From 821f8831b9e2bcd8cfb4323b63495a3e1d27b3a4 Mon Sep 17 00:00:00 2001
From: ph4z <pierrick@poseidon-networks.com>
Date: Wed, 11 Dec 2019 19:19:39 +0100
Subject: [PATCH] Fix resub for authenticated sockets

---
 bfxapi/websockets/generic_websocket.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/bfxapi/websockets/generic_websocket.py b/bfxapi/websockets/generic_websocket.py
index 6c59a81..a01d85d 100644
--- a/bfxapi/websockets/generic_websocket.py
+++ b/bfxapi/websockets/generic_websocket.py
@@ -43,6 +43,9 @@ class Socket():
 
     def set_authenticated(self):
         self.isAuthenticated = True
+        
+    def set_unauthenticated(self):
+        self.isAuthenticated = False
 
     def set_websocket(self, ws):
         self.ws = ws
@@ -147,6 +150,8 @@ class GenericWebsocket:
                 retries = 0
             except (ConnectionClosed, socket.error) as e:
                 self.sockets[sId].set_disconnected()
+                if self.sockets[sId].isAuthenticated:
+                    self.sockets[sId].set_unauthenticated()
                 self._emit('disconnected')
                 if (not self.attempt_retry):
                     return