mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 12:57:52 +00:00
Code Issues Releases Wiki Activity

Refactor get_history method

Browse Source
This commit is contained in:
Dan 2018-12-17 14:18:41 +01:00
parent 7b90a0e7c5
commit 603bc88aa3
1 changed files with 14 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
pyrogram/client/methods/messages/get_history.py
View File

@ -18,7 +18,7 @@
import pyrogram import pyrogram
from pyrogram.api import functions from pyrogram.api import functions
from ...ext import BaseClient, utils from ...ext import BaseClient
class GetHistory(BaseClient): class GetHistory(BaseClient):
@ -59,7 +59,9 @@ class GetHistory(BaseClient):
:class:`Error <pyrogram.Error>` in case of a Telegram RPC error. :class:`Error <pyrogram.Error>` in case of a Telegram RPC error.
""" """
r = self.send( return pyrogram.Messages.parse(
self,
self.send(
functions.messages.GetHistory( functions.messages.GetHistory(
peer=self.resolve_peer(chat_id), peer=self.resolve_peer(chat_id),
offset_id=offset_id, offset_id=offset_id,
@ -71,40 +73,4 @@ class GetHistory(BaseClient):
hash=0 hash=0
) )
) )
users = {i.id: i for i in r.users}
chats = {i.id: i for i in r.chats}
reply_to_messages = {
i.reply_to_msg_id: None
for i in r.messages
if i.reply_to_msg_id
}
if reply_to_messages:
temp = self.get_messages(
chat_id, reply_to_messages,
replies=0
)
assert len(temp) == len(reply_to_messages)
for i in range(len(temp)):
reply_to_messages[temp[i].message_id] = temp[i]
messages = utils.parse_messages(
self, r.messages,
users, chats,
replies=0
)
assert len(messages) == len(r.messages)
for i in range(len(messages)):
if r.messages[i].reply_to_msg_id:
messages[i].reply_to_message = reply_to_messages[r.messages[i].reply_to_msg_id]
return pyrogram.Messages(
total_count=getattr(r, "count", len(r.messages)),
messages=messages
) )