v0.2.1. many features added and cleaned up.

backend/routers/roster_edit.py

@@ -5,7 +5,7 @@ import csv
 import io
 
 from backend.database import get_db
-from backend.models import RosterUnit, IgnoredUnit
+from backend.models import RosterUnit, IgnoredUnit, Emitter
 
 router = APIRouter(prefix="/api/roster", tags=["roster-edit"])
 
@@ -26,9 +26,12 @@ def add_roster_unit(
     device_type: str = Form("seismograph"),
     unit_type: str = Form("series3"),
     deployed: bool = Form(False),
+    retired: bool = Form(False),
     note: str = Form(""),
     project_id: str = Form(None),
     location: str = Form(None),
+    address: str = Form(None),
+    coordinates: str = Form(None),
     # Seismograph-specific fields
     last_calibrated: str = Form(None),
     next_calibration_due: str = Form(None),
@@ -62,9 +65,12 @@ def add_roster_unit(
         device_type=device_type,
         unit_type=unit_type,
         deployed=deployed,
+        retired=retired,
         note=note,
         project_id=project_id,
         location=location,
+        address=address,
+        coordinates=coordinates,
         last_updated=datetime.utcnow(),
         # Seismograph-specific fields
         last_calibrated=last_cal_date,
@@ -96,6 +102,8 @@ def get_roster_unit(unit_id: str, db: Session = Depends(get_db)):
         "note": unit.note or "",
         "project_id": unit.project_id or "",
         "location": unit.location or "",
+        "address": unit.address or "",
+        "coordinates": unit.coordinates or "",
         "last_calibrated": unit.last_calibrated.isoformat() if unit.last_calibrated else "",
         "next_calibration_due": unit.next_calibration_due.isoformat() if unit.next_calibration_due else "",
         "deployed_with_modem_id": unit.deployed_with_modem_id or "",
@@ -115,6 +123,8 @@ def edit_roster_unit(
     note: str = Form(""),
     project_id: str = Form(None),
     location: str = Form(None),
+    address: str = Form(None),
+    coordinates: str = Form(None),
     # Seismograph-specific fields
     last_calibrated: str = Form(None),
     next_calibration_due: str = Form(None),
@@ -152,6 +162,8 @@ def edit_roster_unit(
     unit.note = note
     unit.project_id = project_id
     unit.location = location
+    unit.address = address
+    unit.coordinates = coordinates
     unit.last_updated = datetime.utcnow()
 
     # Seismograph-specific fields
@@ -189,14 +201,27 @@ def set_retired(unit_id: str, retired: bool = Form(...), db: Session = Depends(g
 @router.delete("/{unit_id}")
 def delete_roster_unit(unit_id: str, db: Session = Depends(get_db)):
     """
-    Permanently delete a unit from the roster database.
-    This is different from ignoring - the unit is completely removed.
+    Permanently delete a unit from the database.
+    Checks both roster and emitters tables and deletes from any table where the unit exists.
     """
-    unit = db.query(RosterUnit).filter(RosterUnit.id == unit_id).first()
-    if not unit:
+    deleted = False
+
+    # Try to delete from roster table
+    roster_unit = db.query(RosterUnit).filter(RosterUnit.id == unit_id).first()
+    if roster_unit:
+        db.delete(roster_unit)
+        deleted = True
+
+    # Try to delete from emitters table
+    emitter = db.query(Emitter).filter(Emitter.id == unit_id).first()
+    if emitter:
+        db.delete(emitter)
+        deleted = True
+
+    # If not found in either table, return error
+    if not deleted:
         raise HTTPException(status_code=404, detail="Unit not found")
 
-    db.delete(unit)
     db.commit()
     return {"message": "Unit deleted", "id": unit_id}
 
@@ -274,6 +299,8 @@ async def import_csv(
                 existing_unit.note = row.get('note', existing_unit.note or '')
                 existing_unit.project_id = row.get('project_id', existing_unit.project_id)
                 existing_unit.location = row.get('location', existing_unit.location)
+                existing_unit.address = row.get('address', existing_unit.address)
+                existing_unit.coordinates = row.get('coordinates', existing_unit.coordinates)
                 existing_unit.last_updated = datetime.utcnow()
 
                 results["updated"].append(unit_id)
@@ -287,6 +314,8 @@ async def import_csv(
                     note=row.get('note', ''),
                     project_id=row.get('project_id'),
                     location=row.get('location'),
+                    address=row.get('address'),
+                    coordinates=row.get('coordinates'),
                     last_updated=datetime.utcnow()
                 )
                 db.add(new_unit)