1
0
mirror of https://gitlab.com/MoonTestUse1/AdministrationItDepartmens.git synced 2025-08-14 00:25:46 +02:00

Fix database

This commit is contained in:
MoonTestUse1
2025-01-07 06:33:02 +06:00
parent 362290a332
commit 92d6b9c753

View File

@@ -1,6 +1,7 @@
"""Database module"""
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, declarative_base
from typing import Generator
from .core.config import settings
@@ -18,6 +19,14 @@ engine = create_engine(
# Создаем фабрику сессий
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
def get_db() -> Generator:
"""Get database session"""
db = SessionLocal()
try:
yield db
finally:
db.close()
def init_db():
"""Initialize database"""
# Импортируем модели здесь, чтобы избежать циклических зависимостей