72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
from datetime import date
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Query
|
|
from pydantic import BaseModel
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.auth import get_current_user
|
|
from app.database import get_db
|
|
from app.models.weight import WeightEntry
|
|
|
|
router = APIRouter(
|
|
prefix="/api/weight", tags=["weight"], dependencies=[Depends(get_current_user)]
|
|
)
|
|
|
|
|
|
class WeightCreate(BaseModel):
|
|
date: date
|
|
weight_lbs: float
|
|
body_fat_pct: float | None = None
|
|
notes: str | None = None
|
|
|
|
|
|
class WeightRead(WeightCreate):
|
|
id: int
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
@router.get("", response_model=list[WeightRead])
|
|
def list_weight(
|
|
from_date: date | None = Query(None, alias="from"),
|
|
to_date: date | None = Query(None, alias="to"),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
q = select(WeightEntry).order_by(WeightEntry.date.desc())
|
|
if from_date:
|
|
q = q.where(WeightEntry.date >= from_date)
|
|
if to_date:
|
|
q = q.where(WeightEntry.date <= to_date)
|
|
return db.scalars(q).all()
|
|
|
|
|
|
@router.post("", response_model=WeightRead, status_code=201)
|
|
def create_weight(body: WeightCreate, db: Session = Depends(get_db)):
|
|
entry = WeightEntry(**body.model_dump())
|
|
db.add(entry)
|
|
db.commit()
|
|
db.refresh(entry)
|
|
return entry
|
|
|
|
|
|
@router.put("/{entry_id}", response_model=WeightRead)
|
|
def update_weight(entry_id: int, body: WeightCreate, db: Session = Depends(get_db)):
|
|
entry = db.get(WeightEntry, entry_id)
|
|
if not entry:
|
|
raise HTTPException(status_code=404, detail="Entry not found")
|
|
for key, val in body.model_dump().items():
|
|
setattr(entry, key, val)
|
|
db.commit()
|
|
db.refresh(entry)
|
|
return entry
|
|
|
|
|
|
@router.delete("/{entry_id}", status_code=204)
|
|
def delete_weight(entry_id: int, db: Session = Depends(get_db)):
|
|
entry = db.get(WeightEntry, entry_id)
|
|
if not entry:
|
|
raise HTTPException(status_code=404, detail="Entry not found")
|
|
db.delete(entry)
|
|
db.commit()
|