| from __future__ import with_statement, print_function, absolute_import |
| |
| import os |
| |
| from alembic import context |
| from sqlalchemy import create_engine, pool |
| from logging.config import fileConfig |
| |
| |
| |
| config = context.config |
| |
| |
| |
| fileConfig(config.config_file_name) |
| |
| |
| if 'PAGURE_CONFIG' not in os.environ \ |
| and os.path.exists('/etc/pagure/pagure.cfg'): |
| print('Using configuration file `/etc/pagure/pagure.cfg`') |
| os.environ['PAGURE_CONFIG'] = '/etc/pagure/pagure.cfg' |
| |
| |
| try: |
| import pagure |
| import pagure.lib.model |
| except ImportError: |
| import sys |
| sys.path.insert(0, '.') |
| import pagure |
| import pagure.lib.model |
| |
| |
| |
| |
| target_metadata = pagure.lib.model.BASE.metadata |
| |
| |
| |
| |
| |
| |
| DBURL = config.get_main_option("sqlalchemy.url") |
| if not DBURL: |
| DBURL = pagure.config.config['DB_URL'] |
| |
| |
| def run_migrations_offline(): |
| """Run migrations in 'offline' mode. |
| |
| This configures the context with just a URL |
| and not an Engine, though an Engine is acceptable |
| here as well. By skipping the Engine creation |
| we don't even need a DBAPI to be available. |
| |
| Calls to context.execute() here emit the given string to the |
| script output. |
| |
| """ |
| |
| context.configure(url=DBURL, target_metadata=target_metadata) |
| |
| with context.begin_transaction(): |
| context.run_migrations() |
| |
| |
| def run_migrations_online(): |
| """Run migrations in 'online' mode. |
| |
| In this scenario we need to create an Engine |
| and associate a connection with the context. |
| |
| """ |
| connectable = create_engine(DBURL, 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() |