server/build: fix alembic environment script

This commit is contained in:
rr- 2017-02-05 23:29:21 +01:00
parent 72056e0cd2
commit 74c583f11d
1 changed files with 3 additions and 3 deletions

View File

@ -9,7 +9,7 @@ import logging.config
dir_to_self = os.path.dirname(os.path.realpath(__file__)) dir_to_self = os.path.dirname(os.path.realpath(__file__))
sys.path.append(os.path.join(dir_to_self, *[os.pardir] * 2)) sys.path.append(os.path.join(dir_to_self, *[os.pardir] * 2))
import szurubooru.db.base import szurubooru.model.base
import szurubooru.config import szurubooru.config
alembic_config = alembic.context.config alembic_config = alembic.context.config
@ -18,7 +18,7 @@ logging.config.fileConfig(alembic_config.config_file_name)
szuru_config = szurubooru.config.config szuru_config = szurubooru.config.config
alembic_config.set_main_option('sqlalchemy.url', szuru_config['database']) alembic_config.set_main_option('sqlalchemy.url', szuru_config['database'])
target_metadata = szurubooru.db.Base.metadata target_metadata = szurubooru.model.Base.metadata
def run_migrations_offline(): def run_migrations_offline():
@ -51,7 +51,7 @@ def run_migrations_online():
connectable = sa.engine_from_config( connectable = sa.engine_from_config(
alembic_config.get_section(alembic_config.config_ini_section), alembic_config.get_section(alembic_config.config_ini_section),
prefix='sqlalchemy.', prefix='sqlalchemy.',
poolclass=sqlalchemy.pool.NullPool) poolclass=sa.pool.NullPool)
with connectable.connect() as connection: with connectable.connect() as connection:
alembic.context.configure( alembic.context.configure(