перенос миграций с alembic

shared/datebase_base.py

@@ -1,3 +1,5 @@
+import os
+
 import pydantic
 from tortoise import Tortoise
 
@@ -10,6 +12,19 @@ class DatabaseConfig(pydantic.BaseModel):
     MAX_OVERFLOW: int = 10
 
 
+TORTOISE_ORM = {
+    'connections': {'default': os.getenv('DB__URL')},
+    'apps': {
+        'models': {
+            'models': ['src.domain', 'aerich.models'],
+            'default_connection': 'default',
+        },
+    },
+    'use_tz': True,
+    'timezone': 'UTC',
+}
+
+
 class DatabaseBase:
     def __init__(self, config: DatabaseConfig) -> None:
         self.config = config
@@ -17,7 +32,7 @@ class DatabaseBase:
     async def connect(self) -> None:
         await Tortoise.init(
             db_url=str(self.config.URL),
-            modules={'models': ['src.domain']},
+            modules={'models': ['src.domain', 'aerich.models']},
             use_tz=True,
             timezone='UTC',
         )
@@ -47,35 +62,19 @@ class DatabaseBase:
 
         conn = connections.get('default')
 
-        # Проверяем существование таблицы alembic_version
+        # Проверяем существование таблицы aerich
         result = await conn.execute_query(
-            "SELECT EXISTS (SELECT FROM information_schema.tables WHERE table_name = 'alembic_version')"
+            "SELECT EXISTS (SELECT FROM information_schema.tables WHERE table_name = 'aerich')"
         )
         if not result[1][0]['exists']:
-            logging.error('Migrations not applied. Run: alembic upgrade head')
+            logging.warning('Migrations not applied. Run: aerich upgrade')
             return
 
-        # Получаем текущую версию
-        result = await conn.execute_query('SELECT version_num FROM alembic_version')
-        if not result[1]:
-            logging.error('No migration version found in database. Run: alembic upgrade head')
-            return
-
-        current_version = result[1][0]['version_num']
-
-        # Сравниваем с head версией
+        # Получаем последнюю примененную версию
         try:
-            from alembic.config import Config
-            from alembic.script import ScriptDirectory
-
-            alembic_cfg = Config('alembic.ini')
-            script = ScriptDirectory.from_config(alembic_cfg)
-            head_revision = script.get_current_head()
-
-            if head_revision and current_version != head_revision:
-                logging.warning(
-                    f'Database version {current_version} is outdated (latest: {head_revision}). '
-                    f'Run: alembic upgrade head'
-                )
+            result = await conn.execute_query('SELECT version, app FROM aerich ORDER BY id DESC LIMIT 1')
+            if result[1]:
+                current_version = result[1][0]['version']
+                logging.info(f'Current migration version: {current_version}')
         except Exception as e:
-            logging.debug(f'Could not check migration head: {e}')
+            logging.debug(f'Could not check migration version: {e}')