From 4fadfe31cece35817a29fdb1e730e491e05488e8 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 | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/api/main.py b/api/main.py
index f3d9adad..781dc23d 100644
--- a/api/main.py
+++ b/api/main.py
@@ -59,6 +59,7 @@
     UserGroup,
 )
 from .metrics import Metrics
+from pydantic import BaseModel
 
 
 @asynccontextmanager
@@ -771,6 +772,55 @@ async def put_node(node_id: str, node: Node,
     return obj
 
 
+class NodeUpdateRequest(BaseModel):
+    nodes: List[str]
+    field: str
+    value: str
+
+
+@app.put('/batch/nodeset', response_model=int)
+async def put_batch_nodeset(data: NodeUpdateRequest,
+                            user: str = Depends(get_current_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):
     """Set node ownership information for a hierarchy of nodes"""