| from logging.config import fileConfig |
| from sqlalchemy import engine_from_config |
| from sqlalchemy import pool |
| from alembic import context |
| import sys |
| from pathlib import Path |
|
|
| |
| sys.path.append(str(Path(__file__).parent.parent)) |
|
|
| from app.database.base import Base |
| from app.database import models_intents |
| from app.core.config import settings |
|
|
| config = context.config |
| if config.config_file_name is not None: |
| fileConfig(config.config_file_name) |
|
|
| target_metadata = Base.metadata |
|
|
| def run_migrations_offline() -> None: |
| url = settings.database_url |
| context.configure( |
| url=url, |
| target_metadata=target_metadata, |
| literal_binds=True, |
| dialect_opts={"paramstyle": "named"}, |
| ) |
|
|
| with context.begin_transaction(): |
| context.run_migrations() |
|
|
| def run_migrations_online() -> None: |
| configuration = config.get_section(config.config_ini_section) |
| configuration["sqlalchemy.url"] = settings.database_url |
| connectable = engine_from_config( |
| configuration, |
| 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() |
|
|