TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 12:51:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make send_media_group return the new Messages object

Browse Source
This commit is contained in:
Dan 2019-02-05 11:50:32 +01:00
parent 6109129f73
commit a2263ad8ce
1 changed files with 76 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

110
pyrogram/client/methods/messages/send_media_group.py
View File

@ -17,20 +17,23 @@
# along with Pyrogram. If not, see <http://www.gnu.org/licenses/>. # along with Pyrogram. If not, see <http://www.gnu.org/licenses/>.
import binascii import binascii
import logging
import mimetypes import mimetypes
import os import os
import struct import struct
import time
from typing import Union, List from typing import Union, List
import pyrogram import pyrogram
from pyrogram.api import functions, types from pyrogram.api import functions, types
from pyrogram.api.errors import FileIdInvalid from pyrogram.api.errors import FileIdInvalid, FloodWait
from pyrogram.client.ext import BaseClient, utils from pyrogram.client.ext import BaseClient, utils
log = logging.getLogger(__name__)
class SendMediaGroup(BaseClient): class SendMediaGroup(BaseClient):
# TODO: Add progress parameter # TODO: Add progress parameter
# TODO: Return new Message object
# TODO: Figure out how to send albums using URLs # TODO: Figure out how to send albums using URLs
def send_media_group(self, def send_media_group(self,
chat_id: Union[int, str], chat_id: Union[int, str],
@ -38,7 +41,6 @@ class SendMediaGroup(BaseClient):
disable_notification: bool = None, disable_notification: bool = None,
reply_to_message_id: int = None): reply_to_message_id: int = None):
"""Use this method to send a group of photos or videos as an album. """Use this method to send a group of photos or videos as an album.
On success, an Update containing the sent Messages is returned.
Args: Args:
chat_id (``int`` | ``str``): chat_id (``int`` | ``str``):
@ -57,6 +59,13 @@ class SendMediaGroup(BaseClient):
reply_to_message_id (``int``, *optional*): reply_to_message_id (``int``, *optional*):
If the message is a reply, ID of the original message. If the message is a reply, ID of the original message.
Returns:
On success, a :obj:`Messages <pyrogram.Messages>` object is returned containing all the
single messages sent.
Raises:
:class:`Error <pyrogram.Error>` in case of a Telegram RPC error.
""" """
multi_media = [] multi_media = []
@ -65,14 +74,21 @@ class SendMediaGroup(BaseClient):
if isinstance(i, pyrogram.InputMediaPhoto): if isinstance(i, pyrogram.InputMediaPhoto):
if os.path.exists(i.media): if os.path.exists(i.media):
media = self.send( while True:
functions.messages.UploadMedia( try:
peer=self.resolve_peer(chat_id), media = self.send(
media=types.InputMediaUploadedPhoto( functions.messages.UploadMedia(
file=self.save_file(i.media) peer=self.resolve_peer(chat_id),
media=types.InputMediaUploadedPhoto(
file=self.save_file(i.media)
)
)
) )
) except FloodWait as e:
) log.warning("Sleeping for {}s".format(e.x))
time.sleep(e.x)
else:
break
media = types.InputMediaPhoto( media = types.InputMediaPhoto(
id=types.InputPhoto( id=types.InputPhoto(
@ -106,25 +122,32 @@ class SendMediaGroup(BaseClient):
) )
elif isinstance(i, pyrogram.InputMediaVideo): elif isinstance(i, pyrogram.InputMediaVideo):
if os.path.exists(i.media): if os.path.exists(i.media):
media = self.send( while True:
functions.messages.UploadMedia( try:
peer=self.resolve_peer(chat_id), media = self.send(
media=types.InputMediaUploadedDocument( functions.messages.UploadMedia(
file=self.save_file(i.media), peer=self.resolve_peer(chat_id),
thumb=None if i.thumb is None else self.save_file(i.thumb), media=types.InputMediaUploadedDocument(
mime_type=mimetypes.types_map[".mp4"], file=self.save_file(i.media),
attributes=[ thumb=None if i.thumb is None else self.save_file(i.thumb),
types.DocumentAttributeVideo( mime_type=mimetypes.types_map[".mp4"],
supports_streaming=i.supports_streaming or None, attributes=[
duration=i.duration, types.DocumentAttributeVideo(
w=i.width, supports_streaming=i.supports_streaming or None,
h=i.height duration=i.duration,
), w=i.width,
types.DocumentAttributeFilename(os.path.basename(i.media)) h=i.height
] ),
types.DocumentAttributeFilename(os.path.basename(i.media))
]
)
)
) )
) except FloodWait as e:
) log.warning("Sleeping for {}s".format(e.x))
time.sleep(e.x)
else:
break
media = types.InputMediaDocument( media = types.InputMediaDocument(
id=types.InputDocument( id=types.InputDocument(
@ -165,11 +188,30 @@ class SendMediaGroup(BaseClient):
) )
) )
return self.send( while True:
functions.messages.SendMultiMedia( try:
peer=self.resolve_peer(chat_id), r = self.send(
multi_media=multi_media, functions.messages.SendMultiMedia(
silent=disable_notification or None, peer=self.resolve_peer(chat_id),
reply_to_msg_id=reply_to_message_id multi_media=multi_media,
silent=disable_notification or None,
reply_to_msg_id=reply_to_message_id
)
)
except FloodWait as e:
log.warning("Sleeping for {}s".format(e.x))
time.sleep(e.x)
else:
break
return pyrogram.Messages._parse(
self,
types.messages.Messages(
messages=[m.message for m in filter(
lambda u: isinstance(u, (types.UpdateNewMessage, types.UpdateNewChannelMessage)),
r.updates
)],
users=r.users,
chats=r.chats
) )
) )