3232class DownloadMedia (Scaffold ):
3333 async def download_media (
3434 self ,
35- media : Union ["types.Message" , str ],
35+ message : Union ["types.Message" , str ],
3636 file_name : str = DEFAULT_DOWNLOAD_DIR ,
3737 block : bool = True ,
3838 progress : callable = None ,
@@ -41,7 +41,7 @@ async def download_media(
4141 """Download the media from a message.
4242
4343 Parameters:
44- media (:obj:`~pyrogram.types.Message` | ``str``):
44+ message (:obj:`~pyrogram.types.Message` | ``str``):
4545 Pass a Message containing the media, the media itself (message.audio, message.video, ...) or a file id
4646 as string.
4747
@@ -103,14 +103,16 @@ def progress(current, total):
103103 available_media = ("audio" , "document" , "photo" , "sticker" , "animation" , "video" , "voice" , "video_note" ,
104104 "new_chat_photo" )
105105
106- if isinstance (media , types .Message ):
106+ if isinstance (message , types .Message ):
107107 for kind in available_media :
108- media = getattr (media , kind , None )
108+ media = getattr (message , kind , None )
109109
110110 if media is not None :
111111 break
112112 else :
113113 raise ValueError ("This message doesn't contain any downloadable media" )
114+ else :
115+ media = message
114116
115117 if isinstance (media , str ):
116118 file_id_str = media
0 commit comments