12 lines
296 B
Python
12 lines
296 B
Python
import os
|
|
from sqlmodel import create_engine, SQLModel, Session
|
|
|
|
DATA_DIR = "data"
|
|
os.makedirs(DATA_DIR, exist_ok=True)
|
|
|
|
engine = create_engine(f"sqlite:///{DATA_DIR}/database.db")
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
def get_session():
|
|
with Session(engine) as session:
|
|
yield session |