From 4f197855f4af38f5dc49ce4ff76e062d25227623 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Sat, 28 Nov 2020 17:37:39 +0100
Subject: [PATCH] Fix messed up variable names

---
 pyrogram/methods/messages/download_media.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/pyrogram/methods/messages/download_media.py b/pyrogram/methods/messages/download_media.py
index c2d4d0b4..db9493f3 100644
--- a/pyrogram/methods/messages/download_media.py
+++ b/pyrogram/methods/messages/download_media.py
@@ -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