X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sfa%2Fstorage%2Falchemy.py;h=64c39cf8f3062411b70f271bfaa6d633399514f8;hb=30d9951e075d93127c3909dcb41be09b420b3525;hp=e9f96dd6875f42ea4962f8e36cfd63fdf51e41a8;hpb=5f5f29e2bc00befe5b7b601806ef5a7e44349b75;p=sfa.git

diff --git a/sfa/storage/alchemy.py b/sfa/storage/alchemy.py
index e9f96dd6..64c39cf8 100644
--- a/sfa/storage/alchemy.py
+++ b/sfa/storage/alchemy.py
@@ -1,5 +1,3 @@
-from types import StringTypes
-
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 
@@ -14,7 +12,7 @@ from sfa.util.sfalogging import logger
 class Alchemy:
 
     def __init__ (self, config):
-        dbname="sfa"
+        dbname = "sfa"
         # will be created lazily on-demand
         self._session = None
         # the former PostgreSQL.py used the psycopg2 directly and was doing
@@ -39,7 +37,7 @@ class Alchemy:
             except:
                 pass
         self.engine=None
-        raise Exception,"Could not connect to database %s as %s with psycopg2"%(dbname,config.SFA_DB_USER)
+        raise Exception("Could not connect to database %s as %s with psycopg2"%(dbname,config.SFA_DB_USER))
 
 
     # expects boolean True: debug is ON or False: debug is OFF
@@ -49,21 +47,34 @@ class Alchemy:
     def check (self):
         self.engine.execute ("select 1").scalar()
 
-    def session (self):
+    def global_session (self):
         if self._session is None:
             Session=sessionmaker ()
             self._session=Session(bind=self.engine)
+            logger.debug('alchemy.global_session created session %s'%self._session)
         return self._session
 
-    def close_session (self):
+    def close_global_session (self):
         if self._session is None: return
+        logger.debug('alchemy.close_global_session %s'%self._session)
         self._session.close()
         self._session=None
 
+    # create a dbsession to be managed separately
+    def session (self):
+        Session=sessionmaker()
+        session=Session (bind=self.engine)
+        logger.debug('alchemy.session created session %s'%session)
+        return session
+
+    def close_session (self, session):
+        logger.debug('alchemy.close_session closed session %s'%session)
+        session.close()
+
 ####################
 from sfa.util.config import Config
 
 alchemy=Alchemy (Config())
 engine=alchemy.engine
-dbsession=alchemy.session()
+global_dbsession=alchemy.global_session()