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

Fix messed up variable names

Browse Source
This commit is contained in:
Dan 2020-11-28 17:37:39 +01:00
parent 90cee1ea89
commit 4f197855f4
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyrogram/methods/messages/download_media.py
View File

@ -32,7 +32,7 @@ DEFAULT_DOWNLOAD_DIR = "downloads/"
class DownloadMedia(Scaffold):
async def download_media(
self,
media: Union["types.Message", str],
message: Union["types.Message", str],
file_name: str = DEFAULT_DOWNLOAD_DIR,
block: bool = True,
progress: callable = None,
@ -41,7 +41,7 @@ class DownloadMedia(Scaffold):
"""Download the media from a message.
Parameters:
media (:obj:`~pyrogram.types.Message` | ``str``):
message (:obj:`~pyrogram.types.Message` | ``str``):
Pass a Message containing the media, the media itself (message.audio, message.video, ...) or a file id
as string.
@ -103,14 +103,16 @@ class DownloadMedia(Scaffold):
available_media = ("audio", "document", "photo", "sticker", "animation", "video", "voice", "video_note",
"new_chat_photo")
if isinstance(media, types.Message):
if isinstance(message, types.Message):
for kind in available_media:
media = getattr(media, kind, None)
media = getattr(message, kind, None)
if media is not None:
break
else:
raise ValueError("This message doesn't contain any downloadable media")
else:
media = message
if isinstance(media, str):
file_id_str = media