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

Tidy if/else branch on _recreate_socket

Browse Source
This commit is contained in:
Lonami Exo 2017-05-30 11:42:14 +02:00
parent 2399b7e55e
commit d321096714
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
telethon/network/tcp_client.py
View File

@ -20,15 +20,13 @@ class TcpClient:
self._lock = Lock() self._lock = Lock()
def _recreate_socket(self): def _recreate_socket(self):
self._socket = None if self._proxy is None:
if self._proxy is not None: self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
else:
import socks import socks
self._socket = socks.socksocket(socket.AF_INET, socket.SOCK_STREAM) self._socket = socks.socksocket(socket.AF_INET, socket.SOCK_STREAM)
self._socket.set_proxy(*self._proxy) self._socket.set_proxy(*self._proxy)
if not self._socket:
self._socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
def connect(self, ip, port): def connect(self, ip, port):
"""Connects to the specified IP and port number""" """Connects to the specified IP and port number"""
if not self.connected: if not self.connected: