Allow to hook load and save functions to use external session storage…

… implementations

pyrogram/client/__init__.py

@@ -17,6 +17,6 @@
 # along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
 from .chat_action import ChatAction
-from .client import Client
+from .client import Client, Proxy
 from .parse_mode import ParseMode
 from .emoji import Emoji

pyrogram/client/client.py

@@ -175,6 +175,9 @@ def __init__(self,
 
         self.download_queue = Queue()
 
+        self.load_session_hook = None
+        self.save_session_hook = None
+
     def start(self):
         """Use this method to start the Client after creating it.
         Requires no parameters.
@@ -183,7 +186,10 @@ def start(self):
             :class:`pyrogram.Error`
         """
         self.load_config()
-        self.load_session(self.session_name)
+        if callable(self.load_session_hook):
+            self.load_session_hook(self)
+        else:
+            self.load_session(self.session_name)
 
         self.session = Session(
             self.dc_id,
@@ -202,7 +208,10 @@ def start(self):
             else:
                 self.authorize_bot()
 
-            self.save_session()
+            if callable(self.save_session_hook):
+                self.save_session_hook(self)
+            else:
+                self.save_session()
 
         if self.token is None:
             self.get_dialogs()