yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-07 15:34:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update tcp.py

Browse source
This commit is contained in:
rick 2025-08-11 13:10:45 +07:00 committed by GitHub
parent cb38d6a02b
commit 8ba3e4bb96
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pyrogram/connection/transport/tcp/tcp.py
View file

@ -91,10 +91,8 @@ class TCP:
) )
sock.settimeout(TCP.TIMEOUT) sock.settimeout(TCP.TIMEOUT)
await self.loop.sock_connect( # fix: Use run_in_executor for socks.socksocket
sock=sock, await self.loop.run_in_executor(None, sock.connect, destination)
address=destination
)
sock.setblocking(False) sock.setblocking(False)
@ -123,7 +121,7 @@ class TCP:
async def connect(self, address: Tuple[str, int]) -> None: async def connect(self, address: Tuple[str, int]) -> None:
try: try:
await asyncio.wait_for(self._connect(address), TCP.TIMEOUT) await asyncio.wait_for(self._connect(address), TCP.TIMEOUT)
except asyncio.TimeoutError: # Re-raise as TimeoutError. asyncio.TimeoutError is deprecated in 3.11 except asyncio.TimeoutError: # re raise as TimeoutError. asyncio.TimeoutError is deprecated in 3.11
raise TimeoutError("Connection timed out") raise TimeoutError("Connection timed out")
async def close(self) -> None: async def close(self) -> None: