Add outer loop to ws listener

main.py

@@ -17,6 +17,7 @@ from aiohttp import ClientSession, CookieJar
 import settings
 
 from pyunifiprotect.unifi_protect_server import UpvServer
+from pyunifiprotect.exceptions import NvrError
 
 RELAY_ON = False
 RELAY_OFF = True
@@ -44,10 +45,10 @@ def ring_bell(mac):
     except KeyError:
         logging.error('Doorbell %s not found!', mac)
 
-def subscriber(updated):
-    logging.debug('Subscription: updated=%s', updated)
+def subscriber(updates):
+    logging.debug('Subscription: updates=%s', updates)
 
-    for _, data in updated.items():
+    for _, data in updates.items():
         if data['event_type'] == 'ring' and data['event_ring_on']:
             logging.info('%s: %s is ringing!', data['mac'], data['name'])
             ring_bell(data['mac'])
@@ -57,6 +58,7 @@ def feed_watchdog():
         wdt.write('1')
 
 async def ws_listener():
+    while True:
         session = ClientSession(cookie_jar=CookieJar(unsafe=True))
 
         unifiprotect = UpvServer(
@@ -73,8 +75,25 @@ async def ws_listener():
         unsub = unifiprotect.subscribe_websocket(subscriber)
 
         while True:
+            try:
+                updates = await unifiprotect.update()
+                logging.debug('Updates: %s', str(updates))
+            except NvrError:
+                logging.error('Problem connecting to Unifi Protect. Reconnecting...')
+                break
+
+            logging.debug('unifiprotect: %s', unifiprotect.__dict__)
+            logging.debug('ws_session %s', unifiprotect.ws_session.__dict__)
+
+            breakpoint()
+
+            #active_ws = await unifiprotect.check_ws()
+            #if not active_ws:
+            #    logging.error('Websocket unactive. Reconnecting...')
+
             if allow_watchdog and not DEBUG:
                 feed_watchdog()
+
             await asyncio.sleep(1)
 
         await session.close()
@@ -93,7 +112,7 @@ def init():
 
     for _, doorbell in settings.DOORBELLS.items():
         GPIO.setup(doorbell['gpio'], GPIO.OUT)
-        pulse_relay(doorbell['gpio'])
+        #pulse_relay(doorbell['gpio'])
         time.sleep(1)
     logging.info('GPIO initialized')