diff --git a/pyrogram/connection/connection.py b/pyrogram/connection/connection.py index 4b9cf6a8..118f0d83 100644 --- a/pyrogram/connection/connection.py +++ b/pyrogram/connection/connection.py @@ -27,15 +27,15 @@ log = logging.getLogger(__name__) class Connection: MODES = { - 0: Full, - 1: Abridged, - 2: Intermediate + 0: TCPFull, + 1: TCPAbridged, + 2: TCPIntermediate } def __init__(self, address: tuple, proxy: dict, mode: int = 1): self.address = address self.proxy = proxy - self.mode = self.MODES.get(mode, Abridged) + self.mode = self.MODES.get(mode, TCPAbridged) self.lock = threading.Lock() self.connection = None diff --git a/pyrogram/connection/transport/tcp/__init__.py b/pyrogram/connection/transport/tcp/__init__.py index 0298e04b..d1daff63 100644 --- a/pyrogram/connection/transport/tcp/__init__.py +++ b/pyrogram/connection/transport/tcp/__init__.py @@ -16,6 +16,6 @@ # You should have received a copy of the GNU Lesser General Public License # along with Pyrogram. If not, see . -from .abridged import Abridged -from .full import Full -from .intermediate import Intermediate +from .tcp_abridged import TCPAbridged +from .tcp_full import TCPFull +from .tcp_intermediate import TCPIntermediate diff --git a/pyrogram/connection/transport/tcp/abridged.py b/pyrogram/connection/transport/tcp/tcp_abridged.py similarity index 98% rename from pyrogram/connection/transport/tcp/abridged.py rename to pyrogram/connection/transport/tcp/tcp_abridged.py index 076badad..472f4799 100644 --- a/pyrogram/connection/transport/tcp/abridged.py +++ b/pyrogram/connection/transport/tcp/tcp_abridged.py @@ -23,7 +23,7 @@ from .tcp import TCP log = logging.getLogger(__name__) -class Abridged(TCP): +class TCPAbridged(TCP): def __init__(self, proxy: dict): super().__init__(proxy) diff --git a/pyrogram/connection/transport/tcp/full.py b/pyrogram/connection/transport/tcp/tcp_full.py similarity index 99% rename from pyrogram/connection/transport/tcp/full.py rename to pyrogram/connection/transport/tcp/tcp_full.py index 5707cdfb..1b131678 100644 --- a/pyrogram/connection/transport/tcp/full.py +++ b/pyrogram/connection/transport/tcp/tcp_full.py @@ -25,7 +25,7 @@ from .tcp import TCP log = logging.getLogger(__name__) -class Full(TCP): +class TCPFull(TCP): def __init__(self, proxy: dict): super().__init__(proxy) self.seq_no = None diff --git a/pyrogram/connection/transport/tcp/intermediate.py b/pyrogram/connection/transport/tcp/tcp_intermediate.py similarity index 96% rename from pyrogram/connection/transport/tcp/intermediate.py rename to pyrogram/connection/transport/tcp/tcp_intermediate.py index 46546d1d..4b2e2596 100644 --- a/pyrogram/connection/transport/tcp/intermediate.py +++ b/pyrogram/connection/transport/tcp/tcp_intermediate.py @@ -19,12 +19,12 @@ import logging from struct import pack, unpack -from .abridged import TCP +from .tcp import TCP log = logging.getLogger(__name__) -class Intermediate(TCP): +class TCPIntermediate(TCP): def __init__(self, proxy: dict): super().__init__(proxy)