File size: 763 Bytes
e92d5c0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
from contextlib import contextmanager
from typing import Generator
from sqlalchemy.engine import Engine
from sqlmodel import Session, SQLModel, create_engine
from . import constants
_ENGINE_SINGLETON: Engine | None = None
def get_engine() -> Engine:
"""Get the engine."""
global _ENGINE_SINGLETON
if _ENGINE_SINGLETON is None:
_ENGINE_SINGLETON = create_engine(constants.DATABASE_URL)
return _ENGINE_SINGLETON
@contextmanager
def get_session() -> Generator[Session, None, None]:
"""Get a session from the engine."""
engine = get_engine()
with Session(engine) as session:
yield session
def init_db() -> None:
"""Initialize the database."""
engine = get_engine()
SQLModel.metadata.create_all(engine)
|