More robust websocket retry handler for Todoist

platypush/backend/todoist.py

@@ -58,15 +58,7 @@ class TodoistBackend(Backend):
 
         return hndl
 
-    def _on_error(self):
-        # noinspection PyUnusedLocal
-        def hndl(ws, error):
-            self.logger.warning('Todoist websocket error: {}'.format(error))
-        return hndl
-
-    def _on_close(self):
-        # noinspection PyUnusedLocal
-        def hndl(ws):
+    def _retry_hndl(self):
         self._ws = None
         self.logger.warning('Todoist websocket connection closed')
         self._connected = False
@@ -75,6 +67,19 @@ class TodoistBackend(Backend):
             self._connect()
             time.sleep(10)
 
+    def _on_error(self):
+        # noinspection PyUnusedLocal
+        def hndl(ws, error):
+            self.logger.warning('Todoist websocket error: {}'.format(error))
+            self._retry_hndl()
+
+        return hndl
+
+    def _on_close(self):
+        def hndl(*_, **__):
+            self.logger.info('Todoist websocket closed')
+            self._retry_hndl()
+
         return hndl
 
     def _on_open(self):