Skip to content

Commit 4f19785

Browse files
committed
Fix messed up variable names
1 parent 90cee1e commit 4f19785

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

pyrogram/methods/messages/download_media.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
class 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

Comments
 (0)