Cleaned up session management.

This commit is contained in:
Alexis Lahouze
2013-12-03 23:01:17 +01:00
parent b19bb5fc7a
commit a300db9845
4 changed files with 119 additions and 178 deletions

View File

@ -6,19 +6,19 @@ from sqlalchemy.orm import scoped_session, sessionmaker
db = SQLAlchemy()
@contextmanager
def session_scope(engine):
if engine:
session = scoped_session(sessionmaker(autocommit = False, autoflush = False, bind = engine))
#Base.query = session.query_property()
def session_scope():
#session = scoped_session(sessionmaker(autocommit = False, autoflush = False, bind = engine))
session = db.session
#Base.query = session.query_property()
try:
yield session
session.commit()
except:
session.rollback()
raise
finally:
session.close()
try:
yield session
session.commit()
except:
session.rollback()
raise
finally:
session.close()
import pkgutil