Thomas Peetz
27 lines
756 B
Python
27 lines
756 B
Python
import logging
|
|
import os
|
|
from typing import Annotated
|
|
|
|
from fastapi import Depends
|
|
from src.schema.base import Base
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker, Session
|
|
|
|
connect_string = ('mariadb+mariadbconnector://{}:{}@{}:{}/{}'.format(
|
|
os.environ.get('DB_USER', 'kontor'),
|
|
os.environ.get('DB_PASSWORD', 'kontor'),
|
|
os.environ.get('DB_HOST', 'mariadb'),
|
|
os.environ.get('DB_PORT', 3306),
|
|
os.environ.get('DB_NAME', 'kontor')
|
|
))
|
|
engine = create_engine(connect_string)
|
|
SessionLocal = sessionmaker(bind=engine)
|
|
Base.metadata.create_all(bind=engine, checkfirst=True)
|
|
|
|
def get_db():
|
|
logging.info("get_db")
|
|
with SessionLocal() as db:
|
|
yield db
|
|
|
|
SessionDep = Annotated[Session, Depends(get_db)]
|