Fix issue with returning invisible operation modes

Author: klejejs

ThermiaOnlineAPI/api/ThermiaAPI.py

@@ -286,14 +286,18 @@ def get_group_operational_operation(self, device: ThermiaHeatPump):
 
         if operation_modes_data is not None:
             operation_modes_map = map(
-                lambda values: {
-                    values.get("value"): values.get("name").split(
-                        "REG_VALUE_OPERATION_MODE_"
-                    )[1],
-                },
+                lambda values: (
+                    {
+                        values.get("value"): values.get("name").split(
+                            "REG_VALUE_OPERATION_MODE_"
+                        )[1],
+                    }
+                    if values.get("visible")
+                    else {}
+                ),
                 operation_modes_data,
             )
-            operation_modes_list = list(operation_modes_map)
+            operation_modes_list = list(filter(lambda x: x != {}, operation_modes_map))
             operation_modes = ChainMap(*operation_modes_list)
 
             current_operation_mode = [
@@ -677,7 +681,9 @@ def __authenticate(self) -> bool:
                 "client_id": THERMIA_AZURE_AUTH_CLIENT_ID_AND_SCOPE,
                 "redirect_uri": THERMIA_AZURE_AUTH_REDIRECT_URI,
                 "scope": THERMIA_AZURE_AUTH_CLIENT_ID_AND_SCOPE,
-                "code": request_confirmed.url.split("code=")[1],
+                "code": utils.get_list_value_or_default(
+                    request_confirmed.url.split("code="), 1, ""
+                ),
                 "code_verifier": code_challenge,
                 "grant_type": "authorization_code",
             }

ThermiaOnlineAPI/utils/utils.py

@@ -23,6 +23,13 @@ def get_dict_value_or_default(dictionary, key, default: T) -> T:
     return dictionary[key]
 
 
+def get_list_value_or_default(list, idx, default: T) -> T:
+    try:
+        return list[idx]
+    except IndexError:
+        return default
+
+
 def pretty_print(json_object):
     print(json.dumps(json_object, indent=4, sort_keys=True))
     print("\n")