hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 21:10:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add proxy support using PySocks (closes #37)

Browse Source
This commit is contained in:
Epix Zhang
2017-03-21 00:16:34 +08:00
committed by Lonami
parent b4c81a60b3
commit 350c11d66f
5 changed files with 35 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
telethon/network/tcp_client.py
View File

@@ -9,9 +9,20 @@ from telethon.utils import BinaryWriter
class TcpClient:
def __init__(self):
def __init__(self, proxy=None):
self.connected = False
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
if proxy:
try:
import socks
self.socket = socks.socksocket(socket.AF_INET, socket.SOCK_STREAM)
self.socket.set_proxy(*proxy)
except (ImportError, SystemError):
print("Can't import PySocks, fallback to vanilla socket. "
"Proxy settings are ignored. "
"Try to install PySocks via pip")
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
else:
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# Support for multi-threading advantages and safety
self.cancelled = False # Has the read operation been cancelled?