From 92c115733dc34059e49f883d3d19a31037908ed4 Mon Sep 17 00:00:00 2001 From: Jahongir Qurbonov Date: Wed, 16 Oct 2024 18:43:45 +0500 Subject: [PATCH] Refactor handle_migrate --- client/src/telethon/_impl/client/client/auth.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/client/src/telethon/_impl/client/client/auth.py b/client/src/telethon/_impl/client/client/auth.py index 757771e7..d7b4f014 100644 --- a/client/src/telethon/_impl/client/client/auth.py +++ b/client/src/telethon/_impl/client/client/auth.py @@ -57,15 +57,11 @@ async def complete_login(client: Client, auth: abcs.auth.Authorization) -> User: async def handle_migrate(client: Client, dc_id: Optional[int]) -> None: assert client._sender assert dc_id is not None - sender, client._session.dcs = await connect_sender( + await client._sender.disconnect() + client._sender, client._session.dcs = await connect_sender( client._config, client._session.dcs, DataCenter(id=dc_id) ) - async with client._sender._step_lock: - old_sender = client._sender - client._sender = sender - await old_sender.disconnect() - async def bot_sign_in(self: Client, token: str) -> User: request = functions.auth.import_bot_authorization(