From 4118d73486021b95b755886f1355878eca3dba8c Mon Sep 17 00:00:00 2001
From: Denys Fedoryshchenko <denys.f@collabora.com>
Date: Thu, 5 Dec 2024 01:01:41 +0200
Subject: [PATCH] feat(main.py): Implement /nodes/set endpoint

Implement batch "field to value" bulk set to improve speed of some
batch operations.
This way we can update field on multiple nodes to particular value
at once.

Signed-off-by: Denys Fedoryshchenko <denys.f@collabora.com>
---
 api/main.py | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/api/main.py b/api/main.py
index f3d9adad..c8272991 100644
--- a/api/main.py
+++ b/api/main.py
@@ -770,6 +770,53 @@ async def put_node(node_id: str, node: Node,
         await db.create(evhist)
     return obj
 
+class NodeUpdateRequest(BaseModel):
+    nodes: List[str]
+    field: str
+    value: str
+
+@app.put('/nodes/set', response_model=int)
+async def put_nodes_set(data: NodeUpdateRequest,
+                        user: str = Depends(authorize_user)):
+    """
+    Set a field to a value for multiple nodes
+    TBD: Make db.bulkupdate to update multiple nodes in one go
+    """
+    metrics.add('http_requests_total', 1)
+    updated = 0
+    nodes = data.nodes
+    field = data.field
+    value = data.value
+    for node_id in nodes:
+        node_from_id = await db.find_by_id(Node, node_id)
+        if not node_from_id:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail=f"Node not found with id: {node_id}"
+            )
+        # verify ownership
+        if not user.username == node_from_id.owner:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail="Unauthorized to complete the operation"
+            )
+        # right now we support only field:
+        # processed_by_kcidb_bridge, also value should be boolean
+        if field == 'processed_by_kcidb_bridge':
+            if value == 'true' or value == 'True':
+                value = True
+            elif value == 'false' or value == 'False':
+                value = False
+            setattr(node_from_id, field, value)
+            await db.update(node_from_id)
+            updated += 1
+        else:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="Field not supported"
+            )
+    return updated
+
 
 async def _set_node_ownership_recursively(user: User, hierarchy: Hierarchy,
                                           submitter: str, treeid: str):