From c85aa5dab2941ddacd154dc2fc672c318001281c Mon Sep 17 00:00:00 2001
From: Dan <14043624+delivrance@users.noreply.github.com>
Date: Sun, 1 Jul 2018 19:43:43 +0200
Subject: [PATCH] Fix Photo downloads

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

diff --git a/pyrogram/client/methods/download_media.py b/pyrogram/client/methods/download_media.py
index 89b0f61d..20179d08 100644
--- a/pyrogram/client/methods/download_media.py
+++ b/pyrogram/client/methods/download_media.py
@@ -77,7 +77,12 @@ class DownloadMedia(BaseClient):
         """
         if isinstance(message, pyrogram_types.Message):
             if message.photo:
-                media = message.photo.sizes[-1]
+                media = pyrogram_types.Document(
+                    file_id=message.photo.sizes[-1].file_id,
+                    file_size=message.photo.sizes[-1].file_size,
+                    mime_type="",
+                    date=message.photo.date
+                )
             elif message.audio:
                 media = message.audio
             elif message.document:
@@ -96,6 +101,7 @@ class DownloadMedia(BaseClient):
                 return
         elif isinstance(message, (
                 pyrogram_types.Photo,
+                pyrogram_types.PhotoSize,
                 pyrogram_types.Audio,
                 pyrogram_types.Document,
                 pyrogram_types.Video,
@@ -104,7 +110,15 @@ class DownloadMedia(BaseClient):
                 pyrogram_types.Sticker,
                 pyrogram_types.GIF
         )):
-            media = message
+            if isinstance(message, pyrogram_types.Photo):
+                media = pyrogram_types.Document(
+                    file_id=message.sizes[-1].file_id,
+                    file_size=message.sizes[-1].file_size,
+                    mime_type="",
+                    date=message.date
+                )
+            else:
+                media = message
         elif isinstance(message, str):
             media = pyrogram_types.Document(
                 file_id=message,