app/core/db.py - master - tsps/ai_tools

"""
Подключение к PostgreSQL через SQLAlchemy Core.
Синхронный движок (psycopg), как в case_loader
"""

from sqlalchemy import create_engine, Engine
from sqlalchemy.pool import QueuePool

from app.core.config import config

engine: Engine = create_engine(
    config.DATABASE_URL,
    poolclass=QueuePool,
    pool_size=5,
    max_overflow=10,
    pool_pre_ping=True  # автоматическая проверка "живости" соединения
)


def get_db_connection():
    """
    Возвращает контекст-менеджер для соединения с БД.
    Использование:
        with get_db_connection() as conn:
            conn.execute(stmt)
            conn.commit()
    """
    return engine.connect()


def close_db_engine():
    """Закрыть пул соединений (используется в тестах / при завершении сервиса)."""
    engine.dispose()