From e1dac6c0e241c6332024aa0d3c6b9a7e84ae67d2 Mon Sep 17 00:00:00 2001
From: Legenda24 <48826466+Legenda24@users.noreply.github.com>
Date: Sun, 20 Dec 2020 21:02:54 +0500
Subject: [PATCH] Add new method get_media_group (#550)

* Update __init__.py

Support for GetMediaGroup

* Create get_media_group.py

Added new method - get_media_group

* Update get_media_group.py

Add pyro stuff

* Update get_media_group.py

* Update compiler.py

Co-authored-by: Dan <14043624+delivrance@users.noreply.github.com>
---
 compiler/docs/compiler.py                    |  1 +
 pyrogram/methods/messages/__init__.py        |  2 +
 pyrogram/methods/messages/get_media_group.py | 59 ++++++++++++++++++++
 3 files changed, 62 insertions(+)
 create mode 100644 pyrogram/methods/messages/get_media_group.py

diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py
index 7bc7308f..b1c9b932 100644
--- a/compiler/docs/compiler.py
+++ b/compiler/docs/compiler.py
@@ -168,6 +168,7 @@ def pyrogram_api():
             send_chat_action
             delete_messages
             get_messages
+            get_media_group
             get_history
             get_history_count
             read_history
diff --git a/pyrogram/methods/messages/__init__.py b/pyrogram/methods/messages/__init__.py
index 4c9319a7..f91b7941 100644
--- a/pyrogram/methods/messages/__init__.py
+++ b/pyrogram/methods/messages/__init__.py
@@ -30,6 +30,7 @@ from .edit_message_text import EditMessageText
 from .forward_messages import ForwardMessages
 from .get_history import GetHistory
 from .get_history_count import GetHistoryCount
+from .get_media_group import GetMediaGroup
 from .get_messages import GetMessages
 from .iter_history import IterHistory
 from .read_history import ReadHistory
@@ -65,6 +66,7 @@ class Messages(
     EditMessageText,
     ForwardMessages,
     GetHistory,
+    GetMediaGroup,
     GetMessages,
     SendAudio,
     SendChatAction,
diff --git a/pyrogram/methods/messages/get_media_group.py b/pyrogram/methods/messages/get_media_group.py
new file mode 100644
index 00000000..ab20f8ec
--- /dev/null
+++ b/pyrogram/methods/messages/get_media_group.py
@@ -0,0 +1,59 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-2020 Dan <https://github.com/delivrance>
+#
+#  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 <http://www.gnu.org/licenses/>.
+
+import logging
+from typing import Union, List
+
+from pyrogram.scaffold import Scaffold
+from pyrogram.types import list
+
+log = logging.getLogger(__name__)
+
+
+class GetMediaGroup(Scaffold):
+    async def get_media_group(
+        self,
+        chat_id: Union[int, str],
+        message_id: int
+    ) -> List["types.Message"]:
+        """Get the media group a message belongs to.
+        
+        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).
+
+            message_id (``int``):
+                The id of one of the messages that belong to the media group.
+                
+        Returns:
+            List of :obj:`~pyrogram.types.Message`: On success, a list of messages of the media group is returned.
+            
+        Raises:
+            ValueError: In case the passed message id doesn't belong to a media group.
+        """
+        # There can be maximum 10 items in a media group. 
+        messages = await self.get_messages(chat_id, [msg_id for msg_id in range(message_id - 9, message_id + 10)], replies=0)
+
+        media_group_id = messages[9].media_group_id
+
+        if media_group_id is None:
+            raise ValueError("The message doesn't belong to a media group")
+
+        return list.List(msg for msg in messages if msg.media_group_id == media_group_id)