def db_session(): engine = create_engine(get_connection_string(), echo=False) Session = sessionmaker(bind=engine) session = Session() try: yield session session.commit() except: session.rollback() raise finally: session.close()
from __future__ import with_statement from alembic import context from sqlalchemy import engine_from_config, pool from logging.config import fileConfig import os, sys sys.path.append(os.getcwd()) from misc.util import get_connection_string # this is the Alembic Config object, which provides # access to the values within the .ini file in use. config = context.config config.set_main_option('sqlalchemy.url', get_connection_string()) # Interpret the config file for Python logging. # This line sets up loggers basically. fileConfig(config.config_file_name) # add your model's MetaData object here # for 'autogenerate' support # from myapp import mymodel # target_metadata = mymodel.Base.metadata target_metadata = None # other values from the config, defined by the needs of env.py, # can be acquired: # my_important_option = config.get_main_option("my_important_option") # ... etc. def run_migrations_offline():