TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 20:59:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix handling proxies with domain names

Browse Source
This commit is contained in:
Dan 2018-09-22 14:44:12 +02:00
parent 28abcaac50
commit 8ea556b65f
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -18,6 +18,7 @@
import logging import logging
import socket import socket
import ipaddress
try: try:
import socks import socks
@ -39,11 +40,14 @@ class TCP(socks.socksocket):
port = proxy.get("port", None) port = proxy.get("port", None)
try: try:
socket.inet_aton(hostname) ip_address = ipaddress.ip_address(hostname)
except socket.error: except ValueError:
super().__init__(socket.AF_INET6)
else:
super().__init__(socket.AF_INET) super().__init__(socket.AF_INET)
else:
if isinstance(ip_address, ipaddress.IPv6Address):
super().__init__(socket.AF_INET6)
else:
super().__init__(socket.AF_INET)
self.set_proxy( self.set_proxy(
proxy_type=socks.SOCKS5, proxy_type=socks.SOCKS5,