Merge pull request #206 from MTrab:Fix-invalid-zone-detection

Fix invalid zone detection
MTrab · Apr 16, 2024 · 0b53cc9 · 0b53cc9
2 parents e66009b + 9a88283
commit 0b53cc9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyworxcloud/__init__.py b/pyworxcloud/__init__.py
@@ -912,7 +912,10 @@ def setzone(self, serial_number: str, zone: str | int) -> None:
             if not isinstance(zone, int):
                 zone = int(zone)
 
-            if device.zone["starting_point"][zone] == 0:
+            if (
+                zone not in device.zone["starting_point"]
+                or device.zone["starting_point"][zone] == 0
+            ):
                 raise ZoneNotDefined("Cannot request this zone as it is not defined.")
 
             if not zone in device.zone["indicies"]: