Scaffold SQLAlchemy models, routers, and Alembic migrations

- app/models/: Printer, Filament, PrintJob, Todo, PricingConfig with full
  SQLAlchemy 2.x mapped columns and relationships
- app/routers/: list + get-by-id endpoints for all five resources
- app/db.py: SQLAlchemy engine + SessionLocal + get_db dependency
- alembic/: env.py reads MYSQL_* env vars; 0001_initial_schema mirrors db.sql
- alembic.ini: script_location configured, URL injected at runtime

Since db.sql already created the tables, stamp the DB before running future
migrations:  alembic stamp head

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

app/db.py

@@ -1,22 +1,33 @@
 import os
-import pymysql
+
+from sqlalchemy import create_engine, text
+from sqlalchemy.orm import Session, sessionmaker
 
 
-def _params() -> dict:
-    return {
-        "host": os.environ["MYSQL_HOST"],
-        "port": int(os.environ.get("MYSQL_PORT", 3306)),
-        "user": os.environ["MYSQL_USER"],
-        "password": os.environ["MYSQL_PASSWORD"],
-        "database": os.environ["MYSQL_DATABASE"],
-        "connect_timeout": 3,
-    }
+def _db_url() -> str:
+    return (
+        f"mysql+pymysql://{os.environ['MYSQL_USER']}:{os.environ['MYSQL_PASSWORD']}"
+        f"@{os.environ['MYSQL_HOST']}:{os.environ.get('MYSQL_PORT', 3306)}"
+        f"/{os.environ['MYSQL_DATABASE']}"
+    )
+
+
+engine = create_engine(_db_url(), pool_pre_ping=True)
+SessionLocal = sessionmaker(bind=engine)
+
+
+def get_db():
+    db: Session = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
 
 
 def check_db() -> tuple[bool, str]:
     try:
-        conn = pymysql.connect(**_params())
-        conn.close()
+        with engine.connect() as conn:
+            conn.execute(text("SELECT 1"))
         return True, "ok"
     except Exception as e:
         return False, str(e)

app/main.py

@@ -4,6 +4,7 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI
 
 from app.db import check_db
+from app.routers import filaments, pricing, print_jobs, printers, todos
 
 logger = logging.getLogger("pops")
 
@@ -20,6 +21,12 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(title="POPS", lifespan=lifespan)
 
+app.include_router(printers.router)
+app.include_router(filaments.router)
+app.include_router(print_jobs.router)
+app.include_router(todos.router)
+app.include_router(pricing.router)
+
 
 @app.get("/")
 def root():

app/models/__init__.py

@@ -0,0 +1,8 @@
+from app.models.base import Base
+from app.models.filament import Filament
+from app.models.pricing import PricingConfig
+from app.models.print_job import PrintJob
+from app.models.printer import Printer
+from app.models.todo import Todo
+
+__all__ = ["Base", "Printer", "Filament", "PrintJob", "Todo", "PricingConfig"]

app/models/base.py

@@ -0,0 +1,6 @@
+from sqlalchemy.orm import DeclarativeBase
+
+
+class Base(DeclarativeBase):
+    def to_dict(self) -> dict:
+        return {c.name: getattr(self, c.name) for c in self.__table__.columns}

app/models/filament.py

@@ -0,0 +1,38 @@
+import enum
+from datetime import datetime
+from decimal import Decimal
+
+from sqlalchemy import DateTime, Enum, Numeric, String, func
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from app.models.base import Base
+
+
+class Material(enum.Enum):
+    PLA = "PLA"
+    PETG = "PETG"
+    ABS = "ABS"
+    ASA = "ASA"
+    TPU = "TPU"
+    Nylon = "Nylon"
+    PC = "PC"
+    Other = "Other"
+
+
+class Filament(Base):
+    __tablename__ = "filaments"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    brand: Mapped[str | None] = mapped_column(String(100))
+    material: Mapped[Material] = mapped_column(Enum(Material), default=Material.PLA)
+    color: Mapped[str | None] = mapped_column(String(100))
+    color_hex: Mapped[str | None] = mapped_column(String(7))
+    weight_total_g: Mapped[Decimal] = mapped_column(Numeric(8, 2))
+    weight_remaining_g: Mapped[Decimal] = mapped_column(Numeric(8, 2))
+    price_per_kg_eur: Mapped[Decimal | None] = mapped_column(Numeric(8, 2))
+    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime, server_default=func.now(), onupdate=func.now()
+    )
+
+    jobs: Mapped[list["PrintJob"]] = relationship(back_populates="filament")

app/models/pricing.py

@@ -0,0 +1,20 @@
+from datetime import date, datetime
+from decimal import Decimal
+
+from sqlalchemy import Date, DateTime, Numeric, String, func
+from sqlalchemy.orm import Mapped, mapped_column
+
+from app.models.base import Base
+
+
+class PricingConfig(Base):
+    __tablename__ = "pricing_config"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    name: Mapped[str] = mapped_column(String(100), default="default")
+    electricity_kwh_eur: Mapped[Decimal] = mapped_column(Numeric(6, 4), default=Decimal("0.30"))
+    printer_watt: Mapped[Decimal] = mapped_column(Numeric(8, 2), default=Decimal("200.00"))
+    hourly_rate_eur: Mapped[Decimal] = mapped_column(Numeric(8, 2), default=Decimal("0.00"))
+    markup_percent: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=Decimal("20.00"))
+    valid_from: Mapped[date] = mapped_column(Date, server_default=func.current_date())
+    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())

app/models/print_job.py

@@ -0,0 +1,41 @@
+import enum
+from datetime import datetime
+from decimal import Decimal
+
+from sqlalchemy import DateTime, Enum, ForeignKey, Integer, Numeric, String, Text, func
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from app.models.base import Base
+
+
+class JobStatus(enum.Enum):
+    queued = "queued"
+    printing = "printing"
+    done = "done"
+    failed = "failed"
+    cancelled = "cancelled"
+
+
+class PrintJob(Base):
+    __tablename__ = "print_jobs"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    printer_id: Mapped[int] = mapped_column(ForeignKey("printers.id"))
+    filament_id: Mapped[int | None] = mapped_column(ForeignKey("filaments.id"))
+    name: Mapped[str] = mapped_column(String(200))
+    file_name: Mapped[str | None] = mapped_column(String(255))
+    status: Mapped[JobStatus] = mapped_column(Enum(JobStatus), default=JobStatus.queued)
+    queue_position: Mapped[int | None] = mapped_column(Integer)
+    started_at: Mapped[datetime | None] = mapped_column(DateTime)
+    finished_at: Mapped[datetime | None] = mapped_column(DateTime)
+    duration_minutes: Mapped[int | None] = mapped_column(Integer)
+    filament_used_g: Mapped[Decimal | None] = mapped_column(Numeric(8, 2))
+    cost_eur: Mapped[Decimal | None] = mapped_column(Numeric(8, 2))
+    notes: Mapped[str | None] = mapped_column(Text)
+    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime, server_default=func.now(), onupdate=func.now()
+    )
+
+    printer: Mapped["Printer"] = relationship(back_populates="jobs")
+    filament: Mapped["Filament | None"] = relationship(back_populates="jobs")

app/models/printer.py

@@ -0,0 +1,33 @@
+import enum
+from datetime import datetime
+
+from sqlalchemy import DateTime, Enum, String, func
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from app.models.base import Base
+
+
+class PrinterStatus(enum.Enum):
+    idle = "idle"
+    printing = "printing"
+    error = "error"
+    offline = "offline"
+
+
+class Printer(Base):
+    __tablename__ = "printers"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    name: Mapped[str] = mapped_column(String(100))
+    model: Mapped[str | None] = mapped_column(String(100))
+    location: Mapped[str | None] = mapped_column(String(100))
+    status: Mapped[PrinterStatus] = mapped_column(
+        Enum(PrinterStatus), default=PrinterStatus.offline
+    )
+    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime, server_default=func.now(), onupdate=func.now()
+    )
+
+    jobs: Mapped[list["PrintJob"]] = relationship(back_populates="printer")
+    todos: Mapped[list["Todo"]] = relationship(back_populates="printer")

app/models/todo.py

@@ -0,0 +1,21 @@
+from datetime import datetime
+
+from sqlalchemy import Boolean, DateTime, ForeignKey, String, func
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from app.models.base import Base
+
+
+class Todo(Base):
+    __tablename__ = "todos"
+
+    id: Mapped[int] = mapped_column(primary_key=True)
+    title: Mapped[str] = mapped_column(String(255))
+    done: Mapped[bool] = mapped_column(Boolean, default=False)
+    printer_id: Mapped[int | None] = mapped_column(ForeignKey("printers.id"))
+    created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime, server_default=func.now(), onupdate=func.now()
+    )
+
+    printer: Mapped["Printer | None"] = relationship(back_populates="todos")

app/routers/filaments.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from app.db import get_db
+from app.models.filament import Filament
+
+router = APIRouter(prefix="/filaments", tags=["filaments"])
+
+
+@router.get("/")
+def list_filaments(db: Session = Depends(get_db)):
+    return [f.to_dict() for f in db.query(Filament).all()]
+
+
+@router.get("/{filament_id}")
+def get_filament(filament_id: int, db: Session = Depends(get_db)):
+    filament = db.query(Filament).filter(Filament.id == filament_id).first()
+    if not filament:
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="Filament not found")
+    return filament.to_dict()

app/routers/pricing.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from app.db import get_db
+from app.models.pricing import PricingConfig
+
+router = APIRouter(prefix="/pricing", tags=["pricing"])
+
+
+@router.get("/")
+def list_pricing_configs(db: Session = Depends(get_db)):
+    return [p.to_dict() for p in db.query(PricingConfig).all()]
+
+
+@router.get("/{config_id}")
+def get_pricing_config(config_id: int, db: Session = Depends(get_db)):
+    config = db.query(PricingConfig).filter(PricingConfig.id == config_id).first()
+    if not config:
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="Pricing config not found")
+    return config.to_dict()

app/routers/print_jobs.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from app.db import get_db
+from app.models.print_job import PrintJob
+
+router = APIRouter(prefix="/jobs", tags=["jobs"])
+
+
+@router.get("/")
+def list_jobs(db: Session = Depends(get_db)):
+    return [j.to_dict() for j in db.query(PrintJob).all()]
+
+
+@router.get("/{job_id}")
+def get_job(job_id: int, db: Session = Depends(get_db)):
+    job = db.query(PrintJob).filter(PrintJob.id == job_id).first()
+    if not job:
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="Job not found")
+    return job.to_dict()

app/routers/printers.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from app.db import get_db
+from app.models.printer import Printer
+
+router = APIRouter(prefix="/printers", tags=["printers"])
+
+
+@router.get("/")
+def list_printers(db: Session = Depends(get_db)):
+    return [p.to_dict() for p in db.query(Printer).all()]
+
+
+@router.get("/{printer_id}")
+def get_printer(printer_id: int, db: Session = Depends(get_db)):
+    printer = db.query(Printer).filter(Printer.id == printer_id).first()
+    if not printer:
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="Printer not found")
+    return printer.to_dict()

app/routers/todos.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from app.db import get_db
+from app.models.todo import Todo
+
+router = APIRouter(prefix="/todos", tags=["todos"])
+
+
+@router.get("/")
+def list_todos(db: Session = Depends(get_db)):
+    return [t.to_dict() for t in db.query(Todo).all()]
+
+
+@router.get("/{todo_id}")
+def get_todo(todo_id: int, db: Session = Depends(get_db)):
+    todo = db.query(Todo).filter(Todo.id == todo_id).first()
+    if not todo:
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="Todo not found")
+    return todo.to_dict()