feat: типизация для tortoise ORM
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user