p
16 lines
416 B
Python
16 lines
416 B
Python
from sqlmodel import SQLModel, create_engine, Session
|
|
from app.core.config import settings
|
|
|
|
connect_args = {}
|
|
if "sqlite" in settings.DATABASE_URL:
|
|
connect_args["check_same_thread"] = False
|
|
|
|
engine = create_engine(settings.DATABASE_URL, echo=False, connect_args=connect_args)
|
|
|
|
def init_db():
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
def get_session():
|
|
with Session(engine) as session:
|
|
yield session
|