1 files changed, 33 insertions, 0 deletions
diff --git a/netdisco/discoverables/xiaomi_gw.py b/netdisco/discoverables/xiaomi_gw.py
new file mode 100644
index 0000000..5c9ab0d
--- /dev/null
+++ b/netdisco/discoverables/xiaomi_gw.py
@@ -0,0 +1,33 @@
+"""Discover Xiaomi Mi Home (aka Lumi) Gateways."""
+from . import MDNSDiscoverable
+from ..const import ATTR_MAC_ADDRESS, ATTR_PROPERTIES
+
+
+class Discoverable(MDNSDiscoverable):
+    """Add support for discovering Xiaomi Gateway"""
+
+    def __init__(self, nd):
+        """Initialize the discovery."""
+        super(Discoverable, self).__init__(nd, '_miio._udp.local.')
+
+    def info_from_entry(self, entry):
+        """Return most important info from mDNS entries."""
+        info = super().info_from_entry(entry)
+
+        # Workaround of misparsing of mDNS properties. It's unclear
+        # whether it's bug in zeroconf module or in the Gateway, but
+        # returned properties look like:
+        # {b'poch': b'0:mac=286c07aaaaaa\x00'} instead of expected:
+        # {b'epoch': b'0', b'mac': '286c07aaaaaa'}
+        if "poch" in info[ATTR_PROPERTIES]:
+            misparsed = info[ATTR_PROPERTIES]["poch"]
+            misparsed = misparsed.rstrip("\0")
+            for val in misparsed.split(":"):
+                if val.startswith("mac="):
+                    info[ATTR_MAC_ADDRESS] = val[len("mac="):]
+
+        return info
+
+    def get_entries(self):
+        """Return Xiaomi Gateway devices."""
+        return self.find_by_device_name('lumi-gateway-')