diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py index 1d4ba9e4..9513899a 100644 --- a/compiler/docs/compiler.py +++ b/compiler/docs/compiler.py @@ -144,6 +144,7 @@ def pyrogram_api(): send_message forward_messages copy_message + copy_media_group send_photo send_audio send_document diff --git a/pyrogram/methods/messages/__init__.py b/pyrogram/methods/messages/__init__.py index 879b17f1..5ebedfde 100644 --- a/pyrogram/methods/messages/__init__.py +++ b/pyrogram/methods/messages/__init__.py @@ -17,6 +17,7 @@ # along with Pyrogram. If not, see . from .copy_message import CopyMessage +from .copy_media_group import CopyMediaGroup from .delete_messages import DeleteMessages from .download_media import DownloadMedia from .edit_inline_caption import EditInlineCaption @@ -98,6 +99,7 @@ class Messages( SendDice, SearchMessages, SearchGlobal, - CopyMessage + CopyMessage, + CopyMediaGroup ): pass diff --git a/pyrogram/methods/messages/copy_media_group.py b/pyrogram/methods/messages/copy_media_group.py new file mode 100644 index 00000000..7845863c --- /dev/null +++ b/pyrogram/methods/messages/copy_media_group.py @@ -0,0 +1,133 @@ +# Pyrogram - Telegram MTProto API Client Library for Python +# Copyright (C) 2017-2021 Dan +# +# This file is part of Pyrogram. +# +# Pyrogram is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License as published +# by the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Pyrogram is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public License +# along with Pyrogram. If not, see . + +from typing import Union, List, Optional + +from pyrogram import types, utils, raw +from pyrogram.scaffold import Scaffold + + +class CopyMediaGroup(Scaffold): + async def copy_media_group( + self, + chat_id: Union[int, str], + from_chat_id: Union[int, str], + message_id: int, + captions: Union[List[str], str] = None, + disable_notification: bool = None, + reply_to_message_id: int = None, + schedule_date: int = None, + ) -> List["types.Message"]: + """Copy a media group by providing one of the message ids. + + Parameters: + chat_id (``int`` | ``str``): + Unique identifier (int) or username (str) of the target chat. + 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). + + from_chat_id (``int`` | ``str``): + Unique identifier (int) or username (str) of the source chat where the original media group was sent. + 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). + + message_id (``int``): + Message identifier in the chat specified in *from_chat_id*. + + captions (``str`` | List of ``str`` , *optional*): + New caption for media, 0-1024 characters after entities parsing for each media. + If not specified, the original caption is kept. + Pass "" (empty string) to remove the caption. + + If a ``string`` is passed, it becomes a caption only for the first media. + If a list of ``string`` passed, each element becomes caption for each media element. + You can pass ``None`` in list to keep the original caption (see examples below). + + disable_notification (``bool``, *optional*): + Sends the message silently. + Users will receive a notification with no sound. + + reply_to_message_id (``int``, *optional*): + If the message is a reply, ID of the original message. + + schedule_date (``int``, *optional*): + Date when the message will be automatically sent. Unix time. + + Returns: + List of :obj:`~pyrogram.types.Message`: On success, a list of copied messages is returned. + + Example: + .. code-block:: python + + # Copy a media group + app.copy_media_group("me", source_chat, message_id) + app.copy_media_group("me", source_chat, message_id, captions="single caption") + app.copy_media_group("me", source_chat, message_id, captions=["caption 1", None, ""]) + """ + + media_group = await self.get_media_group(from_chat_id, message_id) + multi_media = [] + + for i, message in enumerate(media_group): + if message.photo: + file_id = message.photo.file_id + elif message.audio: + file_id = message.audio.file_id + elif message.document: + file_id = message.document.file_id + elif message.video: + file_id = message.video.file_id + else: + raise ValueError("Message with this type can't be copied.") + + media = utils.get_input_media_from_file_id(file_id=file_id) + multi_media.append( + raw.types.InputSingleMedia( + media=media, + random_id=self.rnd_id(), + **await self.parser.parse( + captions[i] if isinstance(captions, list) and i < len(captions) and captions[i] else + captions if isinstance(captions, str) and i == 0 else + message.caption if message.caption and message.caption != "None" and not type(captions) is str else "") + ) + ) + + r = await self.send( + raw.functions.messages.SendMultiMedia( + peer=await self.resolve_peer(chat_id), + multi_media=multi_media, + silent=disable_notification or None, + reply_to_msg_id=reply_to_message_id, + schedule_date=schedule_date + ), + sleep_threshold=60 + ) + + return await utils.parse_messages( + self, + raw.types.messages.Messages( + messages=[m.message for m in filter( + lambda u: isinstance(u, (raw.types.UpdateNewMessage, + raw.types.UpdateNewChannelMessage, + raw.types.UpdateNewScheduledMessage)), + r.updates + )], + users=r.users, + chats=r.chats + ) + )