now creating dewars, pucks and samples from spreadsheet and replacing dewars if a dewar with the same name exists

backend/app/routers/dewar.py

@@ -1,9 +1,12 @@
 from fastapi import APIRouter, HTTPException, status, Depends
 from sqlalchemy.orm import Session, joinedload
 from typing import List
-import uuid
+import logging
+from sqlalchemy.exc import SQLAlchemyError
+from pydantic import ValidationError
 from app.schemas import Dewar as DewarSchema, DewarCreate, DewarUpdate
-from app.models import Dewar as DewarModel, Puck as PuckModel
+from app.models import Dewar as DewarModel, Puck as PuckModel, \
+    Sample as SampleModel  # Assuming SampleModel is defined in models
 from app.dependencies import get_db
 
 router = APIRouter()
@@ -17,24 +20,55 @@ async def get_dewars(db: Session = Depends(get_db)):
 
 @router.post("/", response_model=DewarSchema, status_code=status.HTTP_201_CREATED)
 async def create_dewar(dewar: DewarCreate, db: Session = Depends(get_db)) -> DewarSchema:
-    db_dewar = DewarModel(
-        dewar_name=dewar.dewar_name,
-        tracking_number=dewar.tracking_number,
-        status=dewar.status,
-        ready_date=dewar.ready_date,
-        shipping_date=dewar.shipping_date,
-        arrival_date=dewar.arrival_date,
-        returning_date=dewar.returning_date,
-        qrcode=dewar.qrcode,
-        contact_person_id=dewar.contact_person_id,
-        return_address_id=dewar.return_address_id
-    )
+    try:
+        db_dewar = DewarModel(
+            dewar_name=dewar.dewar_name,
+            tracking_number=dewar.tracking_number,
+            status=dewar.status,
+            ready_date=dewar.ready_date,
+            shipping_date=dewar.shipping_date,
+            arrival_date=dewar.arrival_date,
+            returning_date=dewar.returning_date,
+            qrcode=dewar.qrcode,
+            contact_person_id=dewar.contact_person_id,
+            return_address_id=dewar.return_address_id
+        )
 
-    db.add(db_dewar)
-    db.commit()
-    db.refresh(db_dewar)
+        db.add(db_dewar)
+        db.commit()
+        db.refresh(db_dewar)
 
-    return db_dewar
+        for puck_data in dewar.pucks:
+            puck = PuckModel(
+                dewar_id=db_dewar.id,
+                puck_name=puck_data.puck_name,
+                puck_type=puck_data.puck_type,
+                puck_location_in_dewar=puck_data.puck_location_in_dewar,
+            )
+            db.add(puck)
+            db.commit()
+            db.refresh(puck)
+
+            for sample_data in puck_data.samples:
+                sample = SampleModel(
+                    puck_id=puck.id,
+                    sample_name=sample_data.sample_name,
+                    position=sample_data.position,
+                    # Ensure only valid attributes are set
+                    data_collection_parameters=sample_data.data_collection_parameters,
+                )
+                db.add(sample)
+                db.commit()
+                db.refresh(sample)
+
+        return db_dewar
+
+    except SQLAlchemyError as e:
+        logging.error(f"Database error occurred: {e}")
+        raise HTTPException(status_code=500, detail="Internal server error")
+    except ValidationError as e:
+        logging.error(f"Validation error occurred: {e}")
+        raise HTTPException(status_code=400, detail="Validation error")
 
 
 @router.get("/{dewar_id}", response_model=DewarSchema)