Merge branch 'master' into 40-parsingError

Augugrumi · Feb 21, 2018 · 9c5719f · 9c5719f
2 parents f7db844 + 6da42e1
commit 9c5719f
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/torrearchimedebot/bot/slack_controller.py b/torrearchimedebot/bot/slack_controller.py
@@ -112,11 +112,19 @@ def __init__(self):
             SlackController.starterbot_id = \
                 SlackController.slack_client.api_call("auth.test")["user_id"]
             while True:
-                command, channel = SlackController.parse_bot_commands(
-                    SlackController.slack_client.rtm_read())
-                if command:
-                    SlackController.handle_command(command, channel)
-                t.sleep(SlackController.RTM_READ_DELAY)
+                try:
+                    command, channel = SlackController.parse_bot_commands(
+                        SlackController.slack_client.rtm_read())
+                    if command:
+                        SlackController.handle_command(command, channel)
+                    t.sleep(SlackController.RTM_READ_DELAY)
+                except WebSocketConnectionClosedException:
+                    logger.getLogger().error("Lost connection to Slack, reconnecting...")
+                    if not sc.rtm_connect():
+                        logger.getLogger().info("Failed to reconnect to Slack")
+                        time.sleep(SlackController.RTM_READ_DELAY)
+                    else:
+                        logger.getLogger().info("Reconnected to Slack")
         else:
             logging.getLogger().info("SLACK - Connection failed. Exception " + \
             "traceback printed above.")