From f55d75d3bb2ae918e3a6e1ce71d505fd4ec022b6 Mon Sep 17 00:00:00 2001
From: wulan17 <wulan17@nusantararom.org>
Date: Sat, 14 Jan 2023 00:20:05 +0700
Subject: [PATCH] PyroFork: Add support to send animation in media group

Signed-off-by: wulan17 <wulan17@nusantararom.org>
---
 pyrogram/methods/messages/send_media_group.py | 22 ++++++++++++++-----
 requirements.txt                              |  1 +
 2 files changed, 18 insertions(+), 5 deletions(-)

diff --git a/pyrogram/methods/messages/send_media_group.py b/pyrogram/methods/messages/send_media_group.py
index 7e2eafbc..16bfbfaa 100644
--- a/pyrogram/methods/messages/send_media_group.py
+++ b/pyrogram/methods/messages/send_media_group.py
@@ -20,6 +20,7 @@ import logging
 import os
 import re
 from datetime import datetime
+from pymediainfo import MediaInfo
 from typing import Union, List
 
 import pyrogram
@@ -40,7 +41,8 @@ class SendMediaGroup:
             "types.InputMediaPhoto",
             "types.InputMediaVideo",
             "types.InputMediaAudio",
-            "types.InputMediaDocument"
+            "types.InputMediaDocument",
+            "types.InputMediaAnimation"
         ]],
         disable_notification: bool = None,
         message_thread_id: int = None,
@@ -58,7 +60,7 @@ class SendMediaGroup:
                 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).
 
-            media (List of :obj:`~pyrogram.types.InputMediaPhoto`, :obj:`~pyrogram.types.InputMediaVideo`, :obj:`~pyrogram.types.InputMediaAudio` and :obj:`~pyrogram.types.InputMediaDocument`):
+            media (List of :obj:`~pyrogram.types.InputMediaPhoto`, :obj:`~pyrogram.types.InputMediaVideo`, :obj:`~pyrogram.types.InputMediaAudio`, :obj:`~pyrogram.types.InputMediaDocument` and :obj:`~pyrogram.types.InputMediaAnimation`):
                 A list describing photos and videos to be sent, must include 2–10 items.
 
             disable_notification (``bool``, *optional*):
@@ -159,9 +161,17 @@ class SendMediaGroup:
                         ),
                         spoiler=i.has_spoiler
                     )
-            elif isinstance(i, types.InputMediaVideo):
+            elif (
+                isinstance(i, types.InputMediaVideo)
+                or
+                isinstance(i, types.InputMediaAnimation)
+            ):
                 if isinstance(i.media, str):
+                    is_animation = False
                     if os.path.isfile(i.media):
+                        videoInfo = MediaInfo.parse(i.media)
+                        if not any([track.track_type == 'Audio' for track in videoInfo.tracks]):
+                            is_animation = True
                         media = await self.invoke(
                             raw.functions.messages.UploadMedia(
                                 peer=await self.resolve_peer(chat_id),
@@ -170,14 +180,16 @@ class SendMediaGroup:
                                     thumb=await self.save_file(i.thumb),
                                     spoiler=i.has_spoiler,
                                     mime_type=self.guess_mime_type(i.media) or "video/mp4",
+                                    nosound_video=is_animation,
                                     attributes=[
                                         raw.types.DocumentAttributeVideo(
-                                            supports_streaming=i.supports_streaming or None,
+                                            supports_streaming=True if is_animation else (i.supports_streaming or None),
                                             duration=i.duration,
                                             w=i.width,
                                             h=i.height
                                         ),
-                                        raw.types.DocumentAttributeFilename(file_name=os.path.basename(i.media))
+                                        raw.types.DocumentAttributeFilename(file_name=os.path.basename(i.media)),
+                                        raw.types.DocumentAttributeAnimated() if is_animation else None
                                     ]
                                 )
                             )
diff --git a/requirements.txt b/requirements.txt
index bf664153..27a148bb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,3 @@
 pyaes==1.6.1
+pymediainfo==6.0.1
 pysocks==1.7.1