From 38d8a54cc1a2a8cb1a58146b1a3b0bdaa2ceecbb Mon Sep 17 00:00:00 2001 From: Serhii Dylda Date: Mon, 9 Nov 2020 20:05:09 +0100 Subject: [PATCH] Fix protocol typing for python-socks. --- telethon/network/connection/connection.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/telethon/network/connection/connection.py b/telethon/network/connection/connection.py index 4a8cfc69..1703e608 100644 --- a/telethon/network/connection/connection.py +++ b/telethon/network/connection/connection.py @@ -74,11 +74,11 @@ class Connection(abc.ABC): # to be backwards compatible with PySocks proxy format, # (since socks.SOCKS5 == 2, socks.SOCKS4 == 1, socks.HTTP == 3) - if proxy_type == 2 or proxy_type == "socks5": + if isinstance(proxy_type, ProxyType.SOCKS5) or proxy_type == 2 or proxy_type == "socks5": protocol = ProxyType.SOCKS5 - elif proxy_type == 1 or proxy_type == "socks4": + elif isinstance(proxy_type, ProxyType.SOCKS4) or proxy_type == 1 or proxy_type == "socks4": protocol = ProxyType.SOCKS4 - elif proxy_type == 3 or proxy_type == "http": + elif isinstance(proxy_type, ProxyType.HTTP) or proxy_type == 3 or proxy_type == "http": protocol = ProxyType.HTTP else: raise ValueError("Unknown proxy protocol type: {}".format(proxy_type))