diff --git a/pyrogram/methods/bots/send_inline_bot_result.py b/pyrogram/methods/bots/send_inline_bot_result.py index 241798bf..f7eb6f41 100644 --- a/pyrogram/methods/bots/send_inline_bot_result.py +++ b/pyrogram/methods/bots/send_inline_bot_result.py @@ -17,10 +17,11 @@ # You should have received a copy of the GNU Lesser General Public License # along with Pyrofork. If not, see . -from typing import List, Union, Optional +from datetime import datetime +from typing import Union, List, Optional import pyrogram -from pyrogram import enums, raw, types, utils +from pyrogram import raw, enums, types, utils class SendInlineBotResult: @@ -29,12 +30,16 @@ class SendInlineBotResult: chat_id: Union[int, str], query_id: int, result_id: str, - disable_notification: bool = None, - message_thread_id: int = None, - reply_to_message_id: int = None, - quote_text: str = None, - quote_entities: List["types.MessageEntity"] = None, - parse_mode: Optional["enums.ParseMode"] = None + disable_notification: Optional[bool] = None, + message_thread_id: Optional[int] = None, + reply_to_message_id: Optional[int] = None, + reply_to_chat_id: Union[int, str] = None, + reply_to_story_id: Optional[int] = None, + quote_text: Optional[str] = None, + parse_mode: Optional["enums.ParseMode"] = None, + quote_entities: Optional[List["types.MessageEntity"]] = None, + quote_offset: Optional[int] = None, + schedule_date: datetime = None, ) -> "types.Message": """Send an inline bot result. Bot results can be retrieved using :meth:`~pyrogram.Client.get_inline_bot_results` @@ -60,42 +65,42 @@ class SendInlineBotResult: message_thread_id (``int``, *optional*): Unique identifier of a message thread to which the message belongs. - for supergroups only + For supergroups only. reply_to_message_id (``bool``, *optional*): If the message is a reply, ID of the original message. - quote_text (``str``, *optional*): - Text to quote. - for reply_to_message only. + reply_to_chat_id (``int``, *optional*): + If the message is a reply, ID of the original chat. - quote_entities (List of :obj:`~pyrogram.types.MessageEntity`, *optional*): - List of special entities that appear in quote_text, which can be specified instead of *parse_mode*. - for reply_to_message only. + reply_to_story_id (``int``, *optional*): + If the message is a reply, ID of the target story. + + quote_text (``str``, *optional*): + Text of the quote to be sent. parse_mode (:obj:`~pyrogram.enums.ParseMode`, *optional*): - By default, quote_text are parsed using both Markdown and HTML styles. + By default, texts are parsed using both Markdown and HTML styles. You can combine both syntaxes together. - For quote_text. + + quote_entities (List of :obj:`~pyrogram.types.MessageEntity`, *optional*): + List of special entities that appear in quote text, which can be specified instead of *parse_mode*. + + quote_offset (``int``, *optional*): + Offset for quote in original message. + + schedule_date (:py:obj:`~datetime.datetime`, *optional*): + Date when the message will be automatically sent. Returns: - :obj:`~pyrogram.types.Message`: On success, the sent message is returned. + :obj:`~pyrogram.types.Message`: On success, the sent message is returned or False if no message was sent. Example: .. code-block:: python await app.send_inline_bot_result(chat_id, query_id, result_id) """ - - reply_to = await utils.get_reply_to( - client=self, - chat_id=chat_id, - reply_to_message_id=reply_to_message_id, - message_thread_id=message_thread_id, - quote_text=quote_text, - quote_entities=quote_entities, - parse_mode=parse_mode - ) + quote_text, quote_entities = (await utils.parse_text_entities(self, quote_text, parse_mode, quote_entities)).values() r = await self.invoke( raw.functions.messages.SendInlineBotResult( @@ -104,15 +109,27 @@ class SendInlineBotResult: id=result_id, random_id=self.rnd_id(), silent=disable_notification or None, - reply_to=reply_to + reply_to=utils.get_reply_to( + reply_to_message_id=reply_to_message_id, + reply_to_peer=await self.resolve_peer(reply_to_chat_id) if reply_to_chat_id else None, + reply_to_story_id=reply_to_story_id, + message_thread_id=message_thread_id, + quote_text=quote_text, + quote_entities=quote_entities, + quote_offset=quote_offset, + ), + schedule_date=utils.datetime_to_timestamp(schedule_date), ) ) for i in r.updates: if isinstance(i, (raw.types.UpdateNewMessage, - raw.types.UpdateNewChannelMessage)): + raw.types.UpdateNewChannelMessage, + raw.types.UpdateNewScheduledMessage)): return await types.Message._parse( self, i.message, {i.id: i for i in r.users}, {i.id: i for i in r.chats}, - ) + is_scheduled=isinstance(i, raw.types.UpdateNewScheduledMessage), + business_connection_id=getattr(i, "connection_id", None) + ) \ No newline at end of file diff --git a/pyrogram/methods/messages/copy_media_group.py b/pyrogram/methods/messages/copy_media_group.py index 8d73b862..cdd75b6e 100644 --- a/pyrogram/methods/messages/copy_media_group.py +++ b/pyrogram/methods/messages/copy_media_group.py @@ -18,10 +18,10 @@ # along with Pyrofork. If not, see . from datetime import datetime -from typing import Union, List +from typing import Union, List, Optional import pyrogram -from pyrogram import types, utils, raw +from pyrogram import types, utils, raw, enums class CopyMediaGroup: @@ -31,10 +31,18 @@ class CopyMediaGroup: from_chat_id: Union[int, str], message_id: int, captions: Union[List[str], str] = None, + has_spoilers: Union[List[bool], bool] = None, disable_notification: bool = None, message_thread_id: int = None, reply_to_message_id: int = None, + reply_to_chat_id: Union[int, str] = None, + reply_to_story_id: int = None, + quote_text: str = None, + parse_mode: Optional["enums.ParseMode"] = None, + quote_entities: List["types.MessageEntity"] = None, + quote_offset: int = None, schedule_date: datetime = None, + invert_media: bool = None, protect_content: bool = None, ) -> List["types.Message"]: """Copy a media group by providing one of the message ids. @@ -53,7 +61,7 @@ class CopyMediaGroup: For your personal cloud (Saved Messages) you can simply use "me" or "self". For a contact that exists in your Telegram address book you can use his phone number (str). You can also use chat public link in form of *t.me/* (str). - + message_id (``int``): Message identifier in the chat specified in *from_chat_id*. @@ -72,14 +80,36 @@ class CopyMediaGroup: message_thread_id (``int``, *optional*): Unique identifier for the target message thread (topic) of the forum. - for forum supergroups only. + For supergroups only. reply_to_message_id (``int``, *optional*): If the message is a reply, ID of the original message. + reply_to_chat_id (``int``, *optional*): + If the message is a reply, ID of the original chat. + + reply_to_story_id (``int``, *optional*): + If the message is a reply, ID of the target story. + + quote_text (``str``, *optional*): + Text of the quote to be sent. + + parse_mode (:obj:`~pyrogram.enums.ParseMode`, *optional*): + By default, texts are parsed using both Markdown and HTML styles. + You can combine both syntaxes together. + + quote_entities (List of :obj:`~pyrogram.types.MessageEntity`, *optional*): + List of special entities that appear in quote text, which can be specified instead of *parse_mode*. + + quote_offset (``int``, *optional*): + Offset for quote in original message. + schedule_date (:py:obj:`~datetime.datetime`, *optional*): Date when the message will be automatically sent. + invert_media (``bool``, *optional*): + Pass True, if the caption must be shown above the message media. + protect_content (``bool``, *optional*): Protects the contents of the sent message from forwarding and saving @@ -93,18 +123,15 @@ class CopyMediaGroup: await app.copy_media_group(to_chat, from_chat, 123) await app.copy_media_group(to_chat, from_chat, 123, captions="single caption") - + await app.copy_media_group(to_chat, from_chat, 123, captions=["caption 1", None, ""]) """ + quote_text, quote_entities = (await utils.parse_text_entities(self, quote_text, parse_mode, quote_entities)).values() media_group = await self.get_media_group(from_chat_id, message_id) multi_media = [] - reply_to = None - if reply_to_message_id or message_thread_id: - reply_to = types.InputReplyToMessage(reply_to_message_id=reply_to_message_id, message_thread_id=message_thread_id) - for i, message in enumerate(media_group): if message.photo: file_id = message.photo.file_id @@ -117,7 +144,19 @@ class CopyMediaGroup: else: raise ValueError("Message with this type can't be copied.") - media = utils.get_input_media_from_file_id(file_id=file_id) + media = utils.get_input_media_from_file_id( + file_id=file_id, + has_spoiler=( + has_spoilers[i] + if isinstance(has_spoilers, list) + and i < len(has_spoilers) + else ( + has_spoilers + if isinstance(has_spoilers, bool) + else message.has_media_spoiler + ) + ), + ) multi_media.append( raw.types.InputSingleMedia( media=media, @@ -135,9 +174,18 @@ class CopyMediaGroup: peer=await self.resolve_peer(chat_id), multi_media=multi_media, silent=disable_notification or None, - reply_to=reply_to, + reply_to=utils.get_reply_to( + reply_to_message_id=reply_to_message_id, + message_thread_id=message_thread_id, + reply_to_peer=await self.resolve_peer(reply_to_chat_id) if reply_to_chat_id else None, + reply_to_story_id=reply_to_story_id, + quote_text=quote_text, + quote_entities=quote_entities, + quote_offset=quote_offset, + ), + schedule_date=utils.datetime_to_timestamp(schedule_date), noforwards=protect_content, - schedule_date=utils.datetime_to_timestamp(schedule_date) + invert_media=invert_media ), sleep_threshold=60 ) @@ -154,4 +202,4 @@ class CopyMediaGroup: users=r.users, chats=r.chats ) - ) + ) \ No newline at end of file