45 lines
991 B
Python
45 lines
991 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from sqlalchemy import create_engine
|
||
|
|
from sqlalchemy.engine import Engine
|
||
|
|
from sqlalchemy.orm import Session, sessionmaker
|
||
|
|
|
||
|
|
from app.core.config import get_settings
|
||
|
|
|
||
|
|
_engine: Engine | None = None
|
||
|
|
_session_factory: sessionmaker[Session] | None = None
|
||
|
|
|
||
|
|
|
||
|
|
def configure_session_factory() -> None:
|
||
|
|
global _engine, _session_factory
|
||
|
|
|
||
|
|
settings = get_settings()
|
||
|
|
|
||
|
|
if _engine is not None:
|
||
|
|
_engine.dispose()
|
||
|
|
|
||
|
|
_engine = create_engine(
|
||
|
|
settings.resolved_database_url,
|
||
|
|
echo=settings.sqlalchemy_echo,
|
||
|
|
pool_pre_ping=True,
|
||
|
|
)
|
||
|
|
_session_factory = sessionmaker(bind=_engine, autoflush=False, autocommit=False)
|
||
|
|
|
||
|
|
|
||
|
|
def get_engine() -> Engine:
|
||
|
|
global _engine
|
||
|
|
|
||
|
|
if _engine is None:
|
||
|
|
configure_session_factory()
|
||
|
|
|
||
|
|
return _engine
|
||
|
|
|
||
|
|
|
||
|
|
def get_session_factory() -> sessionmaker[Session]:
|
||
|
|
global _session_factory
|
||
|
|
|
||
|
|
if _session_factory is None:
|
||
|
|
configure_session_factory()
|
||
|
|
|
||
|
|
return _session_factory
|