From fb4503338d355f1e5c5ab02a98bd6b51b9559fd5 Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Fri, 1 Sep 2023 11:46:49 +0200 Subject: [PATCH] Add disconnect method to sender Unlike Rust, there's no implicit dropping of connections. --- client/src/telethon/_impl/mtsender/sender.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/client/src/telethon/_impl/mtsender/sender.py b/client/src/telethon/_impl/mtsender/sender.py index 8567ae70..22055479 100644 --- a/client/src/telethon/_impl/mtsender/sender.py +++ b/client/src/telethon/_impl/mtsender/sender.py @@ -120,6 +120,10 @@ class Sender: Enqueuer(request_queue), ) + async def disconnect(self): + self._writer.close() + await self._writer.wait_closed() + async def invoke(self, request: RemoteCall[Return]) -> bytes: rx = self._enqueue_body(bytes(request)) return await self._step_until_receive(rx)