19 lines
504 B
Python
19 lines
504 B
Python
import os
|
|
|
|
from sqlmodel import Session, SQLModel, create_engine # type: ignore
|
|
|
|
DATABASE_URL = os.getenv("DATABASE_URL", "sqlite:///./notes.db")
|
|
|
|
# Only use check_same_thread for SQLite
|
|
connect_args = {"check_same_thread": False} if DATABASE_URL.startswith("sqlite") else {}
|
|
|
|
engine = create_engine(DATABASE_URL, echo=True, connect_args=connect_args)
|
|
|
|
|
|
def create_db_and_tables():
|
|
SQLModel.metadata.create_all(engine)
|
|
|
|
|
|
def get_session():
|
|
with Session(engine) as session:
|
|
yield session
|