Back to threading

src/pytevolve.py

@@ -43,7 +43,7 @@ class Tevolve:
 
 
 
-    def token_manager(self, event, mqtt_manager):
+    def token_manager(self, event):
 
         while 1:
 
@@ -59,10 +59,7 @@ class Tevolve:
                     self.token_primary = x.json()["access_token"]
                     self.time_token_start = time.time()
                     print("GOT TOKEN")
-                    try:
-                        self.r.join()
-                    except Exception as e:
-                        self.r = threading.Thread(target=self.create_websocket).start()
+
 
                     event.set()
                 else:
@@ -271,7 +268,7 @@ class Tevolve:
             pass
 
 
-    def create_websocket(self):
+    def create_websocket(self, mqtt_manager):
 
         global start_time
         global end_time
@@ -289,7 +286,7 @@ class Tevolve:
                 a = json.loads(message[20:])
                 if a[0] == "update":
                     self.last_update = a[1]
-                    self.update_homeassistant_entity(self.last_update)
+                    mqtt_manager.update_homeassistant_entity(self.last_update)
 
 
         def on_open(ws):
@@ -321,6 +318,9 @@ class Tevolve:
 
         def on_close(ws, close_status_code, close_msg):
             print("WEB SOCKET ClOSED")
+            self.r.join()
+            self.r = threading.Thread(target=run, args=(mqtt_manager,))
+            self.r.start()
             # global end_time
             # end_time = datetime.datetime.now()
             # print("starttime: " + str(start_time))
@@ -379,5 +379,6 @@ class Tevolve:
         ws.run_forever(ping_interval=10, ping_payload='42/api/v2/socket_io,["message","ping"]')
         # rel.signal(2, rel.abort)
         # rel.dispatch()
-        self.r = threading.Thread(target=run)
+        self.r = threading.Thread(target=run, args=(mqtt_manager,))
+        self.r.start()