Update Storage abstract class

delivrance · delivrance · commit 108fa0887826 · 2019-09-14T19:39:21.000+02:00
diff --git a/pyrogram/client/storage/storage.py b/pyrogram/client/storage/storage.py
@@ -16,8 +16,15 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+import base64
+import struct
+from typing import List, Tuple
+
 
 class Storage:
+    SESSION_STRING_FORMAT = ">B?256sI?"
+    SESSION_STRING_SIZE = 351
+
     def __init__(self, name: str):
         self.name = name
 
@@ -30,72 +37,47 @@ def save(self):
     def close(self):
         raise NotImplementedError
 
-    def destroy(self):
-        raise NotImplementedError
-
-    def update_peers(self, peers):
-        raise NotImplementedError
-
-    def get_peer_by_id(self, peer_id):
+    def delete(self):
         raise NotImplementedError
 
-    def get_peer_by_username(self, username):
+    def update_peers(self, peers: List[Tuple[int, int, str, str, str]]):
         raise NotImplementedError
 
-    def get_peer_by_phone_number(self, phone_number):
+    def get_peer_by_id(self, peer_id: int):
         raise NotImplementedError
 
-    def export_session_string(self):
+    def get_peer_by_username(self, username: str):
         raise NotImplementedError
 
-    @property
-    def peers_count(self):
+    def get_peer_by_phone_number(self, phone_number: str):
         raise NotImplementedError
 
-    @property
-    def dc_id(self):
+    def dc_id(self, value: int = object):
         raise NotImplementedError
 
-    @dc_id.setter
-    def dc_id(self, value):
+    def test_mode(self, value: bool = object):
         raise NotImplementedError
 
-    @property
-    def test_mode(self):
+    def auth_key(self, value: bytes = object):
         raise NotImplementedError
 
-    @test_mode.setter
-    def test_mode(self, value):
+    def date(self, value: int = object):
         raise NotImplementedError
 
-    @property
-    def auth_key(self):
+    def user_id(self, value: int = object):
         raise NotImplementedError
 
-    @auth_key.setter
-    def auth_key(self, value):
+    def is_bot(self, value: bool = object):
         raise NotImplementedError
 
-    @property
-    def date(self):
-        raise NotImplementedError
-
-    @date.setter
-    def date(self, value):
-        raise NotImplementedError
-
-    @property
-    def user_id(self):
-        raise NotImplementedError
-
-    @user_id.setter
-    def user_id(self, value):
-        raise NotImplementedError
-
-    @property
-    def is_bot(self):
-        raise NotImplementedError
-
-    @is_bot.setter
-    def is_bot(self, value):
-        raise NotImplementedError
+    def export_session_string(self):
+        return base64.urlsafe_b64encode(
+            struct.pack(
+                self.SESSION_STRING_FORMAT,
+                self.dc_id(),
+                self.test_mode(),
+                self.auth_key(),
+                self.user_id(),
+                self.is_bot()
+            )
+        ).decode().rstrip("=")
