From 186dd38ff43712e797b4f138423a7c5f38dc740d Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Tue, 17 Oct 2023 20:45:52 +0200 Subject: [PATCH] Save session in complete_login --- client/src/telethon/_impl/client/client/auth.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/client/src/telethon/_impl/client/client/auth.py b/client/src/telethon/_impl/client/client/auth.py index a05d932c..0056dbbb 100644 --- a/client/src/telethon/_impl/client/client/auth.py +++ b/client/src/telethon/_impl/client/client/auth.py @@ -42,9 +42,17 @@ async def complete_login(client: Client, auth: abcs.auth.Authorization) -> User: try: state = await client(functions.updates.get_state()) client._message_box.set_state(state) + client._session.state = client._message_box.session_state() except Exception: pass + try: + await client._storage.save(client._session) + except Exception: + client._logger.exception( + "failed to save session upon login; you may need to login again in future runs" + ) + return user