from __future__ import with_statement
from alembic import context
from sqlalchemy import engine_from_config, pool
from logging.config import fileConfig
import os
import sys
basepath = os.path.dirname(__file__)
parent_dir = os.path.abspath(os.path.join(basepath, ".."))
sys.path.append(parent_dir)
from dave.config import config as dave_config
config = context.config
fileConfig(config.config_file_name)
from dave import models
target_metadata = models.Base.metadata
def run_migrations_offline():
url = dave_config["database"]
context.configure(
url=url, target_metadata=target_metadata, literal_binds=True)
with context.begin_transaction():
context.run_migrations()
def run_migrations_online():
connectable = engine_from_config(
config.get_section(config.config_ini_section),
prefix='sqlalchemy.',
poolclass=pool.NullPool)
with connectable.connect() as connection:
context.configure(
connection=connection,
target_metadata=target_metadata
)
with context.begin_transaction():
context.run_migrations()
if context.is_offline_mode():
run_migrations_offline()
else:
run_migrations_online()