Fix findings

Author: GabyUnalaq

ankaios_sdk/_components/complete_state.py

@@ -69,7 +69,10 @@
 from .workload import Workload
 from .workload_state import WorkloadStateCollection
 from .manifest import Manifest
-from ..utils import SUPPORTED_API_VERSION, _to_config_item
+from ..utils import SUPPORTED_API_VERSION, _to_config_item, get_logger
+
+
+logger = get_logger()
 
 
 class CompleteState:
@@ -95,19 +98,35 @@ def __init__(self,
         """
         self._complete_state = _ank_base.CompleteState()
         self._set_api_version(SUPPORTED_API_VERSION)
+        if _proto:
+            self._complete_state = _proto
+            logger.debug(
+                "CompleteState initialized from proto message"
+            )
+            return
         if manifest:
             self._complete_state.desiredState.CopyFrom(
                 manifest._to_desired_state()
             )
+            logger.debug(
+                "CompleteState initialized from manifest"
+            )
+            return
         if configs:
             self.set_configs(configs)
+            logger.debug(
+                "CompleteState initialized from configs"
+            )
+            return
         if workloads:
             self._complete_state.desiredState.workloads.workloads.clear()
             for workload in workloads:
                 self._complete_state.desiredState.workloads.workloads[
                     workload.name].CopyFrom(workload._to_proto())
-        if _proto:
-            self._complete_state = _proto
+            logger.debug(
+                "CompleteState initialized from workloads"
+            )
+            return
 
     def __str__(self) -> str:
         """

ankaios_sdk/utils.py

@@ -31,7 +31,7 @@
 """
 
 import logging
-from typing import Union, List, Dict
+from typing import Union
 from enum import Enum
 import threading
 from ._protos import _ank_base
@@ -91,7 +91,7 @@ def get_logger(name="Ankaios logger"):
     return logger
 
 
-def _to_config_item(item: Union[str, List, Dict]
+def _to_config_item(item: Union[str, list, dict]
                     ) -> _ank_base.ConfigItem:
     """
     Returns an ank_base.ConfigItem from a base type.
@@ -102,10 +102,10 @@ def _to_config_item(item: Union[str, List, Dict]
     config_item = _ank_base.ConfigItem()
     if isinstance(item, str):
         config_item.String = item
-    elif isinstance(item, List):
+    elif isinstance(item, list):
         for value in [_to_config_item(value) for value in item]:
             config_item.array.values.append(value)
-    elif isinstance(item, Dict):
+    elif isinstance(item, dict):
         for key, value in item.items():
             config_item.object.fields[key]. \
                 CopyFrom(_to_config_item(value))