Update main.py

Author: ptrlrd

main.py

@@ -1,4 +1,5 @@
 import threading
+import asyncio
 from bot.bot import bot
 from web.app import app
 from shared.config import DISCORD_TOKEN, ENABLE_LOGGING, BOT_STATUS, BOT_NAME
@@ -7,6 +8,21 @@
 import time
 
 
+async def start_bot():
+    retry_count = 0
+    max_retries = 5
+    
+    while retry_count < max_retries:
+        try:
+            await bot.start(DISCORD_TOKEN)
+        except aiohttp.client_exceptions.ClientConnectorError as e:
+            print(f"Connection error: {e}. Retrying in 10 seconds...")
+            retry_count += 1
+            await asyncio.sleep(10)
+        except Exception as e:
+            print(f"Unexpected error: {e}")
+            break
+
 def start_prometheus_server():
     # Start a Prometheus HTTP server on port 8000
     start_http_server(8000, addr='0.0.0.0')
@@ -46,4 +62,4 @@ async def on_disconnect():
     bot.load_extension('bot.news_feed')
     # bot.load_extension('bot.moderation') Turning off because it's not working
     start_prometheus_server()
-    bot.run(DISCORD_TOKEN)
+    asyncio.run(start_bot())