def db_session(db_engine):
"""Get a standalone database session for preparing database state."""
from h import db
session = db.Session(bind=db_engine)
yield session
session.close()
def db_session(request, config):
"""Get a standalone database session for preparing database state."""
from h import db
engine = db.make_engine(config.registry.settings)
session = db.Session(bind=engine)
try:
yield session
finally:
session.close()
engine.dispose()
def remove_old_flags(event):
"""Remove old feature flags from the database."""
# Skip this if we're in a script, not actual app startup. See the comment
# in h.cli:main for an explanation.
if 'H_SCRIPT' in os.environ:
return
engine = db.make_engine(event.app.registry.settings)
session = db.Session(bind=engine)
Feature.remove_old_flags(session)
session.commit()
session.close()
engine.dispose()
def _get_session(settings):
engine = db.make_engine(settings)
return db.Session(bind=engine)
def get_session(settings):
"""Get a DB session from the provided settings."""
return db.Session(bind=db.make_engine(settings))
