23 lines
779 B
Python
23 lines
779 B
Python
import os
|
|
from sqlmodel import create_engine, SQLModel
|
|
from sqlalchemy.orm import sessionmaker
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
|
|
# PostgreSQL example (or use "sqlite:///database.db" for SQLite)
|
|
# DATABASE_URL = "postgresql://user:password@localhost/db_name"
|
|
# DATABASE_URL = "sqlite:///data/wrongmove.db"
|
|
# DATABASE_URL = "mysql://wrongmove:wrongmove@localhost:3306/wrongmove"
|
|
DATABASE_URL = os.environ["DB_CONNECTION_STRING"]
|
|
|
|
|
|
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():
|
|
"""Create all tables (only for development; use migrations in production)."""
|
|
SQLModel.metadata.create_all(engine)
|