mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 05:18:10 +00:00
Code Issues Releases Wiki Activity

Make forward_messages accept iterables

Browse Source
This commit is contained in:
Dan 2018-04-30 14:24:05 +02:00
parent 59ab1591d2
commit b41c009962
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyrogram/client/client.py
View File

@ -1188,7 +1188,7 @@ class Client:
def forward_messages(self, def forward_messages(self,
chat_id: int or str, chat_id: int or str,
from_chat_id: int or str, from_chat_id: int or str,
message_ids: list or int, message_ids,
disable_notification: bool = None): disable_notification: bool = None):
"""Use this method to forward messages of any kind. """Use this method to forward messages of any kind.
@ -1205,8 +1205,9 @@ class Client:
For a contact that exists in your Telegram address book you can use his phone number (str). For a contact that exists in your Telegram address book you can use his phone number (str).
For a private channel/supergroup you can use its *t.me/joinchat/* link. For a private channel/supergroup you can use its *t.me/joinchat/* link.
message_ids (``list`` | ``int``): message_ids (``iterable``):
A list of Message identifiers in the chat specified in *from_chat_id* or a single message id. A list of Message identifiers in the chat specified in *from_chat_id* or a single message id.
Iterators and Generators are also accepted.
disable_notification (``bool``, optional): disable_notification (``bool``, optional):
Sends the message silently. Sends the message silently.
@ -1221,8 +1222,8 @@ class Client:
Raises: Raises:
:class:`Error <pyrogram.Error>` :class:`Error <pyrogram.Error>`
""" """
is_list = isinstance(message_ids, list) is_iterable = not isinstance(message_ids, int)
message_ids = message_ids if is_list else [message_ids] message_ids = list(message_ids) if is_iterable else [message_ids]
r = self.send( r = self.send(
functions.messages.ForwardMessages( functions.messages.ForwardMessages(
@ -1245,7 +1246,7 @@ class Client:
utils.parse_message(self, i.message, users, chats) utils.parse_message(self, i.message, users, chats)
) )
return messages if is_list else messages[0] return messages if is_iterable else messages[0]
def send_photo(self, def send_photo(self,
chat_id: int or str, chat_id: int or str,