Add bound method vote to Message

kalmengr · web-flow · commit 2095f9fb5e3c · 2019-07-12T12:05:19.000-04:00
diff --git a/pyrogram/client/types/messages_and_media/message.py b/pyrogram/client/types/messages_and_media/message.py
@@ -2923,6 +2923,43 @@ def download(
             progress=progress,
             progress_args=progress_args,
         )
+    def vote(
+        self,
+        option: int,
+    ) -> "Poll":
+        """Bound method *vote* of :obj:`Message`.
+
+        Use as a shortcut for:
+
+        .. code-block:: python
+
+            client.vote_poll(
+                chat_id=message.chat.id,
+                message_id=message.message_id,
+                option=1
+            )
+
+        Example:
+            .. code-block:: python
+
+                message.vote(6)
+
+        Parameters:
+            option (``int``):
+                Index of the poll option you want to vote for (0 to 9).
+            
+        Returns:
+            On success, the poll with the chosen option is returned.
+
+        Raises:
+            RPCError: In case of a Telegram RPC error.
+        """
+
+        return self._client.vote_poll(
+            chat_id=self.chat.id,
+            message_id= self.message_id,
+            option=option
+        )
 
     def pin(self, disable_notification: bool = None) -> "Message":
         """Bound method *pin* of :obj:`Message`.
