yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2025-12-29 12:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

PyroFork: Add support to send animation in media group

Browse source 
Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
wulan17 2023-01-14 00:20:05 +07:00
parent c484e82810
commit f55d75d3bb
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
2 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
pyrogram/methods/messages/send_media_group.py
View file

@ -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
]
)
)

1
requirements.txt
View file

@ -1,2 +1,3 @@
pyaes==1.6.1
pymediainfo==6.0.1
pysocks==1.7.1