Интегрирован wesp в сайт

WESP_REL/migrations/env.py

@@ -0,0 +1,151 @@
+import logging
+import os
+import sys
+from logging.config import fileConfig
+from typing import Optional
+
+from alembic import context
+from sqlalchemy import engine_from_config, event, pool
+
+sys.path.append(os.path.dirname(os.path.dirname(__file__)))
+
+# Подсказка create_app: не поднимать bootstrap/фон до upgrade.
+os.environ.setdefault("WESP_ALEMBIC", "1")
+
+
+def _configure_alembic_logging() -> None:
+    """Логи Alembic без второго console handler (иначе каждая строка ×2).
+
+    При upgrade из create_app root уже настроен — fileConfig из alembic.ini дублировал вывод.
+    Прямой ``alembic upgrade`` из shell (без handlers) — читаем ini как раньше.
+    """
+    mig = logging.getLogger("alembic.runtime.migration")
+    if logging.getLogger().handlers:
+        for name in ("alembic", "alembic.runtime", "alembic.runtime.migration"):
+            logging.getLogger(name).setLevel(logging.INFO)
+        mig.propagate = True
+        return
+    ini = context.config.config_file_name
+    if ini:
+        fileConfig(ini)
+    mig.propagate = bool(mig.handlers)
+
+
+from app import _alembic_host_app, create_app, db  # noqa: E402
+from config import get_config_class  # noqa: E402
+
+config = context.config
+
+_configure_alembic_logging()
+
+# Повторный create_app открывал бы SQLite поверх пулов Flask → database is locked.
+if _alembic_host_app is not None:
+    app = _alembic_host_app
+else:
+    app = create_app(get_config_class(), run_migrations=False)
+
+target_metadata = db.metadata
+
+
+def _recipes_db_url() -> str:
+    """URI основной БД: из окружения (выставляет create_app при автомиграции) или из конфига."""
+    return os.getenv("WESP_RECIPES_DB_URI") or app.config["SQLALCHEMY_DATABASE_URI"]
+
+
+def _reports_db_url() -> Optional[str]:
+    """URI отчётов; None — второй проход Alembic не выполняется."""
+    return os.getenv("WESP_REPORTS_DB_URI") or app.config.get("SQLALCHEMY_BINDS", {}).get(
+        "reports"
+    )
+
+
+def run_migrations_offline() -> None:
+    url = _recipes_db_url()
+    context.configure(
+        url=url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        compare_type=True,
+    )
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def _migration_engine(url: str):
+    from app.services.migration_pace import apply_sqlite_migration_pragmas
+
+    engine = engine_from_config(
+        config.get_section(config.config_ini_section),
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+        url=url,
+    )
+    if not str(url).lower().startswith("sqlite:"):
+        return engine
+
+    @event.listens_for(engine, "connect")
+    def _sqlite_migration_pragmas(dbapi_connection, connection_record):  # noqa: ARG001
+        apply_sqlite_migration_pragmas(dbapi_connection)
+
+    return engine
+
+
+def run_migrations_online() -> None:
+    from app.services.migration_pace import migration_pause, migration_pause_bind
+
+    migration_pause("перед recipes.db")
+    recipes_url = _recipes_db_url()
+    connectable = _migration_engine(recipes_url)
+
+    with connectable.connect() as connection:
+        context.configure(
+            connection=connection,
+            target_metadata=target_metadata,
+            version_table="alembic_version",
+            compare_type=True,
+        )
+
+        with context.begin_transaction():
+            context.run_migrations()  # wesp_2_0 на recipes.db (1-я строка Running upgrade)
+
+        if str(recipes_url).lower().startswith("sqlite:"):
+            from app.services.migration_pace import sqlite_flush_after_migration
+
+            raw = connection.connection.dbapi_connection  # type: ignore[attr-defined]
+            if raw is not None:
+                sqlite_flush_after_migration(raw)
+
+    reports_url = _reports_db_url()
+    if not reports_url:
+        migration_pause("после recipes.db")
+        return
+
+    migration_pause_bind("recipes.db → reports.db")
+    reports_engine = _migration_engine(reports_url)
+
+    with reports_engine.connect() as connection:
+        context.configure(
+            connection=connection,
+            target_metadata=None,
+            version_table="alembic_version_reports",
+        )
+
+        with context.begin_transaction():
+            context.run_migrations(tag="reports")  # та же ревизия на reports.db (2-я строка)
+
+        if str(reports_url).lower().startswith("sqlite:"):
+            from app.services.migration_pace import sqlite_flush_after_migration
+
+            raw_rep = connection.connection.dbapi_connection  # type: ignore[attr-defined]
+            if raw_rep is not None:
+                sqlite_flush_after_migration(raw_rep)
+
+    migration_pause("после reports.db")
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()
+