Air Humidifier: Handle poweroff exception on set_mode

rytilahti · Apr 10, 2019 · bafb3a0 · bafb3a0
1 parent feea7fc
commit bafb3a0
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/miio/airhumidifier.py b/miio/airhumidifier.py
@@ -6,7 +6,7 @@
 import click
 
 from .click_common import command, format_output, EnumType
-from .device import Device, DeviceInfo, DeviceException
+from .device import Device, DeviceInfo, DeviceError, DeviceException
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -329,7 +329,14 @@ def off(self):
     )
     def set_mode(self, mode: OperationMode):
         """Set mode."""
-        return self.send("set_mode", [mode.value])
+        try:
+            return self.send("set_mode", [mode.value])
+        except DeviceError as error:
+            # {'code': -6011, 'message': 'device_poweroff'}
+            if error.code == -6011:
+                self.on()
+                return self.send("set_mode", [mode.value])
+            raise
 
     @command(
         click.argument("brightness", type=EnumType(LedBrightness, False)),