mirror of
https://github.com/maxdorninger/MediaManager.git
synced 2026-04-17 23:23:25 +02:00
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
import logging
|
|
import pprint
|
|
from contextvars import ContextVar
|
|
from typing import Annotated, Any, Generator
|
|
|
|
from fastapi import Depends
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session, declarative_base, sessionmaker
|
|
|
|
from .config import DbConfig
|
|
|
|
log = logging.getLogger(__name__)
|
|
config = DbConfig()
|
|
|
|
db_url = "postgresql+psycopg" + "://" + config.USER + ":" + config.PASSWORD + "@" + config.HOST + ":" + str(
|
|
config.PORT) + "/" + config.DBNAME
|
|
|
|
engine = create_engine(db_url, echo=False)
|
|
log.debug("initializing sqlalchemy declarative base")
|
|
Base = declarative_base()
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
|
|
|
|
|
|
def init_db() -> None:
|
|
log.debug("initializing database with following tables")
|
|
for table in Base.metadata.tables:
|
|
log.debug(f"Table: {table.title()}")
|
|
Base.metadata.create_all(engine)
|
|
|
|
|
|
def get_session() -> Generator[Session, Any, None]:
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
db.commit()
|
|
except Exception as e:
|
|
db.rollback()
|
|
log.critical(f"error occurred: {e}")
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
db_session: ContextVar[Session] = ContextVar('db_session')
|
|
|
|
|
|
DbSessionDependency = Annotated[Session, Depends(get_session)]
|