LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 09:26:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

TcpClient: Combine two identical function calls into one

Browse Source
This commit is contained in:
Dmitry D. Chernov 2017-05-10 03:05:14 +10:00 committed by Lonami
parent 15e06f9ce2
commit 7ffe2b3130
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
telethon/network/tcp_client.py
View File

@ -11,6 +11,7 @@ from telethon.utils import BinaryWriter
class TcpClient: class TcpClient:
def __init__(self, proxy=None): def __init__(self, proxy=None):
self.connected = False self.connected = False
if proxy: if proxy:
try: try:
import socks import socks
@ -20,8 +21,9 @@ class TcpClient:
print("Can't import PySocks, fallback to vanilla socket. " print("Can't import PySocks, fallback to vanilla socket. "
"Proxy settings are ignored. " "Proxy settings are ignored. "
"Try to install PySocks via pip") "Try to install PySocks via pip")
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) proxy = None
else:
if not proxy:
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# Support for multi-threading advantages and safety # Support for multi-threading advantages and safety