Handle minified poll updates

delivrance · delivrance · commit 6f2c625cd1c3 · 2019-05-05T15:44:53.000+02:00
diff --git a/pyrogram/client/ext/dispatcher.py b/pyrogram/client/ext/dispatcher.py
@@ -82,7 +82,7 @@ def __init__(self, client, workers: int):
                 lambda upd, usr, cht: (pyrogram.InlineQuery._parse(self.client, upd, usr), InlineQueryHandler),
 
             (types.UpdateMessagePoll,):
-                lambda upd, usr, cht: (pyrogram.Poll._parse(self.client, upd), PollHandler)
+                lambda upd, usr, cht: (pyrogram.Poll._parse_update(self.client, upd), PollHandler)
         }
 
         self.update_parsers = {key: value for key_tuple, value in self.update_parsers.items() for key in key_tuple}
diff --git a/pyrogram/client/types/messages_and_media/poll.py b/pyrogram/client/types/messages_and_media/poll.py
@@ -16,7 +16,7 @@
 # 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 List
+from typing import List, Union
 
 import pyrogram
 from pyrogram.api import types
@@ -71,12 +71,11 @@ def __init__(
         self.chosen_option = chosen_option
 
     @staticmethod
-    def _parse(client, media_poll: types.MessageMediaPoll) -> "Poll":
+    def _parse(client, media_poll: Union[types.MessageMediaPoll, types.UpdateMessagePoll]) -> "Poll":
         poll = media_poll.poll
         results = media_poll.results.results
         total_voters = media_poll.results.total_voters
         chosen_option = None
-
         options = []
 
         for i, answer in enumerate(poll.answers):
@@ -107,3 +106,35 @@ def _parse(client, media_poll: types.MessageMediaPoll) -> "Poll":
             chosen_option=chosen_option,
             client=client
         )
+
+    @staticmethod
+    def _parse_update(client, update: types.UpdateMessagePoll):
+        if update.poll is not None:
+            return Poll._parse(client, update)
+
+        results = update.results.results
+        chosen_option = None
+        options = []
+
+        for i, result in enumerate(results):
+            if result.chosen:
+                chosen_option = i
+
+            options.append(
+                PollOption(
+                    text="",
+                    voter_count=result.voters,
+                    data=result.option,
+                    client=client
+                )
+            )
+
+        return Poll(
+            id=str(update.poll_id),
+            question="",
+            options=options,
+            is_closed=False,
+            total_voters=update.results.total_voters,
+            chosen_option=chosen_option,
+            client=client
+        )

Original file line number	Diff line number	Diff line change
`@@ -82,7 +82,7 @@ def __init__(self, client, workers: int):`
`82`	`82`	`lambda upd, usr, cht: (pyrogram.InlineQuery._parse(self.client, upd, usr), InlineQueryHandler),`
`83`	`83`
`84`	`84`	`(types.UpdateMessagePoll,):`
`85`		`- lambda upd, usr, cht: (pyrogram.Poll._parse(self.client, upd), PollHandler)`
	`85`	`+ lambda upd, usr, cht: (pyrogram.Poll._parse_update(self.client, upd), PollHandler)`
`86`	`86`	`}`
`87`	`87`
`88`	`88`	`self.update_parsers = {key: value for key_tuple, value in self.update_parsers.items() for key in key_tuple}`