Handle proxy socket connections using thread executors

pyrogram/connection/transport/tcp/tcp.py

@@ -20,6 +20,7 @@ import asyncio
 import ipaddress
 import logging
 import socket
+from concurrent.futures import ThreadPoolExecutor
 
 import socks
 
@@ -38,6 +39,8 @@ class TCP:
         self.lock = asyncio.Lock()
         self.loop = asyncio.get_event_loop()
 
+        self.proxy = proxy
+
         if proxy:
             hostname = proxy.get("hostname")
 
@@ -71,6 +74,10 @@ class TCP:
             self.socket.setblocking(False)
 
     async def connect(self, address: tuple):
+        if self.proxy:
+            with ThreadPoolExecutor(1) as executor:
+                await self.loop.run_in_executor(executor, self.socket.connect, address)
+        else:
             try:
                 await asyncio.wait_for(asyncio.get_event_loop().sock_connect(self.socket, address), TCP.TIMEOUT)
             except asyncio.TimeoutError:  # Re-raise as TimeoutError. asyncio.TimeoutError is deprecated in 3.11