mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 12:57:52 +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:
:class:`Error <pyrogram.Error>`
"""
error_message = "This message doesn't contain any downloadable media"
if isinstance(message, pyrogram_types.Message):
if message.photo:
media = pyrogram_types.Document(
@ -98,7 +100,7 @@ class DownloadMedia(BaseClient):
elif message.gif:
media = message.gif
else:
return
raise ValueError(error_message)
elif isinstance(message, (
pyrogram_types.Photo,
pyrogram_types.PhotoSize,
@ -126,7 +128,7 @@ class DownloadMedia(BaseClient):
mime_type=""
)
else:
return
raise ValueError(error_message)
done = Event()
path = [None]