From f51fb1ce3252c79f6c3c657155d481a397ff96d5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andrius=20=C5=A0tikonas?= <andrius@stikonas.eu>
Date: Fri, 20 Nov 2020 23:39:20 +0000
Subject: [PATCH] Add set_format.

---
 neohub.py | 18 ++++++++++++++++++
 test.py   |  6 +++---
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/neohub.py b/neohub.py
index c71568d..7993282 100644
--- a/neohub.py
+++ b/neohub.py
@@ -109,6 +109,7 @@ class NeoHub:
         result = await self._send(message, reply)
         return result
 
+
     async def set_temp_format(self, temp_format):
         """
         Set temperature format to C or F
@@ -119,3 +120,20 @@ class NeoHub:
 
         result = await self._send(message, reply)
         return result
+
+
+    async def set_format(self, format):
+        """
+        Sets timer format
+
+        "NONPROGRAMMABLE" - non programmable (time clocks cannot be non programmable)
+        "24HOURSFIXED" - same format every day of the week
+        "5DAY/2DAY" - 5 day / 2 day
+        "7DAY"      - 7 day (every day different)
+        """
+
+        message = {"SET_FORMAT": format}
+        reply = {"result": "Format was set"}
+
+        result = await self._send(message, reply)
+        return result
diff --git a/test.py b/test.py
index 36cc089..6bb042f 100755
--- a/test.py
+++ b/test.py
@@ -12,9 +12,9 @@ import logging
 async def run():
     hub = neohub.NeoHub()
     await hub.connect()
-    firmware_version = await hub.firmware()
-    print(firmware_version)
-    result = await hub.set_temp_format("C")
+    system = await hub.get_system()
+    print(vars(system))
+    result = await hub.set_format("NONPROGRAMMABLE")
     print(result)