TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't use raw functions when getting messages

Browse Source
This commit is contained in:
Dan 2018-04-17 00:26:30 +02:00
parent ebe806e2b7
commit d78dfb4f9e
1 changed files with 5 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
pyrogram/client/message_parser.py
View File

@ -19,7 +19,7 @@
from struct import pack
import pyrogram
from pyrogram.api import types, functions
from pyrogram.api import types
from .utils import encode
# TODO: Organize the code better?
@ -173,40 +173,6 @@ def parse_thumb(thumb: types.PhotoSize or types.PhotoCachedSize) -> pyrogram.Pho
)
def get_message_reply(client, chat_id: int or str, message_id: int):
peer = client.resolve_peer(chat_id)
message_id = [types.InputMessageReplyTo(message_id)]
if isinstance(peer, types.InputPeerChannel):
rpc = functions.channels.GetMessages(
channel=peer,
id=message_id
)
else:
rpc = functions.messages.GetMessages(
id=message_id
)
return client.send(rpc)
def get_message_pinned(client, chat_id: int or str):
peer = client.resolve_peer(chat_id)
message_id = [types.InputMessagePinned()]
if isinstance(peer, types.InputPeerChannel):
rpc = functions.channels.GetMessages(
channel=peer,
id=message_id
)
else:
rpc = functions.messages.GetMessages(
id=message_id
)
return client.send(rpc)
# TODO: Reorganize code, maybe split parts as well
def parse_message(
client,
@ -499,16 +465,8 @@ def parse_message(
)
if message.reply_to_msg_id and replies:
reply_to_message = get_message_reply(client, m.chat.id, message.id)
message = reply_to_message.messages[0]
users = {i.id: i for i in reply_to_message.users}
chats = {i.id: i for i in reply_to_message.chats}
if isinstance(message, types.Message):
m.reply_to_message = parse_message(client, message, users, chats, replies - 1)
elif isinstance(message, types.MessageService):
m.reply_to_message = parse_message_service(client, message, users, chats)
m.reply_to_message = client.get_messages(m.chat.id, [message.reply_to_msg_id])
m.reply_to_message = m.reply_to_message[0] if m.reply_to_message else None
return m
@ -607,16 +565,7 @@ def parse_message_service(
)
if isinstance(action, types.MessageActionPinMessage):
pin_message = get_message_pinned(client, m.chat.id)
message = pin_message.messages[0]
users = {i.id: i for i in pin_message.users}
chats = {i.id: i for i in pin_message.chats}
if isinstance(message, types.Message):
m.pinned_message = parse_message(client, message, users, chats)
elif isinstance(message, types.MessageService):
# TODO: We can't pin a service message, can we?
m.pinned_message = parse_message_service(client, message, users, chats)
m.pinned_message = client.get_messages(m.chat.id, [message.reply_to_msg_id])
m.pinned_message = m.pinned_message[0] if m.pinned_message else None
return m