Break-out determine_contexts() function to reduce complexity

Author: bouwew

plugwise/__init__.py

@@ -602,6 +602,24 @@ async def _set_schedule_state_legacy(
         await self._request(uri, method="put", data=data)
         self._schedule_old_states[loc_id][name] = new_state
 
+    def determine_contexts(self, loc_id: str, schedule_rule_id: str) -> etree
+        """Helper function for set_schedule_state()."""
+        locator = f'.//*[@id="{schedule_rule_id}"]/contexts'
+        contexts = self._domain_objects.find(locator)
+        locator = f'.//*[@id="{loc_id}"].../...'
+        if (subject := contexts.find(locator)) is None:
+            subject = f'<context><zone><location id="{loc_id}" /></zone></context>'
+            subject = etree.fromstring(subject)
+
+        if new_state == "off":
+            self._last_active[loc_id] = name
+            contexts.remove(subject)
+        if new_state == "on":
+            contexts.append(subject)
+
+        contexts = etree.tostring(contexts, encoding="unicode").rstrip()
+        return contexts
+
     async def set_schedule_state(
         self,
         loc_id: str,
@@ -647,21 +665,7 @@ async def set_schedule_state(
             template_id = self._domain_objects.find(locator).attrib["id"]
             template = f'<template id="{template_id}" />'
 
-        locator = f'.//*[@id="{schedule_rule_id}"]/contexts'
-        contexts = self._domain_objects.find(locator)
-        locator = f'.//*[@id="{loc_id}"].../...'
-        if (subject := contexts.find(locator)) is None:
-            subject = f'<context><zone><location id="{loc_id}" /></zone></context>'
-            subject = etree.fromstring(subject)
-
-        if new_state == "off":
-            self._last_active[loc_id] = name
-            contexts.remove(subject)
-        if new_state == "on":
-            contexts.append(subject)
-
-        contexts = etree.tostring(contexts, encoding="unicode").rstrip()
-
+        self.determine_contexts(loc_id, schedule_rule_id)
         uri = f"{RULES};id={schedule_rule_id}"
         data = (
             f'<rules><rule id="{schedule_rule_id}"><name><![CDATA[{name}]]></name>'