attzonko#28 re-establish ws connection

rwky · Dec 11, 2016 · 92c4012 · 92c4012
1 parent de38593
commit 92c4012
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/mattermost_bot/mattermost.py b/mattermost_bot/mattermost.py
@@ -5,6 +5,7 @@
 
 import requests
 import websocket
+import websocket._exceptions
 
 logger = logging.getLogger(__name__)
 
@@ -186,7 +187,12 @@ def messages(self, ignore_own_msg=False, filter_action=None):
         if not self.connect_websocket():
             return
         while True:
-            data = self.websocket.recv()
+            try:
+                data = self.websocket.recv()
+            except websocket._exceptions.WebSocketException:
+                if not self.connect_websocket():
+                    raise
+                continue
             if data:
                 try:
                     post = json.loads(data)