From b251f0c131ea7f002fd11f15db7c17b285386ad9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andrius=20=C5=A0tikonas?= <andrius@stikonas.eu>
Date: Sat, 21 Nov 2020 11:44:40 +0000
Subject: [PATCH] Add functions to get zones/devices.

---
 neohub.py | 26 ++++++++++++++++++++++++++
 test.py   |  5 +----
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/neohub.py b/neohub.py
index fac0310..4b7a9fc 100644
--- a/neohub.py
+++ b/neohub.py
@@ -190,3 +190,29 @@ class NeoHub:
 
         result = await self._send(message, reply)
         return result
+
+
+    async def get_zones(self):
+        """
+        Returns list of zones and their ids
+
+        {"zone1": 1}
+        """
+
+        message = {"GET_ZONES": 0}
+
+        result = await self._send(message)
+        return result
+
+
+    async def get_devices(self):
+        """
+        Returns list of devices
+
+        {"result": ["device1"]}
+        """
+
+        message = {"GET_DEVICES": 0}
+
+        result = await self._send(message)
+        return result
diff --git a/test.py b/test.py
index 39085ce..f70367c 100755
--- a/test.py
+++ b/test.py
@@ -15,11 +15,8 @@ async def run():
     hub = neohub.NeoHub()
     await hub.connect()
     system = await hub.get_system()
-    print(vars(system))
-    result = await hub.cancel_holiday()
+    result = await hub.get_devices()
     print(result)
-    result = await hub.get_holiday()
-    print(vars(result))
 
 
 logging.basicConfig(level=logging.DEBUG)