Add DELETE endpoint for DevicePlant

Author: fjpacheco

app/controller/device_plant_controller.py

@@ -82,3 +82,15 @@ def get_device_plant_relation(req: Request, id_plant: str):
 @withSQLExceptionsHandle
 def get_all_device_plant_relations(req: Request, limit: int):
     return req.app.database.find_all(limit)
+
+
+@withSQLExceptionsHandle
+def delete_device_plant_relation_by_id_device(req: Request, id_device: str):
+    result_rowcount = req.app.database.delete_by_device_id(id_device)
+    if result_rowcount == 0:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="No device-plant relation found with the given device id"
+        )
+
+    return {"message": "Device-plant relation deleted successfully"}

app/database/database.py

@@ -83,3 +83,16 @@ def get_last_measurement(self, id_plant: int) -> Measurement:
         result: Measurement = self.session.scalars(query).one()
 
         return result
+
+    def delete_by_device_id(self, id_device: str) -> int:
+        """
+        Delete a device-plant relation by device id
+        Args:
+            id_device (str): device id
+        Returns:
+            int: number of rows affected. 0 if no rows were affected
+        """
+        query = delete(DevicePlant).where(DevicePlant.id_device == id_device)
+        result = self.session.execute(query)
+        self.session.commit()
+        return result.rowcount

app/router/device_plant_router.py

@@ -55,3 +55,11 @@ async def get_device_plant(req: Request,
     if id_plant is None:
         return controller.get_all_device_plant_relations(req, limit)
     return [controller.get_device_plant_relation(req, id_plant)]
+
+
+@device_plant.delete(
+    "/{id_device}",
+    status_code=status.HTTP_200_OK
+)
+async def delete_device_plant_relation_by_id_device(req: Request, id_device: str):
+    return controller.delete_device_plant_relation_by_id_device(req, id_device)