From fb8dbf6fc9a8a2bce613c0a44fdddb0e1a98c034 Mon Sep 17 00:00:00 2001 From: Jahongir Qurbonov Date: Mon, 21 Oct 2024 18:15:22 +0500 Subject: [PATCH] Rewrite protocol --- client/doc/modules/types.rst | 4 ---- client/src/telethon/_impl/client/client/net.py | 5 ++--- client/src/telethon/_impl/mtsender/__init__.py | 4 ---- 3 files changed, 2 insertions(+), 11 deletions(-) diff --git a/client/doc/modules/types.rst b/client/doc/modules/types.rst index b665ce36..413fdeb8 100644 --- a/client/doc/modules/types.rst +++ b/client/doc/modules/types.rst @@ -140,8 +140,4 @@ Private definitions .. currentmodule:: telethon._impl.mtsender.sender -.. autoclass:: AsyncReader - -.. autoclass:: AsyncWriter - .. autoclass:: Connector diff --git a/client/src/telethon/_impl/client/client/net.py b/client/src/telethon/_impl/client/client/net.py index 808c4b6b..5f4b4a7a 100644 --- a/client/src/telethon/_impl/client/client/net.py +++ b/client/src/telethon/_impl/client/client/net.py @@ -246,10 +246,9 @@ async def invoke_request( sleep_thresh = client._config.flood_sleep_threshold rx = client._sender.enqueue(request) while True: - while not rx.done(): - await step_sender(client) + await step_sender(client) try: - response = rx.result() + response = await rx break except RpcError as e: if e.code == 420 and e.value is not None and e.value < sleep_thresh: diff --git a/client/src/telethon/_impl/mtsender/__init__.py b/client/src/telethon/_impl/mtsender/__init__.py index 4ba3c76d..e0b3e466 100644 --- a/client/src/telethon/_impl/mtsender/__init__.py +++ b/client/src/telethon/_impl/mtsender/__init__.py @@ -1,8 +1,6 @@ from .sender import ( NO_PING_DISCONNECT, PING_DELAY, - AsyncReader, - AsyncWriter, Connector, Sender, connect, @@ -11,8 +9,6 @@ from .sender import ( __all__ = [ "NO_PING_DISCONNECT", "PING_DELAY", - "AsyncReader", - "AsyncWriter", "Connector", "Sender", "connect",