from sqlalchemy.orm import Session
from app.infrastructure.database.database import SessionLocal
from app.infrastructure.database.models import Base


def get_db() -> Session:
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()