From bb96122a69e6a8708a7645c2d75bb89e059805da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andrius=20=C5=A0tikonas?= <andrius@stikonas.eu>
Date: Sat, 21 Nov 2020 11:55:21 +0000
Subject: [PATCH] Add get_device_list add devices_sn.

---
 neohub.py | 27 +++++++++++++++++++++++++++
 test.py   |  2 +-
 2 files changed, 28 insertions(+), 1 deletion(-)

diff --git a/neohub.py b/neohub.py
index 4b7a9fc..ce579e2 100644
--- a/neohub.py
+++ b/neohub.py
@@ -216,3 +216,30 @@ class NeoHub:
 
         result = await self._send(message)
         return result
+
+
+    async def get_device_list(self, zone: str):
+        """
+        Returns list of devices associated with zone
+        """
+
+        message = {"GET_DEVICE_LIST": zone}
+
+        result = await self._send(message)
+        if 'error' in result:
+            return False
+        else:
+            return result[zone]
+
+
+    async def devices_sn(self):
+        """
+        Returns serial numbers of attached devices
+
+        {'name': [id, 'serial', 1], ...}
+        """
+
+        message = {"DEVICES_SN": 0}
+
+        result = await self._send(message)
+        return result
diff --git a/test.py b/test.py
index f70367c..d3da8c5 100755
--- a/test.py
+++ b/test.py
@@ -15,7 +15,7 @@ async def run():
     hub = neohub.NeoHub()
     await hub.connect()
     system = await hub.get_system()
-    result = await hub.get_devices()
+    result = await hub.devices_sn()
     print(result)