Save session in complete_login

LonamiWebs · Oct 17, 2023 · 186dd38 · 186dd38
1 parent 4465644
commit 186dd38
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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