fix(discord): reconnect when websocket connection is unexpectedly lost (

#231)
OpenTTD · Apr 9, 2024 · b2a1859 · b2a1859
1 parent 5c8d9f6
commit b2a1859
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/dibridge/discord.py b/dibridge/discord.py
@@ -207,4 +207,13 @@ def update_presence(self, status):
 
 def start(token, channel_id):
     relay.DISCORD = RelayDiscord(channel_id)
-    relay.DISCORD.run(token, log_handler=None)
+    backoff = discord.backoff.ExponentialBackoff()
+
+    while True:
+        try:
+            relay.DISCORD.run(token, log_handler=None)
+        except Exception:
+            retry = backoff.delay()
+            log.exception("Discord client stopped unexpectedly; will reconnect in %.2f seconds", retry)
+
+            asyncio.run(asyncio.sleep(retry))