Spaces:
Sleeping
Sleeping
File size: 388 Bytes
3abfc90 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | from sqlmodel import SQLModel, create_engine, Session
from .config import get_settings
settings = get_settings()
engine = create_engine(
settings.DATABASE_URL,
echo=True,
connect_args={"check_same_thread": False} # Needed for SQLite
)
def init_db():
SQLModel.metadata.create_all(engine)
def get_session():
with Session(engine) as session:
yield session
|