Add quick retries when connecting to Telegram (#941)

* Add quick 5 retries when connecting to Telegram * Fix attempt initialisation Co-authored-by: Tulir Asokan <[email protected]> * Log only when retrying Co-authored-by: Tulir Asokan <[email protected]> --------- Co-authored-by: Tulir Asokan <[email protected]>
mautrix · Nov 13, 2023 · 1b2f07d · 1b2f07d
1 parent 2a67c96
commit 1b2f07d
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()
+        attempts = 1
+        while True:
+            try:
+                await self.client.connect()
+            except Exception:
+                attempts += 1
+                if attempts > 10:
+                    raise
+                self.log.exception("Exception connecting to Telegram, retrying in 5s...")
+                await asyncio.sleep(5)
         self.log.debug(f"{'Bot' if self.is_relaybot else self.mxid} connected: {self.connected}")
         return self