Initial commit: Sproutly plant tracking app

backend/routers/varieties.py

@@ -0,0 +1,51 @@
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.orm import Session
+from typing import List
+from database import get_db
+from models import Variety
+from schemas import VarietyCreate, VarietyUpdate, VarietyOut
+
+router = APIRouter(prefix="/varieties", tags=["varieties"])
+
+
+@router.get("/", response_model=List[VarietyOut])
+def list_varieties(db: Session = Depends(get_db)):
+    return db.query(Variety).order_by(Variety.category, Variety.name).all()
+
+
+@router.get("/{variety_id}", response_model=VarietyOut)
+def get_variety(variety_id: int, db: Session = Depends(get_db)):
+    v = db.query(Variety).filter(Variety.id == variety_id).first()
+    if not v:
+        raise HTTPException(status_code=404, detail="Variety not found")
+    return v
+
+
+@router.post("/", response_model=VarietyOut, status_code=201)
+def create_variety(data: VarietyCreate, db: Session = Depends(get_db)):
+    v = Variety(**data.model_dump())
+    db.add(v)
+    db.commit()
+    db.refresh(v)
+    return v
+
+
+@router.put("/{variety_id}", response_model=VarietyOut)
+def update_variety(variety_id: int, data: VarietyUpdate, db: Session = Depends(get_db)):
+    v = db.query(Variety).filter(Variety.id == variety_id).first()
+    if not v:
+        raise HTTPException(status_code=404, detail="Variety not found")
+    for field, value in data.model_dump().items():
+        setattr(v, field, value)
+    db.commit()
+    db.refresh(v)
+    return v
+
+
+@router.delete("/{variety_id}", status_code=204)
+def delete_variety(variety_id: int, db: Session = Depends(get_db)):
+    v = db.query(Variety).filter(Variety.id == variety_id).first()
+    if not v:
+        raise HTTPException(status_code=404, detail="Variety not found")
+    db.delete(v)
+    db.commit()