19 lines
400 B
Python
19 lines
400 B
Python
from __future__ import annotations
|
|
|
|
from sqlmodel import SQLModel, create_engine, Session
|
|
|
|
from .config import settings
|
|
|
|
engine = create_engine(
|
|
settings.database_url,
|
|
connect_args={"check_same_thread": False} if settings.database_url.startswith("sqlite") else {},
|
|
)
|
|
|
|
|
|
def init_db() -> None:
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
|
|
def get_session() -> Session:
|
|
return Session(engine)
|