From a9a9dc3cd36edd3b1da23fb8e5c41ac2036e6998 Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Thu, 12 Jul 2018 00:24:57 +0200
Subject: [PATCH] Make download_media raise ValueError in case of non-media
 messages

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

diff --git a/pyrogram/client/methods/download_media.py b/pyrogram/client/methods/download_media.py
index 20179d08..e2fd463f 100644
--- a/pyrogram/client/methods/download_media.py
+++ b/pyrogram/client/methods/download_media.py
@@ -75,6 +75,8 @@ class DownloadMedia(BaseClient):
         Raises:
             :class:`Error <pyrogram.Error>`
         """
+        error_message = "This message doesn't contain any downloadable media"
+
         if isinstance(message, pyrogram_types.Message):
             if message.photo:
                 media = pyrogram_types.Document(
@@ -98,7 +100,7 @@ class DownloadMedia(BaseClient):
             elif message.gif:
                 media = message.gif
             else:
-                return
+                raise ValueError(error_message)
         elif isinstance(message, (
                 pyrogram_types.Photo,
                 pyrogram_types.PhotoSize,
@@ -126,7 +128,7 @@ class DownloadMedia(BaseClient):
                 mime_type=""
             )
         else:
-            return
+            raise ValueError(error_message)
 
         done = Event()
         path = [None]