Visitar URL original
Make download_media raise ValueError in case of non-media messages · usergao/pyrogram@a9a9dc3 · GitHub
Skip to content

Commit a9a9dc3

Browse files
committed
Make download_media raise ValueError in case of non-media messages
1 parent f242ace commit a9a9dc3

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

‎pyrogram/client/methods/download_media.py‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,8 @@ def download_media(self,
7575
Raises:
7676
:class:`Error <pyrogram.Error>`
7777
"""
78+
error_message = "This message doesn't contain any downloadable media"
79+
7880
if isinstance(message, pyrogram_types.Message):
7981
if message.photo:
8082
media = pyrogram_types.Document(
@@ -98,7 +100,7 @@ def download_media(self,
98100
elif message.gif:
99101
media = message.gif
100102
else:
101-
return
103+
raise ValueError(error_message)
102104
elif isinstance(message, (
103105
pyrogram_types.Photo,
104106
pyrogram_types.PhotoSize,
@@ -126,7 +128,7 @@ def download_media(self,
126128
mime_type=""
127129
)
128130
else:
129-
return
131+
raise ValueError(error_message)
130132

131133
done = Event()
132134
path = [None]

0 commit comments

Comments
 (0)