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

Clean get_history

Browse Source
This commit is contained in:
Dan 2018-05-09 20:27:29 +02:00
parent 781edda56a
commit 14f23d388c
2 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pyrogram/client/ext/base_client.py
View File

@ -93,3 +93,11 @@ class BaseClient:
progress_args: tuple = () progress_args: tuple = ()
): ):
pass pass
def get_messages(
self,
chat_id: int or str,
message_ids,
replies: int = 1
):
pass

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

@ -16,7 +16,7 @@
# You should have received a copy of the GNU Lesser General Public License # You should have received a copy of the GNU Lesser General Public License
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>. # along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
from pyrogram.api import functions, types from pyrogram.api import functions
from ...ext import BaseClient, utils from ...ext import BaseClient, utils
@ -69,17 +69,28 @@ class GetHistory(BaseClient):
users = {i.id: i for i in r.users} users = {i.id: i for i in r.users}
chats = {i.id: i for i in r.chats} 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} reply_to_messages = {
i.reply_to_msg_id: None
for i in r.messages
if i.reply_to_msg_id
}
if reply_to_messages: if reply_to_messages:
temp = self.get_messages(chat_id, reply_to_messages.keys(), replies=0) temp = self.get_messages(
chat_id, reply_to_messages,
replies=0
)
assert len(temp) == len(reply_to_messages) assert len(temp) == len(reply_to_messages)
for i in range(len(temp)): for i in range(len(temp)):
reply_to_messages[temp[i].message_id] = temp[i] reply_to_messages[temp[i].message_id] = temp[i]
messages = utils.parse_messages(self, r.messages, users, chats, replies=0) messages = utils.parse_messages(
self, r.messages,
users, chats,
replies=0
)
assert len(messages) == len(r.messages) assert len(messages) == len(r.messages)