added pucks and samples

backend/app/routers/puck.py

@@ -0,0 +1,63 @@
+from fastapi import APIRouter, HTTPException, status, Depends
+from sqlalchemy.orm import Session
+from typing import List
+import uuid
+from app.schemas import Puck as PuckSchema, PuckCreate, PuckUpdate
+from app.models import Puck as PuckModel, Sample as SampleModel
+from app.dependencies import get_db
+
+router = APIRouter()
+
+
+@router.get("/", response_model=List[PuckSchema])
+async def get_pucks(db: Session = Depends(get_db)):
+    return db.query(PuckModel).all()
+
+
+@router.get("/{puck_id}", response_model=PuckSchema)
+async def get_puck(puck_id: str, db: Session = Depends(get_db)):
+    puck = db.query(PuckModel).filter(PuckModel.id == puck_id).first()
+    if not puck:
+        raise HTTPException(status_code=404, detail="Puck not found")
+    return puck
+
+
+@router.post("/", response_model=PuckSchema, status_code=status.HTTP_201_CREATED)
+async def create_puck(puck: PuckCreate, db: Session = Depends(get_db)) -> PuckSchema:
+    puck_id = f'PUCK-{uuid.uuid4().hex[:8].upper()}'
+    db_puck = PuckModel(
+        id=puck_id,
+        puck_name=puck.puck_name,
+        puck_type=puck.puck_type,
+        puck_location_in_dewar=puck.puck_location_in_dewar,
+        dewar_id=puck.dewar_id
+    )
+    db.add(db_puck)
+    db.commit()
+    db.refresh(db_puck)
+    return db_puck
+
+
+@router.put("/{puck_id}", response_model=PuckSchema)
+async def update_puck(puck_id: str, updated_puck: PuckUpdate, db: Session = Depends(get_db)):
+    puck = db.query(PuckModel).filter(PuckModel.id == puck_id).first()
+    if not puck:
+        raise HTTPException(status_code=404, detail="Puck not found")
+
+    for key, value in updated_puck.dict(exclude_unset=True).items():
+        setattr(puck, key, value)
+
+    db.commit()
+    db.refresh(puck)
+    return puck
+
+
+@router.delete("/{puck_id}", status_code=status.HTTP_204_NO_CONTENT)
+async def delete_puck(puck_id: str, db: Session = Depends(get_db)):
+    puck = db.query(PuckModel).filter(PuckModel.id == puck_id).first()
+    if not puck:
+        raise HTTPException(status_code=404, detail="Puck not found")
+
+    db.delete(puck)
+    db.commit()
+    return