105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
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, SetTellPosition, PuckEventCreate
|
|
from app.models import Puck as PuckModel, Sample as SampleModel, PuckEvent as PuckEventModel
|
|
from app.dependencies import get_db
|
|
from datetime import datetime
|
|
|
|
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
|
|
|
|
@router.put("/{puck_id}/tell_position", status_code=status.HTTP_200_OK)
|
|
async def set_tell_position(
|
|
puck_id: int,
|
|
request: SetTellPosition,
|
|
db: Session = Depends(get_db)
|
|
):
|
|
# Get the requested tell_position
|
|
tell_position = request.tell_position
|
|
|
|
# Define valid positions
|
|
valid_positions = [f"{letter}{num}" for letter in "ABCDEF" for num in range(1, 6)] + ["null", None]
|
|
|
|
# Validate tell_position
|
|
if tell_position not in valid_positions:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail=f"Invalid tell_position value. Must be one of {valid_positions}.",
|
|
)
|
|
|
|
# Set the correct tell_position logic
|
|
actual_position = None if tell_position in ["null", None] else tell_position
|
|
|
|
# Create a new PuckEvent (always a new event, even with null/None)
|
|
new_puck_event = PuckEventModel(
|
|
puck_id=puck_id,
|
|
tell_position=actual_position, # Null for disassociation, else the valid position
|
|
event_type="tell_position_set", # Example event type
|
|
timestamp=datetime.utcnow(),
|
|
)
|
|
db.add(new_puck_event)
|
|
db.commit()
|
|
db.refresh(new_puck_event)
|
|
|
|
# Send the response
|
|
return {
|
|
"message": "New tell position event created successfully",
|
|
"tell_position": new_puck_event.tell_position,
|
|
"timestamp": new_puck_event.timestamp,
|
|
}
|