"""Alembic environment для ВЕКТОРа. URL: env DATABASE_URL > alembic.ini sqlalchemy.url. Метаданные моделей: backend.models (Base). """ from __future__ import annotations import os import sys from logging.config import fileConfig from sqlalchemy import engine_from_config, pool from alembic import context # backend/ и корень репо в sys.path, чтобы import backend.models работал # при любом CWD (alembic -c backend/alembic.ini из корня или из backend/) _BACKEND_DIR = os.path.dirname(os.path.abspath(__file__)) for _p in (_BACKEND_DIR, os.path.dirname(_BACKEND_DIR)): if _p not in sys.path: sys.path.insert(0, _p) config = context.config if config.config_file_name is not None: fileConfig(config.config_file_name) # URL: env побеждает ini db_url = os.getenv('DATABASE_URL') if db_url: config.set_main_option('sqlalchemy.url', db_url) from backend.models import Base # noqa: E402 (после set_main_option) target_metadata = Base.metadata def run_migrations_offline() -> None: """Запуск без БД (--sql): выгрузить SQL в скрипт.""" url = config.get_main_option('sqlalchemy.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: """Обычный запуск: подключиться и накатить.""" 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) with context.begin_transaction(): context.run_migrations() if context.is_offline_mode(): run_migrations_offline() else: run_migrations_online()