another race condition based on initial delay to connect

Author: donmahallem

pyghthouse/connection/wsconnector.py

@@ -1,9 +1,15 @@
+from enum import Enum
 from threading import Thread, Lock
 from websocket import WebSocketApp, setdefaulttimeout, ABNF
 from msgpack import packb, unpackb
 from ssl import CERT_NONE
 
 
+class ConnectionState(Enum):
+    NONE=0
+    CONNECTING=1
+    CONNECTED=2
+    FAILED=3
 class WSConnector:
 
     class REID:
@@ -26,7 +32,7 @@ def __init__(self, username: str, token: str, address: str, on_msg=None, ignore_
         self.ws = None
         self.lock = Lock()
         self.reid = self.REID()
-        self.running = False
+        self.__connection_state = ConnectionState.NONE
         self.ignore_ssl_cert = ignore_ssl_cert
         setdefaulttimeout(60)
 
@@ -40,26 +46,33 @@ def start(self):
                                on_message=None if self.on_msg is None else self._handle_msg,
                                on_open=self._ready, on_error=self._fail)
         self.lock.acquire() # wait for connection to be established
+        self.__connection_state = ConnectionState.CONNECTING
         kwargs = {"sslopt": {"cert_reqs": CERT_NONE}} if self.ignore_ssl_cert else None
         Thread(target=self.ws.run_forever, kwargs=kwargs).start()
+        
 
     def _fail(self, ws, err):
+        self.__connection_state = ConnectionState.FAILED
         self.lock.release()
         raise err
 
     def stop(self):
         if self.ws is not None:
             with self.lock:
                 print("Closing the connection.")
-                self.running = False
+                self.__connection_state = ConnectionState.NONE
                 self.ws.close()
                 self.ws = None
 
     def _ready(self, ws):
         print(f"Connected to {self.address}.")
-        self.running = True
+        self.__connection_state = ConnectionState.CONNECTED
         self.lock.release()
 
+    @property
+    def connection_state(self):
+        return self.__connection_state
+
     def _handle_msg(self, ws, msg):
         if isinstance(msg, bytes):
             msg = unpackb(msg)

pyghthouse/ph.py

@@ -6,7 +6,7 @@
 import numpy as np
 
 from pyghthouse.data.canvas import PyghthouseCanvas
-from pyghthouse.connection.wsconnector import WSConnector
+from pyghthouse.connection.wsconnector import WSConnector,ConnectionState
 
 
 class VerbosityLevel(Enum):
@@ -208,10 +208,12 @@ def connect(self):
         self.connector.start()
 
     def start(self):
-        if not self.connector.running:
+        if not self.connector.connection_state==ConnectionState.CONNECTED:
             self.connect()
         self.stop()
         self.msg_handler.reset()
+        while not self.connector.connection_state==ConnectionState.CONNECTED:
+            sleep(100)
         self.ph_thread = self.PHThread(self)
         self.ph_thread.start()