Review changes

Signed-off-by: Jakub Boguščiak <[email protected]>

Author: jbogusciak

uniconfig/python/frinx_worker/uniconfig/connection_manager.py

@@ -77,6 +77,13 @@ def execute(self, worker_input: WorkerInput) -> TaskResult[WorkerOutput]:
             return handle_response(response, self.WorkerOutput)
 
         def _prepare_input(self, worker_input: WorkerInput) -> DictAny:
+            """
+            The input now contains multiple choice nodes (keepalive, credentials and other). Until UniConfig can parse
+            choice nodes, this is a workaround to prepare the input for the installation request correctly.
+            https://frinxhelpdesk.atlassian.net/browse/UNIC-1764
+            :param worker_input: Worker input.
+            :return: Request input data as dict.
+            """
             if worker_input.connection_type == "cli":
                 return {
                     "input": {

uniconfig/python/frinx_worker/uniconfig/device_discovery.py

@@ -38,6 +38,12 @@
 
 
 def _unwrap_data(discovery_input: Input) -> dict[str, Any]:
+    """
+    Unwrapping is necessary because of the input now containing choice wrapper nodes which cannot be parsed by
+    UniConfig yet. https://frinxhelpdesk.atlassian.net/browse/UNIC-1764
+    :param discovery_input: Device Discovery input.
+    :return: Unwrapped data as dict.
+    """
     tcp_port: list[dict[str, Any]] = []
     udp_port: list[dict[str, Any]] = []
     address: list[dict[str, Any]] = []