added pucks and samples

backend/app/routers/dewar.py

@@ -1,17 +1,19 @@
 from fastapi import APIRouter, HTTPException, status, Depends
-from sqlalchemy.orm import Session
+from sqlalchemy.orm import Session, joinedload
 from typing import List
 import uuid
-from app.schemas import Dewar as DewarSchema, DewarCreate
-from app.models import Dewar as DewarModel
+from app.schemas import Dewar as DewarSchema, DewarCreate, DewarUpdate, Sample as SampleSchema, Puck as PuckSchema
+from app.models import Dewar as DewarModel, Puck as PuckModel, Sample as SampleModel
 from app.dependencies import get_db
 
 router = APIRouter()
 
+
 @router.get("/", response_model=List[DewarSchema])
 async def get_dewars(db: Session = Depends(get_db)):
     return db.query(DewarModel).all()
 
+
 @router.post("/", response_model=DewarSchema, status_code=status.HTTP_201_CREATED)
 async def create_dewar(dewar: DewarCreate, db: Session = Depends(get_db)) -> DewarSchema:
     dewar_id = f'DEWAR-{uuid.uuid4().hex[:8].upper()}'
@@ -35,4 +37,32 @@ async def create_dewar(dewar: DewarCreate, db: Session = Depends(get_db)) -> Dew
     db.add(db_dewar)
     db.commit()
     db.refresh(db_dewar)
-    return db_dewar
+    return db_dewar
+
+
+@router.get("/{dewar_id}", response_model=DewarSchema)
+async def get_dewar(dewar_id: str, db: Session = Depends(get_db)):
+    dewar = db.query(DewarModel).options(
+        joinedload(DewarModel.pucks).joinedload(PuckModel.positions)
+    ).filter(DewarModel.id == dewar_id).first()
+
+    if not dewar:
+        raise HTTPException(status_code=404, detail="Dewar not found")
+
+    return dewar
+
+
+@router.put("/{dewar_id}", response_model=DewarSchema)
+async def update_dewar(dewar_id: str, dewar_update: DewarUpdate, db: Session = Depends(get_db)) -> DewarSchema:
+    dewar = db.query(DewarModel).filter(DewarModel.id == dewar_id).first()
+
+    if not dewar:
+        raise HTTPException(status_code=404, detail="Dewar not found")
+
+    for key, value in dewar_update.dict(exclude_unset=True).items():
+        setattr(dewar, key, value)
+
+    db.commit()
+    db.refresh(dewar)
+
+    return dewar