Replace xml attrib[] by get()

Author: bouwew

plugwise/common.py

@@ -200,14 +200,14 @@ def _get_groups(self) -> dict[str, GwEntityData]:
 
         for group in self._domain_objects.findall("./group"):
             members: list[str] = []
-            group_id = group.attrib["id"]
+            group_id = group.get("id")
             group_name = group.find("name").text
             group_type = group.find("type").text
             group_appliances = group.findall("appliances/appliance")
             for item in group_appliances:
                 # Check if members are not orphaned - stretch
-                if item.attrib["id"] in self.gw_entities:
-                    members.append(item.attrib["id"])
+                if item.get("id") in self.gw_entities:
+                    members.append(item.get("id"))
 
             if group_type in GROUP_TYPES and members:
                 groups[group_id] = {
@@ -266,7 +266,7 @@ def _get_module_data(
             if key is not None and key not in link_tag:
                 continue
 
-            link_id = appl_search.attrib["id"]
+            link_id = appl_search.get("id")
             loc = f".//services/{link_tag}[@id='{link_id}']...."
             # Not possible to walrus for some reason...
             # xml_2: self._modules for legacy, self._domain_objects for actual

plugwise/helper.py

@@ -117,7 +117,7 @@ def _get_appliances(self) -> None:
         for appliance in self._domain_objects.findall("./appliance"):
             appl = Munch()
             appl.available = None
-            appl.entity_id = appliance.attrib["id"]
+            appl.entity_id = appliance.get("id")
             appl.location = None
             appl.name = appliance.find("name").text
             appl.model = None
@@ -139,7 +139,7 @@ def _get_appliances(self) -> None:
                 continue
 
             if (appl_loc := appliance.find("location")) is not None:
-                appl.location = appl_loc.attrib["id"]
+                appl.location = appl_loc.get("id")
             # Set location to the _home_loc_id when the appliance-location is not found,
             # except for thermostat-devices without a location, they are not active
             elif appl.pwclass not in THERMOSTAT_CLASSES:
@@ -206,7 +206,7 @@ def _get_locations(self) -> None:
         loc = Munch()
         locations = self._domain_objects.findall("./location")
         for location in locations:
-            loc.loc_id = location.attrib["id"]
+            loc.loc_id = location.get("id")
             loc.name = location.find("name").text
             loc._type = location.find("type").text
             self._loc_data[loc.loc_id] = {
@@ -269,7 +269,7 @@ def _appliance_info_finder(self, appl: Munch, appliance: etree.Element) -> Munch
 
     def _appl_gateway_info(self, appl: Munch, appliance: etree.Element) -> Munch:
         """Helper-function for _appliance_info_finder()."""
-        self._gateway_id = appliance.attrib["id"]
+        self._gateway_id = appliance.get("id")
         appl.firmware = str(self.smile.version)
         appl.hardware = self.smile.hw_version
         appl.mac = self.smile.mac_address
@@ -319,7 +319,7 @@ def _get_appliances_with_offset_functionality(self) -> list[str]:
             './/actuator_functionalities/offset_functionality[type="temperature_offset"]/offset/../../..'
         )
         for item in offset_appls:
-            therm_list.append(item.attrib["id"])
+            therm_list.append(item.get("id"))
 
         return therm_list
 
@@ -502,7 +502,7 @@ def _get_plugwise_notifications(self) -> None:
         self._notifications = {}
         for notification in self._domain_objects.findall("./notification"):
             try:
-                msg_id = notification.attrib["id"]
+                msg_id = notification.get("id")
                 msg_type = notification.find("type").text
                 msg = notification.find("message").text
                 self._notifications[msg_id] = {msg_type: msg}
@@ -903,7 +903,7 @@ def _presets(self, loc_id: str) -> dict[str, list[float]]:
             directives = self._domain_objects.find(f'rule[@id="{rule_id}"]/directives')
             for directive in directives:
                 preset = directive.find("then").attrib
-                presets[directive.attrib["preset"]] = [
+                presets[directive.get("preset")] = [
                     float(preset["heating_setpoint"]),
                     float(preset["cooling_setpoint"]),
                 ]
@@ -920,13 +920,13 @@ def _rule_ids_by_name(self, name: str, loc_id: str) -> dict[str, dict[str, str]]
         for rule in self._domain_objects.findall(f'./rule[name="{name}"]'):
             active = rule.find("active").text
             if rule.find(locator) is not None:
-                schedule_ids[rule.attrib["id"]] = {
+                schedule_ids[rule.get("id")] = {
                     "location": loc_id,
                     "name": name,
                     "active": active,
                 }
             else:
-                schedule_ids[rule.attrib["id"]] = {
+                schedule_ids[rule.get("id")] = {
                     "location": NONE,
                     "name": name,
                     "active": active,
@@ -947,13 +947,13 @@ def _rule_ids_by_tag(self, tag: str, loc_id: str) -> dict[str, dict[str, str]]:
                 name = rule.find("name").text
                 active = rule.find("active").text
                 if rule.find(locator2) is not None:
-                    schedule_ids[rule.attrib["id"]] = {
+                    schedule_ids[rule.get("id")] = {
                         "location": loc_id,
                         "name": name,
                         "active": active,
                     }
                 else:
-                    schedule_ids[rule.attrib["id"]] = {
+                    schedule_ids[rule.get("id")] = {
                         "location": NONE,
                         "name": name,
                         "active": active,
@@ -1002,6 +1002,6 @@ def _thermostat_uri(self, loc_id: str) -> str:
         Determine the location-set_temperature uri - from LOCATIONS.
         """
         locator = f'./location[@id="{loc_id}"]/actuator_functionalities/thermostat_functionality'
-        thermostat_functionality_id = self._domain_objects.find(locator).attrib["id"]
+        thermostat_functionality_id = self._domain_objects.find(locator).get("id")
 
         return f"{LOCATIONS};id={loc_id}/thermostat;id={thermostat_functionality_id}"

plugwise/legacy/helper.py

@@ -107,7 +107,7 @@ def _all_appliances(self) -> None:
                 continue  # pragma: no cover
 
             appl.location = self._home_loc_id
-            appl.entity_id = appliance.attrib["id"]
+            appl.entity_id = appliance.get("id")
             appl.name = appliance.find("name").text
             # Extend device_class name when a Circle/Stealth is type heater_central -- Pw-Beta Issue #739
             if (
@@ -148,7 +148,7 @@ def _all_locations(self) -> None:
             return
 
         for location in locations:
-            loc.loc_id = location.attrib["id"]
+            loc.loc_id = location.get("id")
             loc.name = location.find("name").text
             loc._type = location.find("type").text
             # Filter the valid single location for P1 legacy: services not empty
@@ -395,8 +395,8 @@ def _presets(self) -> dict[str, list[float]]:
         for directive in self._domain_objects.findall("rule/directives/when/then"):
             if directive is not None and directive.get("icon") is not None:
                 # Ensure list of heating_setpoint, cooling_setpoint
-                presets[directive.attrib["icon"]] = [
-                    float(directive.attrib["temperature"]),
+                presets[directive.get("icon")] = [
+                    float(directive.get("temperature")),
                     0,
                 ]
 
@@ -411,7 +411,7 @@ def _schedules(self) -> tuple[list[str], str]:
         search = self._domain_objects
         if (result := search.find("./rule[name='Thermostat schedule']")) is not None:
             name = "Thermostat schedule"
-            rule_id = result.attrib["id"]
+            rule_id = result.get("id")
 
         log_type = "schedule_state"
         locator = f"./appliance[type='thermostat']/logs/point_log[type='{log_type}']/period/measurement"
@@ -432,5 +432,5 @@ def _schedules(self) -> tuple[list[str], str]:
     def _thermostat_uri(self) -> str:
         """Determine the location-set_temperature uri - from APPLIANCES."""
         locator = "./appliance[type='thermostat']"
-        appliance_id = self._appliances.find(locator).attrib["id"]
+        appliance_id = self._appliances.find(locator).get("id")
         return f"{APPLIANCES};id={appliance_id}/thermostat"

plugwise/legacy/smile.py

@@ -160,7 +160,7 @@ async def set_preset(self, _: str, preset: str) -> None:
             raise PlugwiseError("Plugwise: invalid preset.")
 
         locator = f'rule/directives/when/then[@icon="{preset}"].../.../...'
-        rule_id = self._domain_objects.find(locator).attrib["id"]
+        rule_id = self._domain_objects.find(locator).get("id")
         data = f"<rules><rule id='{rule_id}'><active>true</active></rule></rules>"
         await self.call_request(RULES, method="put", data=data)
 
@@ -192,7 +192,7 @@ async def set_schedule_state(
         schedule_rule_id: str | None = None
         for rule in self._domain_objects.findall("rule"):
             if rule.find("name").text == name:
-                schedule_rule_id = rule.attrib["id"]
+                schedule_rule_id = rule.get("id")
                 break
 
         if schedule_rule_id is None:
@@ -205,7 +205,7 @@ async def set_schedule_state(
             new_state = "true"
 
         locator = f'.//*[@id="{schedule_rule_id}"]/template'
-        template_id = self._domain_objects.find(locator).attrib["id"]
+        template_id = self._domain_objects.find(locator).get("id")
 
         data = (
             "<rules>"

plugwise/smile.py

@@ -178,7 +178,7 @@ async def set_number(
         if th_func_list := self._domain_objects.findall(locator):
             for th_func in th_func_list:
                 if th_func.find("type").text == key:
-                    thermostat_id = th_func.attrib["id"]
+                    thermostat_id = th_func.get("id")
 
         if thermostat_id is None:
             raise PlugwiseError(f"Plugwise: cannot change setpoint, {key} not found.")
@@ -358,7 +358,7 @@ async def set_schedule_state(
         )
         if self.check_name(ANNA):
             locator = f'.//*[@id="{schedule_rule_id}"]/template'
-            template_id = self._domain_objects.find(locator).attrib["id"]
+            template_id = self._domain_objects.find(locator).get("id")
             template = f'<template id="{template_id}" />'
 
         contexts = self.determine_contexts(loc_id, new_state, schedule_rule_id)
@@ -427,10 +427,10 @@ async def set_switch_state(
             # multiple types of e.g. toggle_functionality present
             if (sw_type := item.find("type")) is not None:
                 if sw_type.text == switch.act_type:
-                    switch_id = item.attrib["id"]
+                    switch_id = item.get("id")
                     break
             else:  # actuators with a single item like relay_functionality
-                switch_id = item.attrib["id"]
+                switch_id = item.get("id")
 
         uri = f"{APPLIANCES};id={appl_id}/{switch.device};id={switch_id}"
         if model == "relay":
@@ -456,7 +456,7 @@ async def _set_groupswitch_member_state(
         switched = 0
         for member in members:
             locator = f'appliance[@id="{member}"]/{switch.actuator}/{switch.func_type}'
-            switch_id = self._domain_objects.find(locator).attrib["id"]
+            switch_id = self._domain_objects.find(locator).get("id")
             uri = f"{APPLIANCES};id={member}/{switch.device};id={switch_id}"
             lock_blocked = self.gw_entities[member]["switches"].get("lock")
             # Assume Plugs under Plugwise control are not part of a group

plugwise/util.py

@@ -86,7 +86,7 @@ def check_heater_central(xml: etree.Element) -> str:
     hc_list: list[dict[str, bool]] = []
     for heater_central in xml.findall(locator):
         hc_count += 1
-        hc_id: str = heater_central.attrib["id"]
+        hc_id: str = heater_central.get("id")
         has_actuators: bool = (
             heater_central.find("actuator_functionalities/") is not None
         )