mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 21:07:59 +00:00
Code Issues Releases Wiki Activity

Make download_media raise ValueError in case of non-media messages

Browse Source
This commit is contained in:
Dan 2018-07-12 00:24:57 +02:00
parent f242aceb31
commit a9a9dc3cd3
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pyrogram/client/methods/download_media.py
View File

@ -75,6 +75,8 @@ class DownloadMedia(BaseClient):
Raises: Raises:
:class:`Error <pyrogram.Error>` :class:`Error <pyrogram.Error>`
""" """
error_message = "This message doesn't contain any downloadable media"
if isinstance(message, pyrogram_types.Message): if isinstance(message, pyrogram_types.Message):
if message.photo: if message.photo:
media = pyrogram_types.Document( media = pyrogram_types.Document(
@ -98,7 +100,7 @@ class DownloadMedia(BaseClient):
elif message.gif: elif message.gif:
media = message.gif media = message.gif
else: else:
return raise ValueError(error_message)
elif isinstance(message, ( elif isinstance(message, (
pyrogram_types.Photo, pyrogram_types.Photo,
pyrogram_types.PhotoSize, pyrogram_types.PhotoSize,
@ -126,7 +128,7 @@ class DownloadMedia(BaseClient):
mime_type="" mime_type=""
) )
else: else:
return raise ValueError(error_message)
done = Event() done = Event()
path = [None] path = [None]