yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2025-12-29 12:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pyrofork: Add support for parsing Reaction as channel

Browse source 
Signed-off-by: wulan17 <wulan17@komodos.id>
This commit is contained in:
wulan17 2025-03-27 02:02:08 +07:00
parent 24d3ea0e48
commit f5296145cd
No known key found for this signature in database
GPG key ID: 737814D4B5FF0420
5 changed files with 31 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pyrogram/methods/messages/send_reaction.py
View file

@ -109,9 +109,11 @@ class SendReaction:
add_to_recent=add_to_recent
)
)
users = {i.id: i for i in r.users}
chats = {i.id: i for i in r.chats}
for i in r.updates:
if isinstance(i, raw.types.UpdateMessageReactions):
return types.MessageReactions._parse(self, i.reactions)
return types.MessageReactions._parse(self, i.reactions, users, chats)
elif story_id is not None:
await self.invoke(
raw.functions.stories.SendReaction(

3
pyrogram/methods/payments/send_paid_reaction.py
View file

@ -68,7 +68,8 @@ class SendPaidReaction:
)
)
users = {i.id: i for i in r.users}
chats = {i.id: i for i in r.chats}
for i in r.updates:
if isinstance(i, raw.types.UpdateMessageReactions):
return types.MessageReactions._parse(self, i.reactions, users)
return types.MessageReactions._parse(self, i.reactions, users, chats)

2
pyrogram/types/messages_and_media/message.py
View file

@ -1143,7 +1143,7 @@ class Message(Object, Update):
from_user = types.User._parse(client, users.get(user_id, None))
sender_chat = types.Chat._parse(client, message, users, chats, is_chat=False) if not from_user else None
reactions = types.MessageReactions._parse(client, message.reactions, users)
reactions = types.MessageReactions._parse(client, message.reactions, users, chats)
if message.via_business_bot_id:
sender_business_bot = types.User._parse(client, users.get(message.via_business_bot_id, None))

5
pyrogram/types/messages_and_media/message_reactions.py
View file

@ -51,7 +51,8 @@ class MessageReactions(Object):
def _parse(
client: "pyrogram.Client",
message_reactions: Optional["raw.base.MessageReactions"] = None,
users: Optional[Dict[int, "raw.types.User"]] = None
users: Optional[Dict[int, "raw.types.User"]] = None,
chats: Dict[int, "raw.types.Chat"] = None
) -> Optional["MessageReactions"]:
if not message_reactions:
return None
@ -63,7 +64,7 @@ class MessageReactions(Object):
for reaction in message_reactions.results
],
top_reactors=[
types.MessageReactor._parse(client, reactor, users)
types.MessageReactor._parse(client, reactor, users, chats)
for reactor in message_reactions.top_reactors
]
)

27
pyrogram/types/messages_and_media/message_reactor.py
View file

@ -42,6 +42,9 @@ class MessageReactor(Object):
from_user (:obj:`~pyrogram.types.User`, *optional*):
Information about the reactor.
sender_chat (:obj:`~pyrogram.types.Chat`, *optional*):
Information about the sender chat.
"""
def __init__(
self,
@ -51,7 +54,8 @@ class MessageReactor(Object):
is_top: bool = None,
is_my: bool = None,
is_anonymous: bool = None,
from_user: "types.User" = None
from_user: "types.User" = None,
sender_chat: "types.Chat" = None
):
super().__init__(client)
@ -60,26 +64,39 @@ class MessageReactor(Object):
self.is_my = is_my
self.is_anonymous = is_anonymous
self.from_user = from_user
self.sender_chat = sender_chat
@staticmethod
def _parse(
client: "pyrogram.Client",
message_reactor: Optional["raw.base.MessageReactor"] = None,
users: Dict[int, "raw.types.User"] = None
users: Dict[int, "raw.types.User"] = None,
chats: Dict[int, "raw.types.Chat"] = None
) -> Optional["MessageReactor"]:
if not message_reactor:
return None
is_anonymous = message_reactor.anonymous
from_user = None
sender_chat = None
if not is_anonymous:
from_user = types.User._parse(client, users.get(message_reactor.peer_id.user_id))
if isinstance(message_reactor.peer, raw.types.PeerUser):
from_user = types.User._parse(
client,
users.get(message_reactor.peer_id.user_id)
)
elif isinstance(message_reactor.peer, raw.types.PeerChannel):
sender_chat = types.Chat._parse_channel_chat(
client,
chats.get(message_reactor.peer_id.channel_id)
)
return MessageReactor(
client=client,
amount=message_reactor.count,
is_top=message_reactor.top,
is_my=message_reactor.my,
is_anonymous=is_anonymous,
from_user=from_user
from_user=from_user,
sender_chat=sender_chat
)