Format codebase

frinx_conductor_workers/frinx_conductor_workers/cli_worker_test.py

@@ -98,19 +98,16 @@
 }
 
 cli_node_already_installed = {
-  "errors": {
-    "error": [
-      {
-        "error-tag": "data-exists",
-        "error-info": {
-          "node-id": "xr6",
-          "topology-id": "protocol"
-        },
-        "error-message": "Node has already been installed using other protocols",
-        "error-type": "application"
-      }
-    ]
-  }
+    "errors": {
+        "error": [
+            {
+                "error-tag": "data-exists",
+                "error-info": {"node-id": "xr6", "topology-id": "protocol"},
+                "error-message": "Node has already been installed using other protocols",
+                "error-type": "application",
+            }
+        ]
+    }
 }
 
 

frinx_conductor_workers/frinx_conductor_workers/logging_helpers.py

@@ -1,4 +1,5 @@
 """This module contains helper functions and objects for logging."""
+
 import io
 import logging
 

frinx_conductor_workers/frinx_conductor_workers/netconf_worker.py

@@ -102,9 +102,9 @@ def execute_mount_netconf(task):
         and task["inputData"]["between-attempts-timeout-millis"] is not None
         and task["inputData"]["between-attempts-timeout-millis"] is not ""
     ):
-        mount_body["input"]["netconf"][
-            "netconf-node-topology:between-attempts-timeout-millis"
-        ] = task["inputData"]["between-attempts-timeout-millis"]
+        mount_body["input"]["netconf"]["netconf-node-topology:between-attempts-timeout-millis"] = (
+            task["inputData"]["between-attempts-timeout-millis"]
+        )
 
     if (
         "connection-timeout-millis" in task["inputData"]

frinx_conductor_workers/frinx_conductor_workers/netconf_worker_test.py

@@ -84,19 +84,16 @@
 }
 
 netconf_node_already_installed = {
-  "errors": {
-    "error": [
-      {
-        "error-tag": "data-exists",
-        "error-info": {
-          "node-id": "xr6",
-          "topology-id": "protocol"
-        },
-        "error-message": "Node has already been installed using other protocols",
-        "error-type": "application"
-      }
-    ]
-  }
+    "errors": {
+        "error": [
+            {
+                "error-tag": "data-exists",
+                "error-info": {"node-id": "xr6", "topology-id": "protocol"},
+                "error-message": "Node has already been installed using other protocols",
+                "error-type": "application",
+            }
+        ]
+    }
 }
 
 

frinx_conductor_workers/frinx_conductor_workers/uniconfig_worker_test.py

@@ -76,7 +76,7 @@
             "node-result": [
                 {
                     "node-id": "xr5",
-                    "configuration": "2019-09-13T08:37:28.331: configure terminal\n2019-09-13T08:37:28.536: interface GigabitEthernet0/0/0/1\nshutdown\nroot\n\n2019-09-13T08:37:28.536: commit\n2019-09-13T08:37:28.536: end\n"
+                    "configuration": "2019-09-13T08:37:28.331: configure terminal\n2019-09-13T08:37:28.536: interface GigabitEthernet0/0/0/1\nshutdown\nroot\n\n2019-09-13T08:37:28.536: commit\n2019-09-13T08:37:28.536: end\n",
                 }
             ]
         },
@@ -94,10 +94,10 @@
                             "data-actual": '{\n  "frinx-openconfig-interfaces:config": {\n    "type": "iana-if-type:ethernetCsmacd",\n    "enabled": false,\n    "name": "GigabitEthernet0/0/0/0"\n  }\n}',
                             "data-intended": '{\n  "frinx-openconfig-interfaces:config": {\n    "type": "iana-if-type:ethernetCsmacd",\n    "enabled": false,\n    "name": "GigabitEthernet0/0/0/0dfhdfghd"\n  }\n}',
                         }
-                    ]
+                    ],
                 }
             ]
-        },
+        }
     }
 }
 
@@ -432,9 +432,7 @@ def test_calculate_diff_with_non_existing_device(self):
 class TestSyncFromNetwork(unittest.TestCase):
     def test_sync_from_network_with_existing_devices(self):
         with patch("frinx_conductor_workers.uniconfig_worker.requests.post") as mock:
-            mock.return_value = MockResponse(
-                bytes(json.dumps({}), encoding="utf-8"), 204, ""
-            )
+            mock.return_value = MockResponse(bytes(json.dumps({}), encoding="utf-8"), 204, "")
             request = frinx_conductor_workers.uniconfig_worker.sync_from_network(
                 {"inputData": {"devices": "xr5, xr6"}}
             )
@@ -455,9 +453,7 @@ def test_sync_from_network_with_non_existing_device(self):
 class TestReplaceConfigWithOper(unittest.TestCase):
     def test_replace_config_with_oper_with_existing_devices(self):
         with patch("frinx_conductor_workers.uniconfig_worker.requests.post") as mock:
-            mock.return_value = MockResponse(
-                bytes(json.dumps({}), encoding="utf-8"), 204, ""
-            )
+            mock.return_value = MockResponse(bytes(json.dumps({}), encoding="utf-8"), 204, "")
             request = frinx_conductor_workers.uniconfig_worker.replace_config_with_oper(
                 {"inputData": {"devices": "xr5, xr6"}}
             )

frinx_python_sdk/src/frinx/common/worker/worker.py

@@ -32,14 +32,11 @@ class WorkerImpl(ABC):
     task_def: TaskDefinition = None
     task_def_template: type[BaseTaskdef] | type[DefaultTaskDefinition] = None
 
-    class WorkerDefinition(TaskDefinition):
-        ...
+    class WorkerDefinition(TaskDefinition): ...
 
-    class WorkerInput(TaskInput):
-        ...
+    class WorkerInput(TaskInput): ...
 
-    class WorkerOutput(TaskOutput):
-        ...
+    class WorkerOutput(TaskOutput): ...
 
     def __init__(
         self, task_def_template: type[BaseTaskdef] | type[DefaultTaskDefinition] = None

frinx_python_sdk/src/frinx/common/workflow/task.py

@@ -173,7 +173,12 @@ class DynamicForkTask(WorkflowTaskImpl):
     type = TaskType.FORK_JOIN_DYNAMIC
     dynamic_fork_tasks_param: str = Field(default="dynamicTasks")
     dynamic_fork_tasks_input_param_name: str = Field(default="dynamicTasksInput")
-    input_parameters: DynamicForkArraysTaskInputParameters | DynamicForkTaskInputParameters | DynamicForkArraysTaskFromDefInputParameters | DynamicForkTaskFromDefInputParameters
+    input_parameters: (
+        DynamicForkArraysTaskInputParameters
+        | DynamicForkTaskInputParameters
+        | DynamicForkArraysTaskFromDefInputParameters
+        | DynamicForkTaskFromDefInputParameters
+    )
 
 
 class ForkTask(WorkflowTaskImpl):

frinx_python_sdk/src/frinx/services/inventory/inventory_worker.py

@@ -153,9 +153,11 @@ def add_device(
                 labelIds=label_ids if type(label_ids) is not None else None,
                 vendor=vendor if type(vendor) is not None and len(vendor) > 0 else None,
                 model=model if type(model) is not None and len(model) > 0 else None,
-                deviceSize=str(device_size)
-                if type(device_size) is not None and len(device_size) > 0
-                else None,
+                deviceSize=(
+                    str(device_size)
+                    if type(device_size) is not None and len(device_size) > 0
+                    else None
+                ),
             )
         )
 

frinx_python_sdk/src/frinx/workers/inventory/inventory_worker.py

@@ -130,8 +130,7 @@ class WorkerDefinition(TaskDefinition):
             timeout_seconds = 3600
             response_timeout_seconds = 3600
 
-        class WorkerInput(TaskInput):
-            ...
+        class WorkerInput(TaskInput): ...
 
         class WorkerOutput(TaskOutput):
             url: str

frinx_python_sdk/src/frinx/workers/uniconfig/uniconfig_worker.py

@@ -232,8 +232,7 @@ class WorkerDefinition(TaskDefinition):
         class WorkerInput(TaskInput):
             uniconfig_contexts: str
 
-        class WorkerOutput(TaskOutput):
-            ...
+        class WorkerOutput(TaskOutput): ...
 
         def execute(self, task: Task) -> TaskResult:
             response = uniconfig.rollback_all_tx(**task.input_data)