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

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]