airpurifier_miot: return OperationMode.Unknown if mode is unknown (#1111

)
rytilahti · Aug 12, 2021 · 59e1603 · 59e1603
1 parent 1859d68
commit 59e1603
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/miio/airpurifier_miot.py b/miio/airpurifier_miot.py
@@ -73,6 +73,7 @@ class AirPurifierMiotException(DeviceException):
 
 
 class OperationMode(enum.Enum):
+    Unknown = -1
     Auto = 0
     Silent = 1
     Favorite = 2
@@ -110,7 +111,12 @@ def aqi(self) -> int:
     @property
     def mode(self) -> OperationMode:
         """Current operation mode."""
-        return OperationMode(self.data["mode"])
+        mode = self.data["mode"]
+        try:
+            return OperationMode(mode)
+        except ValueError:
+            _LOGGER.debug("Unknown mode: %s", mode)
+            return OperationMode.Unknown
 
     @property
     def buzzer(self) -> Optional[bool]: