From efe26bcb196f4fe6a9a6c4db385d48eba47c6593 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Wed, 13 Jun 2018 13:35:41 +0200
Subject: [PATCH] Allow Connection to connect to ipv6 addresses

---
 pyrogram/connection/connection.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/pyrogram/connection/connection.py b/pyrogram/connection/connection.py
index a53295ce..043f7cb7 100644
--- a/pyrogram/connection/connection.py
+++ b/pyrogram/connection/connection.py
@@ -21,6 +21,7 @@ import threading
 import time
 
 from .transport import *
+from ..session.internals import DataCenter
 
 log = logging.getLogger(__name__)
 
@@ -36,16 +37,18 @@ class Connection:
         4: TCPIntermediateO
     }
 
-    def __init__(self, address: tuple, proxy: dict, mode: int = 1):
-        self.address = address
+    def __init__(self, dc_id: int, test_mode: bool, ipv6: bool, proxy: dict, mode: int = 1):
+        self.ipv6 = ipv6
         self.proxy = proxy
+        self.address = DataCenter(dc_id, test_mode, ipv6)
         self.mode = self.MODES.get(mode, TCPAbridged)
+
         self.lock = threading.Lock()
         self.connection = None
 
     def connect(self):
         for i in range(Connection.MAX_RETRIES):
-            self.connection = self.mode(self.proxy)
+            self.connection = self.mode(self.ipv6, self.proxy)
 
             try:
                 log.info("Connecting...")