Device support for the xiaomi smart wifi socket added

mirobo/containers.py

@@ -39,6 +39,34 @@ def pretty_time(x: int) -> datetime:
 }
 
 
+class PlugStatus:
+    """Container for status reports from the plug."""
+    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"
+
+    @property
+    def temperature(self) -> float:
+        return self.data["temperature"]
+
+    @property
+    def current(self) -> float:
+        return self.data["current"]
+
+    def __str__(self) -> str:
+        s = "<PlugStatus power=%s, " % (self.state)
+        s += "temperature=%s, " % (self.temperature)
+        s += "current=%s>" % (self.current)
+        return s
+
+
 class VacuumStatus:
     """Container for status reports from the vacuum."""
     def __init__(self, data: Dict[str, Any]) -> None:

mirobo/plug.py

@@ -0,0 +1,23 @@
+from .device import Device
+from .containers import PlugStatus
+
+
+class Plug(Device):
+    """Main class representing the smart wifi socket / plug."""
+
+    def start(self):
+        """Power on."""
+        return self.send("set_power", ["on"])
+
+    def stop(self):
+        """Power off."""
+        return self.send("set_power", ["off"])
+
+    def status(self):
+        """Retrieve properties."""
+        properties = ['power', 'temperature', 'current']
+        values = self.send(
+            "get_prop",
+            properties
+        )
+        return PlugStatus(dict(zip(properties, values)))

mirobo/protocol.py

@@ -20,7 +20,8 @@
 _LOGGER = logging.getLogger(__name__)
 
 # Map of device ids
-xiaomi_devices_reverse = {0x02f2: "Xiaomi Mi Robot Vacuum",
+xiaomi_devices_reverse = {0x02c1: "Xiaomi Mi Smart WiFi Socket",
+                          0x02f2: "Xiaomi Mi Robot Vacuum",
                           0x00c4: "Xiaomi Smart Mi Air Purifier",
                           0x031a: "Xiaomi Smart home gateway",
                           0x0330: "Yeelight color bulb"