Add support for public polls, in an unstable way

Git-Original-Commit-Id: KurimuzonAkuma/pyrogram@046312a

Co-authored-by: KurimuzonAkuma <[email protected]>

Author: SpEcHiDe

pyrogram/dispatcher.py

@@ -51,7 +51,8 @@
     UpdateEditMessage, UpdateEditChannelMessage,
     UpdateDeleteMessages, UpdateDeleteChannelMessages,
     UpdateBotCallbackQuery, UpdateInlineBotCallbackQuery,
-    UpdateUserStatus, UpdateBotInlineQuery, UpdateMessagePoll,
+    UpdateUserStatus, UpdateBotInlineQuery,
+    UpdateMessagePoll, UpdateMessagePollVote,
     UpdateBotInlineSend, UpdateChatParticipant, UpdateChannelParticipant, UpdateBotStopped,
     UpdateBotChatInviteRequester,
     UpdateBotMessageReaction,
@@ -76,7 +77,7 @@ class Dispatcher:
     CHAT_MEMBER_UPDATES = (UpdateChatParticipant, UpdateChannelParticipant, UpdateBotStopped,)
     USER_STATUS_UPDATES = (UpdateUserStatus,)
     BOT_INLINE_QUERY_UPDATES = (UpdateBotInlineQuery,)
-    POLL_UPDATES = (UpdateMessagePoll,)
+    POLL_UPDATES = (UpdateMessagePoll, UpdateMessagePollVote,)
     CHOSEN_INLINE_RESULT_UPDATES = (UpdateBotInlineSend,)
     CHAT_JOIN_REQUEST_UPDATES = (UpdateBotChatInviteRequester,)
     MESSAGE_BOT_NA_REACTION_UPDATES = (UpdateBotMessageReaction,)
@@ -146,7 +147,9 @@ async def inline_query_parser(update, users, chats):
 
         async def poll_parser(update, users, chats):
             return (
-                pyrogram.types.Poll._parse_update(self.client, update),
+                pyrogram.types.Poll._parse_update(
+                    self.client, update, users, chats
+                ),
                 PollHandler
             )
 

pyrogram/types/messages_and_media/poll.py

@@ -77,6 +77,13 @@ class Poll(Object, Update):
 
         close_date (:py:obj:`~datetime.datetime`, *optional*):
             Point in time when the poll will be automatically closed.
+        
+        user (:obj:`~pyrogram.types.User`, *optional*):
+            The user that changed the answer to the poll, if the voter isn't anonymous.
+        
+        voter_chat (:obj:`~pyrogram.types.Chat`, *optional*):
+            The chat that changed the answer to the poll, if the voter is anonymous.
+
     """
 
     def __init__(
@@ -97,7 +104,9 @@ def __init__(
         explanation: Optional[str] = None,
         explanation_entities: Optional[List["types.MessageEntity"]] = None,
         open_period: Optional[int] = None,
-        close_date: Optional[datetime] = None
+        close_date: Optional[datetime] = None,
+        user: Optional["types.User"] = None,
+        voter_chat: Optional["types.Chat"] = None,
     ):
         super().__init__(client)
 
@@ -116,6 +125,9 @@ def __init__(
         self.explanation_entities = explanation_entities
         self.open_period = open_period
         self.close_date = close_date
+        # TODO: spit this out
+        self.user = user
+        self.voter_chat = voter_chat
 
     @staticmethod
     def _parse(client, media_poll: Union["raw.types.MessageMediaPoll", "raw.types.UpdateMessagePoll"]) -> "Poll":
@@ -193,44 +205,81 @@ def _parse(client, media_poll: Union["raw.types.MessageMediaPoll", "raw.types.Up
         )
 
     @staticmethod
-    def _parse_update(client, update: "raw.types.UpdateMessagePoll"):
-        if update.poll is not None:
-            return Poll._parse(client, update)
-
-        # TODO: FIXME!
-        results = update.results.results
-        chosen_option_id = None
-        correct_option_id = None
-        options = []
-        question = ""
-
-        for i, result in enumerate(results):
-            if result.chosen:
-                chosen_option_id = i
+    def _parse_update(
+        client,
+        update: Union["raw.types.UpdateMessagePoll", "raw.types.UpdateMessagePollVote"],
+        users: dict,
+        chats: dict,
+    ):
+        if isinstance(update, raw.types.UpdateMessagePoll):
+            if update.poll is not None:
+                return Poll._parse(client, update)
+
+            # TODO: FIXME!
+            results = update.results.results
+            chosen_option_id = None
+            correct_option_id = None
+            options = []
+            question = ""
+
+            for i, result in enumerate(results):
+                if result.chosen:
+                    chosen_option_id = i
 
-            if result.correct:
-                correct_option_id = i
+                if result.correct:
+                    correct_option_id = i
 
-            options.append(
-                types.PollOption(
-                    text="",
-                    text_entities=[],
-                    voter_count=result.voters,
-                    data=result.option,
-                    client=client
+                options.append(
+                    types.PollOption(
+                        text="",
+                        text_entities=[],
+                        voter_count=result.voters,
+                        data=result.option,
+                        client=client
+                    )
                 )
+
+            return Poll(
+                id=str(update.poll_id),
+                question=question,
+                options=options,
+                total_voter_count=update.results.total_voters,
+                is_closed=False,
+                chosen_option_id=chosen_option_id,
+                correct_option_id=correct_option_id,
+                client=client
             )
 
-        return Poll(
-            id=str(update.poll_id),
-            question=question,
-            options=options,
-            total_voter_count=update.results.total_voters,
-            is_closed=False,
-            chosen_option_id=chosen_option_id,
-            correct_option_id=correct_option_id,
-            client=client
-        )
+        if isinstance(update, raw.types.UpdateMessagePollVote):
+            user = None
+            voter_chat = None
+            if isinstance(update.peer, raw.types.PeerUser):
+                user = types.Chat._parse_user_chat(client, users[update.peer.user_id])
+
+            elif isinstance(update.peer, raw.types.PeerChat):
+                voter_chat = types.Chat._parse_chat_chat(client, chats[update.peer.chat_id])
+
+            else:
+                voter_chat = types.Chat._parse_channel_chat(client, chats[update.peer.channel_id])
+
+            return Poll(
+                id=str(update.poll_id),
+                question="",
+                options=[
+                    types.PollOption(
+                        text="",
+                        text_entities=[],
+                        voter_count=None,
+                        data=option,
+                        client=client
+                    ) for option in update.options
+                ],
+                total_voter_count=None,
+                is_closed=False,
+                user=user,
+                voter_chat=voter_chat,
+                client=client
+            )
 
     async def stop(
         self,