| 123456789101112131415 |
- from typing import Iterator
- from sqlmodel import SQLModel, create_engine, Session
- from config.settings import settings
- sqlite_file_name = "database.db"
- sqlite_url = f"sqlite:///" + str(settings.OUTPUT_DIR / sqlite_file_name)
- engine = create_engine(sqlite_url, echo=False)
- def get_session() -> Iterator[Session]:
- with Session(engine) as session:
- yield session
- def create_db_and_tables():
- SQLModel.metadata.create_all(engine)
|
