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()