def apply_migrations(db_dsn=None):
db_dsn = db_dsn or settings.DB_DSN
current_dir = Path(__name__).parent
config_dir = Path(current_dir) / 'ketl' / 'db' / 'config'
alembic_cfg = Config(file_=str(config_dir / 'alembic.ini'))
alembic_cfg.set_main_option('script_location', str(config_dir))
alembic_cfg.set_main_option('sqlalchemy.url', db_dsn)
command.upgrade(alembic_cfg, 'head')
def update_db(update=True):
db_connection = DatabaseConnection(database=options.sql_database,
hostname=options.sql_host,
port=options.sql_port,
username=options.sql_user,
password=options.sql_password,
dialect=options.sql_dialect)
alembic_cfg = Config('alembic/alembic.ini')
alembic_cfg.attributes['configure_logger'] = False
alembic_cfg.set_main_option('sqlalchemy.url', unquote_plus(str(db_connection)))
if update:
command.upgrade(alembic_cfg, "head")
else:
command.stamp(alembic_cfg, "head")
def update_db(update=True):
db_connection = DatabaseConnection(database=options.sql_database,
hostname=options.sql_host,
port=options.sql_port,
username=options.sql_user,
password=options.sql_password,
dialect=options.sql_dialect)
alembic_cfg = Config('alembic/alembic.ini')
alembic_cfg.attributes['configure_logger'] = False
alembic_cfg.set_main_option('sqlalchemy.url', unquote_plus(str(db_connection)))
if update:
command.upgrade(alembic_cfg, "head")
else:
command.stamp(alembic_cfg, "head")
def update_db(update=True):
logging.info("Checking for Database Updates...")
db_connection = DatabaseConnection(
database=options.sql_database,
hostname=options.sql_host,
port=options.sql_port,
username=options.sql_user,
password=options.sql_password,
dialect=options.sql_dialect,
)
alembic_cfg = Config("alembic/alembic.ini")
alembic_cfg.attributes["configure_logger"] = False
alembic_cfg.set_main_option("sqlalchemy.url", unquote(str(db_connection)))
if update:
command.upgrade(alembic_cfg, "head")
else:
command.stamp(alembic_cfg, "head")