52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
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()
|