feat: типизация для tortoise ORM

This commit is contained in:
Artem Tsyrulnikov
2025-12-11 12:01:20 +03:00
parent 0f0de396e1
commit 02729ce81b
17 changed files with 181 additions and 263 deletions

View File

@@ -5,7 +5,11 @@ from sqlalchemy import engine_from_config, pool
from src.config import settings
from src.domain import * # noqa
from src.domain import Base
try:
from src.domain import Base as SqlBase # type: ignore[attr-defined]
except ImportError:
SqlBase = None
config = context.config
@@ -13,7 +17,7 @@ if config.config_file_name:
fileConfig(config.config_file_name)
config.set_main_option('sqlalchemy.url', f'{settings.db.URL}?async_fallback=True')
target_metadata = Base.metadata
target_metadata = SqlBase.metadata if SqlBase else None
def run_migrations_offline() -> None: