Add information about test/production servers when logging connections

delivrance · delivrance · commit 4de203a81c02 · 2019-08-13T13:18:51.000+02:00
diff --git a/pyrogram/connection/connection.py b/pyrogram/connection/connection.py
@@ -39,6 +39,7 @@ class Connection:
 
     def __init__(self, dc_id: int, test_mode: bool, ipv6: bool, proxy: dict, mode: int = 3):
         self.dc_id = dc_id
+        self.test_mode = test_mode
         self.ipv6 = ipv6
         self.proxy = proxy
         self.address = DataCenter(dc_id, test_mode, ipv6)
@@ -59,7 +60,8 @@ def connect(self):
                 self.connection.close()
                 time.sleep(1)
             else:
-                log.info("Connected! DC{} - IPv{} - {}".format(
+                log.info("Connected! {} DC{} - IPv{} - {}".format(
+                    "Test" if self.test_mode else "Production",
                     self.dc_id,
                     "6" if self.ipv6 else "4",
                     self.mode.__name__
