19 lines
569 B
Python
19 lines
569 B
Python
import os
|
|
from sqlmodel import create_engine, SQLModel
|
|
from sqlalchemy.orm import sessionmaker
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
|
|
DATABASE_URL = os.getenv("DB_CONNECTION_STRING", "sqlite:///data/wrongmove.db")
|
|
|
|
|
|
debug = os.getenv("ENV", "dev") == "dev"
|
|
engine = create_engine(DATABASE_URL, echo=debug) # `echo=True` for debug logs
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
|
|
|
|
def init_db() -> None:
|
|
"""Create all tables (only for development; use migrations in production)."""
|
|
SQLModel.metadata.create_all(engine)
|