From 81d3d5956c405767d3682bff6880eced7dc3650f Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Fri, 25 May 2018 12:37:03 +0200
Subject: [PATCH] Don't try to indefinitely reconnect at Connection layer

---
 pyrogram/connection/connection.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pyrogram/connection/connection.py b/pyrogram/connection/connection.py
index 118f0d83..5e86c8aa 100644
--- a/pyrogram/connection/connection.py
+++ b/pyrogram/connection/connection.py
@@ -26,6 +26,8 @@ log = logging.getLogger(__name__)
 
 
 class Connection:
+    MAX_RETRIES = 5
+
     MODES = {
         0: TCPFull,
         1: TCPAbridged,
@@ -40,7 +42,7 @@ class Connection:
         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 @@ class Connection:
                 time.sleep(1)
             else:
                 break
+        else:
+            raise TimeoutError
 
     def close(self):
         self.connection.close()