Add log_out method

delivrance · delivrance · commit cfd756bd2439 · 2019-09-08T11:58:34.000+02:00
diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py
@@ -264,6 +264,7 @@ def get_title_list(s: str) -> list:
             send_recovery_code
             recover_password
             accept_terms_of_service
+            log_out
         """,
         advanced="""
         Advanced
diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
@@ -222,7 +222,10 @@ def __enter__(self):
         return self.start()
 
     def __exit__(self, *args):
-        self.stop()
+        try:
+            self.stop()
+        except ConnectionError:
+            pass
 
     @property
     def proxy(self):
@@ -773,6 +776,27 @@ def authorize(self) -> User:
 
         return signed_up
 
+    def log_out(self):
+        """Log out from Telegram and delete the *\\*.session* file.
+
+        When you log out, the current client is stopped and the storage session destroyed.
+        No more API calls can be made until you start the client and re-authorize again.
+
+        Returns:
+            ``bool``: On success, True is returned.
+
+        Example:
+            .. code-block:: python
+
+                # Log out.
+                app.log_out()
+        """
+        self.send(functions.auth.LogOut())
+        self.stop()
+        self.storage.destroy()
+
+        return True
+
     def start(self):
         """Start the client.
 
@@ -1338,7 +1362,7 @@ def updates_worker(self):
                 elif isinstance(updates, types.UpdateShort):
                     self.dispatcher.updates_queue.put((updates.update, {}, {}))
                 elif isinstance(updates, types.UpdatesTooLong):
-                    log.warning(updates)
+                    log.info(updates)
             except Exception as e:
                 log.error(e, exc_info=True)
 
diff --git a/pyrogram/client/storage/file_storage.py b/pyrogram/client/storage/file_storage.py
@@ -19,6 +19,7 @@
 import base64
 import json
 import logging
+import os
 import sqlite3
 from pathlib import Path
 from threading import Lock
@@ -108,3 +109,6 @@ def open(self):
 
         with self.conn:
             self.conn.execute("VACUUM")
+
+    def destroy(self):
+        os.remove(self.database)
diff --git a/pyrogram/client/storage/memory_storage.py b/pyrogram/client/storage/memory_storage.py
@@ -97,6 +97,9 @@ def close(self):
         with self.lock:
             self.conn.close()
 
+    def destroy(self):
+        pass
+
     def update_peers(self, peers: List[Tuple[int, int, str, str, str]]):
         with self.lock:
             self.conn.executemany(
diff --git a/pyrogram/client/storage/storage.py b/pyrogram/client/storage/storage.py
@@ -30,6 +30,9 @@ def save(self):
     def close(self):
         raise NotImplementedError
 
+    def destroy(self):
+        raise NotImplementedError
+
     def update_peers(self, peers):
         raise NotImplementedError
 
