First draft of the water purifier support

README.md

@@ -14,7 +14,7 @@ Thanks for the nice people over [ioBroker forum](http://forum.iobroker.net/viewt
 ## Supported devices
 
 * Xiaomi Mi Robot Vacuum
-* Xiaomi Mi Air Purifier 2
+* Xiaomi Mi Air Purifier Pro & Air Purifier 2
 * Xiaomi Mi Smart WiFi Socket
 * Xiaomi Mi Smart Socket Plug (1 Socket, 1 USB Port)
 * Xiaomi Smart Power Strip (WiFi, 6 Ports)
@@ -23,6 +23,8 @@ Thanks for the nice people over [ioBroker forum](http://forum.iobroker.net/viewt
 * Xiaomi Philips LED Ball Lamp
 * Xiaomi Universal IR Remote Controller (Chuangmi IR)
 * Xiaomi Mi Smart Fan
+* Xiaomi Mi Air Humidifer
+* Xiaomi Mi Water Purifier (Basic support: Turn on & off)
 * Yeelight light bulbs (only a very rudimentary support, use [python-yeelight](https://gitlab.com/stavros/python-yeelight/) for a more complete support)
 
 *Feel free to create a pull request to add support for new devices as well as additional features for supported devices.*

mirobo/__init__.py

@@ -7,6 +7,7 @@
 from mirobo.plug_v1 import PlugV1
 from mirobo.airpurifier import AirPurifier
 from mirobo.airhumidifier import AirHumidifier
+from mirobo.waterpurifier import WaterPurifier
 from mirobo.strip import Strip
 from mirobo.ceil import Ceil
 from mirobo.philips_eyecare import PhilipsEyecare

mirobo/discovery.py

@@ -4,7 +4,7 @@
 import inspect
 import codecs
 from . import (Device, Vacuum, Plug, PlugV1, Strip, AirPurifier, Ceil,
-               PhilipsEyecare, ChuangmiIr, AirHumidifier)
+               PhilipsEyecare, ChuangmiIr, AirHumidifier, WaterPurifier)
 from typing import Union, Callable, Dict, Optional  # noqa: F401
 
 
@@ -26,7 +26,7 @@
     "zhimi-airpurifier-v6": AirPurifier,
     "chuangmi-ir-v2": ChuangmiIr,
     "zhimi-humidifier-v1": AirHumidifier,
-    # "yunmi-waterpuri-v2": WaterPurifier,
+    "yunmi-waterpuri-v2": WaterPurifier,
     # It looks like philips devices cannot be discovered via mdns
     "philips-light-bulb": Ceil,
     "philips-light-ceil": Ceil,

mirobo/waterpurifier.py

@@ -0,0 +1,57 @@
+import logging
+from typing import Any, Dict
+from collections import defaultdict
+from .device import Device
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class WaterPurifier(Device):
+    """Main class representing the waiter purifier."""
+
+    def status(self):
+        """Retrieve properties."""
+
+        properties = ['power']
+
+        values = self.send(
+            "get_prop",
+            properties
+        )
+
+        properties_count = len(properties)
+        values_count = len(values)
+        if properties_count != values_count:
+            _LOGGER.debug(
+                "Count (%s) of requested properties does not match the "
+                "count (%s) of received values.",
+                properties_count, values_count)
+
+        return WaterPurifierStatus(
+            defaultdict(lambda: None, zip(properties, values)))
+
+    def on(self):
+        """Power on."""
+        return self.send("set_power", ["on"])
+
+    def off(self):
+        """Power off."""
+        return self.send("set_power", ["off"])
+
+
+class WaterPurifierStatus:
+    """Container for status reports from the water purifier."""
+
+    def __init__(self, data: Dict[str, Any]) -> None:
+        self.data = data
+
+    @property
+    def power(self) -> str:
+        return self.data["power"]
+
+    @property
+    def is_on(self) -> bool:
+        return self.power == "on"
+
+    def __str__(self) -> str:
+        return "<WaterPurifierStatus power=%s>" % self.power