From 14051edfc5eaafb2306335b4921ed95867531ddd Mon Sep 17 00:00:00 2001
From: Tanner Collin <git@tannercollin.com>
Date: Sun, 19 Dec 2021 05:29:10 +0000
Subject: [PATCH] Misc

---
 main.py | 85 +++++++++++++++++++++++++++++++++++----------------------
 1 file changed, 52 insertions(+), 33 deletions(-)

diff --git a/main.py b/main.py
index fd0f971..ad9b520 100644
--- a/main.py
+++ b/main.py
@@ -1,10 +1,12 @@
 import os, logging
 DEBUG = os.environ.get('DEBUG')
+NO_WATCHDOG = os.environ.get('NO_WATCHDOG')
 logging.basicConfig(
         format='[%(asctime)s] %(levelname)s %(module)s/%(funcName)s - %(message)s',
         level=logging.DEBUG if DEBUG else logging.INFO)
 logging.getLogger('aiohttp').setLevel(logging.DEBUG if DEBUG else logging.WARNING)
 
+import json
 import os
 import sys
 import asyncio
@@ -24,21 +26,30 @@ RELAY_OFF = True
 
 allow_watchdog = False
 
+cooldown_time = time.time()
+
 def set_relay(pin, state):
     GPIO.output(pin, state)
     logging.info('Set relay on pin %s to %s', pin, 'ON' if state == RELAY_ON else 'OFF')
 
 def pulse_relay(pin):
     set_relay(pin, RELAY_ON)
-    time.sleep(0.5)
+    time.sleep(0.5)  # atomic
     set_relay(pin, RELAY_OFF)
 
 def ring_bell(mac):
-    global allow_watchdog
-    if not allow_watchdog and not DEBUG:
+    global allow_watchdog, cooldown_time
+
+    if not allow_watchdog and not DEBUG and not NO_WATCHDOG:
         logging.info('Enabling watchdog...')
         allow_watchdog = True
 
+    if time.time() - cooldown_time < 2:
+        logging.info('Cooldown skipping.')
+        return
+
+    cooldown_time = time.time()
+
     try:
         doorbell = settings.DOORBELLS[mac]
         pulse_relay(doorbell['gpio'])
@@ -58,44 +69,52 @@ def feed_watchdog():
         wdt.write('1')
 
 async def ws_listener():
-    while True:
-        session = ClientSession(cookie_jar=CookieJar(unsafe=True))
+    session = ClientSession(cookie_jar=CookieJar(unsafe=True))
+
+    unifiprotect = UpvServer(
+        session,
+        settings.UFP_ADDRESS,
+        settings.UFP_PORT,
+        settings.UFP_USERNAME,
+        settings.UFP_PASSWORD,
+    )
 
-        unifiprotect = UpvServer(
-            session,
-            settings.UFP_ADDRESS,
-            settings.UFP_PORT,
-            settings.UFP_USERNAME,
-            settings.UFP_PASSWORD,
-        )
+    await unifiprotect.update()
 
-        await unifiprotect.update()
+    unsub = unifiprotect.subscribe_websocket(subscriber)
 
-        unsub = unifiprotect.subscribe_websocket(subscriber)
+    logging.info('Connecting to websocket.')
+    await asyncio.sleep(2)
 
-        logging.info('Connecting to websocket.')
-        await asyncio.sleep(2)
+    while True:
+        try:
+            updates = await unifiprotect.update()
+            logging.debug('')
+            logging.debug('Updates: %s', json.dumps(updates, indent=4))
+        except NvrError:
+            logging.error('Error updating connection. Reconnecting...')
+            break
 
-        while True:
-            try:
-                updates = await unifiprotect.update()
-                logging.debug('Updates: %s', str(updates))
-            except NvrError:
-                logging.error('Problem connecting to Unifi Protect. Reconnecting...')
-                break
+        active_ws = await unifiprotect.check_ws()
+        if not active_ws:
+            logging.error('Websocket unactive. Reconnecting...')
+            break
 
-            active_ws = await unifiprotect.check_ws()
-            if not active_ws:
-                logging.error('Websocket unactive. Reconnecting...')
-                break
+        if allow_watchdog:
+            feed_watchdog()
 
-            if allow_watchdog and not DEBUG:
-                feed_watchdog()
+        await asyncio.sleep(1)
 
-            await asyncio.sleep(1)
+    await session.close()
+    unsub()
 
-        await session.close()
-        unsub()
+async def connect():
+    while True:
+        try:
+            await ws_listener()
+        except NvrError as e:
+            logging.error('Error connecting to Unifi Protect: %s. Trying again...', str(e))
+            await asyncio.sleep(3)
 
 def disable_relays_on_exit(*args):
     logging.info('Exiting, disabling relays...')
@@ -126,5 +145,5 @@ if __name__ == '__main__':
     init()
 
     loop = asyncio.get_event_loop()
-    loop.run_until_complete(ws_listener())
+    loop.run_until_complete(connect())
     loop.close()