Daily Work MCP Server

from __future__ import annotations from collections.abc import Generator from sqlalchemy import create_engine from sqlalchemy.orm import DeclarativeBase, Session, sessionmaker from app.core.config import get_settings class Base(DeclarativeBase): """Base declarative class shared by all ORM models.""" settings = get_settings() connect_args = {"check_same_thread": False} if settings.database_url.startswith("sqlite") else {} engine = create_engine(settings.database_url, connect_args=connect_args) SessionLocal = sessionmaker(bind=engine, autoflush=False, autocommit=False, expire_on_commit=False) def get_session() -> Generator[Session, None, None]: """Provide a SQLAlchemy session for request scope usage.""" session = SessionLocal() try: yield session finally: session.close()