From ae3bc516fb2b08e5d834eecb5814004e6afccfe6 Mon Sep 17 00:00:00 2001 From: KurimuzonAkuma Date: Thu, 24 Oct 2024 13:31:27 +0300 Subject: [PATCH] pyrofork: Add forward_media_group method Signed-off-by: Yasir Aris M --- compiler/docs/compiler.py | 1 + pyrogram/methods/messages/__init__.py | 2 + .../methods/messages/forward_media_group.py | 120 ++++++++++++++++++ 3 files changed, 123 insertions(+) create mode 100644 pyrogram/methods/messages/forward_media_group.py diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py index dc1de959..6a199443 100644 --- a/compiler/docs/compiler.py +++ b/compiler/docs/compiler.py @@ -165,6 +165,7 @@ def pyrogram_api(): messages=""" Messages send_message + forward_media_group forward_messages copy_message copy_media_group diff --git a/pyrogram/methods/messages/__init__.py b/pyrogram/methods/messages/__init__.py index dc2ccd98..a29275c4 100644 --- a/pyrogram/methods/messages/__init__.py +++ b/pyrogram/methods/messages/__init__.py @@ -30,6 +30,7 @@ from .edit_message_caption import EditMessageCaption from .edit_message_media import EditMessageMedia from .edit_message_reply_markup import EditMessageReplyMarkup from .edit_message_text import EditMessageText +from .forward_media_group import ForwardMediaGroup from .forward_messages import ForwardMessages from .get_available_effects import GetAvailableEffects from .get_chat_history import GetChatHistory @@ -81,6 +82,7 @@ class Messages( EditMessageReplyMarkup, EditMessageMedia, EditMessageText, + ForwardMediaGroup, ForwardMessages, GetAvailableEffects, GetMediaGroup, diff --git a/pyrogram/methods/messages/forward_media_group.py b/pyrogram/methods/messages/forward_media_group.py new file mode 100644 index 00000000..59e594a3 --- /dev/null +++ b/pyrogram/methods/messages/forward_media_group.py @@ -0,0 +1,120 @@ +# Pyrogram - Telegram MTProto API Client Library for Python +# Copyright (C) 2017-present 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 datetime import datetime +from typing import Union, List + +import pyrogram +from pyrogram import raw, utils +from pyrogram import types + + +class ForwardMediaGroup: + async def forward_media_group( + self: "pyrogram.Client", + chat_id: Union[int, str], + from_chat_id: Union[int, str], + message_id: int, + message_thread_id: int = None, + disable_notification: bool = None, + schedule_date: datetime = None, + hide_sender_name: bool = None, + hide_captions: bool = None, + protect_content: bool = None + ) -> List["types.Message"]: + """Forward a media group by providing one of the message ids. + + .. include:: /_includes/usable-by/users-bots.rst + + 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 message 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*. + + message_thread_id (``int``, *optional*): + Unique identifier of a message thread to which the message belongs. + For supergroups only. + + disable_notification (``bool``, *optional*): + Sends the message silently. + Users will receive a notification with no sound. + + schedule_date (:py:obj:`~datetime.datetime`, *optional*): + Date when the message will be automatically sent. + + hide_sender_name (``bool``, *optional*): + If True, the original author of the message will not be shown. + + hide_captions (``bool``, *optional*): + If True, the original media captions will be removed. + + protect_content (``bool``, *optional*): + Protects the contents of the sent message from forwarding and saving. + + Returns: + List of :obj:`~pyrogram.types.Message`: On success, a list of forwarded messages is returned. + + Example: + .. code-block:: python + + # Forward a media group + await app.forward_media_group(to_chat, from_chat, 123) + """ + message_ids = [i.id for i in await self.get_media_group(from_chat_id, message_id)] + + r = await self.invoke( + raw.functions.messages.ForwardMessages( + to_peer=await self.resolve_peer(chat_id), + from_peer=await self.resolve_peer(from_chat_id), + id=message_ids, + silent=disable_notification or None, + random_id=[self.rnd_id() for _ in message_ids], + schedule_date=utils.datetime_to_timestamp(schedule_date), + drop_author=hide_sender_name, + drop_media_captions=hide_captions, + noforwards=protect_content, + top_msg_id=message_thread_id + ) + ) + + forwarded_messages = [] + + users = {i.id: i for i in r.users} + chats = {i.id: i for i in r.chats} + + for i in r.updates: + if isinstance(i, (raw.types.UpdateNewMessage, + raw.types.UpdateNewChannelMessage, + raw.types.UpdateNewScheduledMessage)): + forwarded_messages.append( + await types.Message._parse( + self, i.message, + users, chats + ) + ) + + return types.List(forwarded_messages)