Доработка вебхука

services/ingest-service/alembic/env.py

@@ -1,29 +1,22 @@
 from logging.config import fileConfig
-from sqlalchemy import engine_from_config
-from sqlalchemy import pool
 from alembic import context
-import os
-
+import asyncio
 from dotenv import load_dotenv
+
 load_dotenv()
 
-from app.core.database import Base
+from app.infrastructure.database import engine, Base
 # Важно! Импортируем все модели, чтобы они зарегистрировались в Base.metadata
 from app.models import CallEvent
 
 config = context.config
 
-# Переопределяем URL из переменных окружения
-database_url = os.getenv("DATABASE_URL", "postgresql://postgres:postgres@localhost:5432/ingest_db")
-# Преобразуем asyncpg URL в psycopg2
-sync_database_url = database_url.replace("postgresql+asyncpg://", "postgresql://")
-config.set_main_option("sqlalchemy.url", sync_database_url)
-
 if config.config_file_name is not None:
     fileConfig(config.config_file_name)
 
 target_metadata = Base.metadata
 
+
 def run_migrations_offline() -> None:
     """Run migrations in 'offline' mode.
 
@@ -48,26 +41,25 @@ def run_migrations_offline() -> None:
         context.run_migrations()
 
 
+def do_run_migrations(connection):
+    """Запускаем миграции с существующим подключением"""
+    context.configure(connection=connection, target_metadata=target_metadata)
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+async def run_async_migrations():
+    """Асинхронный запуск миграций через async движок"""
+    async with engine.connect() as connection:
+        await connection.run_sync(do_run_migrations)
+
+    await engine.dispose()
+
+
 def run_migrations_online() -> None:
-    """Run migrations in 'online' mode.
-
-    In this scenario we need to create an Engine
-    and associate a connection with the context.
-
-    """
-    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()
+    """Run migrations in 'online' mode через async движок"""
+    asyncio.run(run_async_migrations())
 
 
 if context.is_offline_mode():