Add quick 5 retries when connecting to Telegram

mautrix · Nov 13, 2023 · d47fb1d · d47fb1d
1 parent 2a67c96
commit d47fb1d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/mautrix_telegram/abstract_user.py b/mautrix_telegram/abstract_user.py
@@ -306,7 +306,16 @@ async def has_full_access(self, allow_bot: bool = False) -> bool:
     async def start(self, delete_unless_authenticated: bool = False) -> AbstractUser:
         if not self.client:
             await self._init_client()
-        await self.client.connect()
+        while True:
+            attempts = 1
+            try:
+                await self.client.connect()
+            except Exception:
+                self.log.exception("Exception connecting to Telegram, retrying in 5s...")
+                attempts += 1
+                if attempts > 10:
+                    raise
+                await asyncio.sleep(5)
         self.log.debug(f"{'Bot' if self.is_relaybot else self.mxid} connected: {self.connected}")
         return self