Don't try to indefinitely reconnect at Connection layer

delivrance · delivrance · commit 81d3d5956c40 · 2018-05-25T12:37:03.000+02:00
diff --git a/pyrogram/connection/connection.py b/pyrogram/connection/connection.py
@@ -26,6 +26,8 @@
 
 
 class Connection:
+    MAX_RETRIES = 5
+
     MODES = {
         0: TCPFull,
         1: TCPAbridged,
@@ -40,7 +42,7 @@ def __init__(self, address: tuple, proxy: dict, mode: int = 1):
         self.connection = None
 
     def connect(self):
-        while True:
+        for i in range(Connection.MAX_RETRIES):
             self.connection = self.mode(self.proxy)
 
             try:
@@ -51,6 +53,8 @@ def connect(self):
                 time.sleep(1)
             else:
                 break
+        else:
+            raise TimeoutError
 
     def close(self):
         self.connection.close()
