Add missing attributes in MessageReactions type

Author: KurimuzonAkuma

compiler/docs/compiler.py

@@ -702,6 +702,7 @@ def get_title_list(s: str) -> list:
             PaidMediaInfo
             PaidMediaPreview
             PaidMessagesRefunded
+            PaidReactor
             PaidMessagesPriceChanged
             DirectMessagePriceChanged
             DirectMessagesTopic

pyrogram/types/messages_and_media/__init__.py

@@ -67,6 +67,7 @@
 from .paid_media_preview import PaidMediaPreview
 from .paid_messages_price_changed import PaidMessagesPriceChanged
 from .paid_messages_refunded import PaidMessagesRefunded
+from .paid_reactor import PaidReactor
 from .payment_form import PaymentForm
 from .photo import Photo
 from .poll import Poll
@@ -152,6 +153,7 @@
     "PaidMediaPreview",
     "PaidMessagesPriceChanged",
     "PaidMessagesRefunded",
+    "PaidReactor",
     "PaymentForm",
     "Photo",
     "Poll",

pyrogram/types/messages_and_media/message.py

@@ -1124,7 +1124,7 @@ async def _parse_service(
             paid_messages_refunded=paid_messages_refunded,
             paid_messages_price_changed=paid_messages_price_changed,
             direct_message_price_changed=direct_message_price_changed,
-            reactions=types.MessageReactions._parse(client, message.reactions),
+            reactions=types.MessageReactions._parse(client, message.reactions, users, chats),
             business_connection_id=business_connection_id,
             raw=message,
             client=client
@@ -1356,7 +1356,7 @@ async def _parse_message(
             else:
                 reply_markup = None
 
-        reactions = types.MessageReactions._parse(client, message.reactions)
+        reactions = types.MessageReactions._parse(client, message.reactions, users, chats)
 
         parsed_message = Message(
             id=message.id,

pyrogram/types/messages_and_media/message_reactions.py

@@ -16,10 +16,11 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from typing import Optional, List
+from typing import Dict, List, Optional
 
 import pyrogram
 from pyrogram import raw, types
+
 from ..object import Object
 
 
@@ -29,28 +30,61 @@ class MessageReactions(Object):
     Parameters:
         reactions (List of :obj:`~pyrogram.types.Reaction`):
             Reactions list.
+
+        are_tags (``bool``, *optional*):
+            True, if the reactions are tags and Telegram Premium users can filter messages by them.
+
+        paid_reactors (List of :obj:`~pyrogram.types.PaidReactor`, *optional*):
+            Information about top users that added the paid reaction.
+
+        can_get_added_reactions (``bool``, *optional*):
+            True, if the list of added reactions is available using :meth:`~pyrogram.Client.get_message_added_reactions`.
     """
 
+    # TODO: Add get_message_added_reactions method
+
     def __init__(
         self,
         *,
         client: "pyrogram.Client" = None,
         reactions: Optional[List["types.Reaction"]] = None,
+        are_tags: Optional[bool] = None,
+        paid_reactors: Optional[List["types.PaidReactor"]] = None,
+        can_get_added_reactions: Optional[bool] = None,
     ):
         super().__init__(client)
 
         self.reactions = reactions
+        self.are_tags = are_tags
+        self.paid_reactors = paid_reactors
+        self.can_get_added_reactions = can_get_added_reactions
 
     @staticmethod
     def _parse(
         client: "pyrogram.Client",
-        message_reactions: Optional["raw.base.MessageReactions"] = None
+        message_reactions: Optional["raw.base.MessageReactions"],
+        users: Dict[int, "types.User"],
+        chats: Dict[int, "types.Chat"],
     ) -> Optional["MessageReactions"]:
         if not message_reactions:
             return None
 
         return MessageReactions(
             client=client,
-            reactions=[types.Reaction._parse_count(client, reaction)
-                       for reaction in message_reactions.results]
+            reactions=types.List(
+                [
+                    types.Reaction._parse_count(client, reaction)
+                    for reaction in message_reactions.results
+                ]
+            ),
+            are_tags=message_reactions.reactions_as_tags,
+            paid_reactors=types.List(
+                [
+                    types.PaidReactor._parse(client, paid_reactor, users, chats)
+                    for paid_reactor in message_reactions.top_reactors
+                ]
+            )
+            if message_reactions.top_reactors
+            else None,
+            can_get_added_reactions=message_reactions.can_see_list,
         )

pyrogram/types/messages_and_media/paid_reactor.py

@@ -0,0 +1,86 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Lesser General Public License as published
+#  by the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Pyrogram is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public License
+#  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+from typing import Dict, Optional
+
+import pyrogram
+from pyrogram import raw, types, utils
+
+from ..object import Object
+
+
+class PaidReactor(Object):
+    """Contains information about a user that added paid reactions.
+
+    Parameters:
+        sender (:obj:`~pyrogram.types.Chat`, *optional*):
+            Identifier of the user or chat that added the reactions.
+            May be None for anonymous reactors that aren't the current user
+
+        star_count (``int``, *optional*):
+            True, if the reactions are tags and Telegram Premium users can filter messages by them.
+
+        is_top (``bool``, *optional*):
+            True, if the reactor is one of the most active reactors.
+            May be False if the reactor is the current user.
+
+        is_me (``bool``, *optional*):
+            True, if the paid reaction was added by the current user.
+
+        is_anonymous (``bool``, *optional*):
+            True, if the reactor is anonymous.
+    """
+
+    def __init__(
+        self,
+        *,
+        sender: Optional["types.Chat"] = None,
+        star_count: Optional[int] = None,
+        is_top: Optional[bool] = None,
+        is_me: Optional[bool] = None,
+        is_anonymous: Optional[bool] = None,
+    ):
+        super().__init__()
+
+        self.sender = sender
+        self.star_count = star_count
+        self.is_top = is_top
+        self.is_me = is_me
+        self.is_anonymous = is_anonymous
+
+    @staticmethod
+    def _parse(
+        client: "pyrogram.Client",
+        paid_reactor: Optional["raw.base.MessageReactor"],
+        users: Dict[int, "raw.base.User"],
+        chats: Dict[int, "raw.base.Chat"],
+    ) -> Optional["PaidReactor"]:
+        if not paid_reactor:
+            return None
+
+        chat = chats.get(utils.get_raw_peer_id(paid_reactor.peer_id)) or users.get(
+            utils.get_raw_peer_id(paid_reactor.peer_id)
+        )
+
+        return PaidReactor(
+            sender=types.Chat._parse_chat(client, chat),
+            star_count=paid_reactor.count,
+            is_top=paid_reactor.top,
+            is_me=paid_reactor.my,
+            is_anonymous=paid_reactor.anonymous,
+        )