Optimize code

Author: bouwew

plugwise/smile.py

@@ -512,31 +512,26 @@ async def set_schedule_state(self, loc_id: str, name: str, state: str) -> bool:
 
         schema_rule_id: str = next(iter(schema_rule))
 
+        template = (
+            '<template tag="zone_preset_based_on_time_and_presence_with_override" />'
+        )
         if self.smile_name != "Adam":
             locator = f'.//*[@id="{schema_rule_id}"]/template'
             template_id = self._domain_objects.find(locator).attrib["id"]
             template = f'<template id="{template_id}" />'
-            if state == "off":
-                contexts = "<contexts> </contexts>"
-            if state == "on":
-                contexts = f'<contexts><context><zone><location id="{loc_id}" /></zone></context></contexts>'
-        else:  # Adam
-            template = '<template tag="zone_preset_based_on_time_and_presence_with_override" />'
-            locator = f'.//*[@id="{schema_rule_id}"]/contexts'
-            contexts = self._domain_objects.find(locator)
-            locator = f'.//*[@id="{loc_id}"].../...'
-            subject = contexts.find(locator)
-            if subject is None:
-                subject = f'<context><zone><location id="{loc_id}" /></zone></context>'
-                subject = etree.fromstring(subject)
-
-            if state == "off":
-                self._last_active[loc_id] = name
-                contexts.remove(subject)
-            if state == "on":
-                contexts.append(subject)
-
-            contexts = etree.tostring(contexts, encoding="unicode").rstrip()
+
+        locator = f'.//*[@id="{schema_rule_id}"]/contexts'
+        contexts = self._domain_objects.find(locator)
+        subject = f'<context><zone><location id="{loc_id}" /></zone></context>'
+        subject = etree.fromstring(subject)
+
+        if state == "off":
+            self._last_active[loc_id] = name
+            contexts.remove(subject)
+        if state == "on":
+            contexts.append(subject)
+
+        contexts = etree.tostring(contexts, encoding="unicode").rstrip()
 
         uri = f"{RULES};id={schema_rule_id}"
         data = (