# Alembic env.py — подключение к БД через Settings. import os import sys from logging.config import fileConfig from alembic import context from sqlalchemy import engine_from_config, pool # Добавляем корень проекта в sys.path sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) from iaai_scraper.core.config import Settings from iaai_scraper.storage.models import Base config = context.config VERSION_TABLE = "iaai_parser_alembic_version" # Logging if config.config_file_name is not None: fileConfig(config.config_file_name) # Подставляем URL из Settings (env vars имеют приоритет) settings = Settings() config.set_main_option("sqlalchemy.url", settings.database.url) target_metadata = Base.metadata def run_migrations_offline() -> None: # Run migrations in 'offline' mode. url = config.get_main_option("sqlalchemy.url") context.configure( url=url, target_metadata=target_metadata, version_table=VERSION_TABLE, literal_binds=True, dialect_opts={"paramstyle": "named"}, ) with context.begin_transaction(): context.run_migrations() def run_migrations_online() -> None: # Run migrations in 'online' mode. 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, version_table=VERSION_TABLE, ) with context.begin_transaction(): context.run_migrations() if context.is_offline_mode(): run_migrations_offline() else: run_migrations_online()