nifi add update property

noharm-ai · Dec 12, 2024 · 6a44324 · 6a44324
1 parent 6b85379
commit 6a44324
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/models/enums.py b/models/enums.py
@@ -130,6 +130,7 @@ class NifiQueueActionTypeEnum(Enum):
     TERMINATE_PROCESS = "TERMINATE_PROCESS"
     CUSTOM_CALLBACK = "CUSTOM_CALLBACK"
     REFRESH_TEMPLATE = "REFRESH_TEMPLATE"
+    UPDATE_PROPERTY = "UPDATE_PROPERTY"
 
 
 class DrugAlertTypeEnum(Enum):

diff --git a/services/admin/admin_integration_remote_service.py b/services/admin/admin_integration_remote_service.py
@@ -247,6 +247,13 @@ def _get_new_queue(id_processor: str, action_type: str, data: dict):
     elif NifiQueueActionTypeEnum.REFRESH_TEMPLATE.value == action_type:
         queue.url = f"nifi-api/system-diagnostics"
         queue.method = "GET"
+    elif NifiQueueActionTypeEnum.UPDATE_PROPERTY.value == action_type:
+        queue.url = f"nifi-api/processors/{escape(id_processor)}"
+        queue.method = "PUT"
+        queue.body = {
+            "id": id_processor,
+            "config": {"properties": data["properties"]},
+        }
 
     return queue