mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-22 09:57:19 +00:00
Code Issues Releases Wiki Activity

Workaround proxy sockets not timing out properly

Browse Source
This commit is contained in:
Dan 2022-12-27 22:23:05 +01:00
parent 52effe19d5
commit cf06939a55
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyrogram/connection/transport/tcp/tcp.py
View File

@ -20,6 +20,7 @@ import asyncio
import ipaddress import ipaddress
import logging import logging
import socket import socket
import socks import socks
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -58,6 +59,8 @@ class TCP:
password=proxy.get("password", None) password=proxy.get("password", None)
) )
self.socket.settimeout(TCP.TIMEOUT)
log.info("Using proxy %s", hostname) log.info("Using proxy %s", hostname)
else: else:
self.socket = socket.socket( self.socket = socket.socket(
@ -65,7 +68,7 @@ class TCP:
else socket.AF_INET else socket.AF_INET
) )
self.socket.setblocking(False) self.socket.setblocking(False)
async def connect(self, address: tuple): async def connect(self, address: tuple):
try: try: