Skip to content

Commit ac14845

Browse files
committed
fix websocket close state
1 parent 6390b9b commit ac14845

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

py_boilingdata/__init__.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ def _on_error(self, ws_app, error):
240240

241241
def _on_close(self, ws_app, code, msg):
242242
self.logger.info(f"WS CLOSE: {code} {msg}")
243-
self.is_open = False
243+
self.bd_is_open = False
244244

245245
def _all_messages_received(self, event):
246246
requestId = event["requestId"]
@@ -274,8 +274,9 @@ async def bd_execute(self, sql, cb):
274274

275275
async def connect(self):
276276
"""Connect to BoilingData WebSocket API"""
277-
if self.websocket is not None:
278-
raise Exception("WebSocket already exists")
277+
if self.bd_is_open is True:
278+
self.logger.debug("Already open, not re-connecting.")
279+
return
279280
self.logger.info("Connecting")
280281
self.websocket = websocket.WebSocket()
281282
websocket.enableTrace(self.ws_trace)

0 commit comments

Comments
 (0)