feat: Implement read, create & delete operation on drone table

hotosm · Jul 29, 2024 · 2bd16ca · 2bd16ca
1 parent 6f50087
commit 2bd16ca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/src/backend/app/drones/drone_crud.py b/src/backend/app/drones/drone_crud.py
@@ -125,4 +125,6 @@ async def create_drone(db: Database, drone_info: drone_schemas.DroneIn):
 
     except Exception as e:
         log.exception(e)
-        raise HTTPException(e) from e
+        raise HTTPException(
+            status_code=HTTPStatus.INTERNAL_SERVER_ERROR, detail="Drone creation failed"
+        ) from e
diff --git a/src/backend/app/drones/drone_schemas.py b/src/backend/app/drones/drone_schemas.py
@@ -1,7 +1,7 @@
 from pydantic import BaseModel
 
 
-class DroneBase(BaseModel):
+class DroneIn(BaseModel):
     model: str
     manufacturer: str
     camera_model: str
@@ -16,9 +16,6 @@ class DroneBase(BaseModel):
     weight: float
 
 
-class DroneIn(DroneBase):
-    pass
-
-
-class DroneOut(DroneBase):
+class DroneOut(BaseModel):
     id: int
+    model: str