|
|
|
|
@ -1,21 +1,14 @@
|
|
|
|
|
from sqlalchemy import create_engine |
|
|
|
|
from sqlalchemy.ext.declarative import declarative_base |
|
|
|
|
from sqlalchemy.orm import sessionmaker, Session |
|
|
|
|
from autoLoader.config import DATABASE_URL |
|
|
|
|
from contextlib import contextmanager |
|
|
|
|
|
|
|
|
|
# Создание engine |
|
|
|
|
engine = create_engine( |
|
|
|
|
DATABASE_URL, |
|
|
|
|
connect_args={"check_same_thread": False} if "sqlite" in DATABASE_URL else {} |
|
|
|
|
) |
|
|
|
|
# Создаём engine, но используем Base из apiApp.database |
|
|
|
|
from apiApp.database import Base, engine |
|
|
|
|
|
|
|
|
|
# SessionLocal |
|
|
|
|
# SessionLocal (используем тот же engine) |
|
|
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) |
|
|
|
|
|
|
|
|
|
# Base (используем тот же Base, что и в apiApp.database) |
|
|
|
|
from apiApp.database import Base |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Зависимость для получения сессии БД (для FastAPI) |
|
|
|
|
def get_db(): |
|
|
|
|
@ -41,7 +34,7 @@ def get_db_session():
|
|
|
|
|
db.close() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# Импортируем модели из apiApp.database |
|
|
|
|
from apiApp.database.Audio import Audio |
|
|
|
|
from apiApp.database.AiConclusion import AiConclusion |
|
|
|
|
from apiApp.database.ConclusionVersion import ConclusionVersion |
|
|
|
|
# Импортируем ТОЛЬКО нужные модели из apiApp.database |
|
|
|
|
# Operator не импортируем, чтобы избежать проблем с Call |
|
|
|
|
from autoLoader.database.Audio import Audio |
|
|
|
|
|
|
|
|
|
|