From e38da08e2568b8ab97c4f7098f7ba511373a22b7 Mon Sep 17 00:00:00 2001 From: Dan <14043624+delivrance@users.noreply.github.com> Date: Fri, 13 Apr 2018 15:20:37 +0200 Subject: [PATCH] Load peers from session file --- pyrogram/client/client.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py index a0cec4bc..fce73875 100644 --- a/pyrogram/client/client.py +++ b/pyrogram/client/client.py @@ -48,6 +48,7 @@ from pyrogram.api.errors import ( from pyrogram.crypto import AES from pyrogram.session import Auth, Session from pyrogram.session.internals import MsgId +from . import utils from .input_media import InputMedia from .style import Markdown, HTML @@ -849,6 +850,15 @@ class Client: self.user_id = s["user_id"] self.date = s.get("date", int(time.time())) + for k, v in s.get("peers_by_id", {}).items(): + self.peers_by_id[int(k)] = utils.get_input_peer(int(k), v) + + for k, v in s.get("peers_by_username", {}).items(): + self.peers_by_username[k] = self.peers_by_id[v] + + for k, v in s.get("peers_by_phone", {}).items(): + self.peers_by_phone[k] = self.peers_by_id[v] + def save_session(self): auth_key = base64.b64encode(self.auth_key).decode() auth_key = [auth_key[i: i + 43] for i in range(0, len(auth_key), 43)]