Xtao-Labs/MTPyroger
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix Photo downloads

Browse Source
This commit is contained in:
Dan 2018-07-01 19:43:43 +02:00
parent c660d3a7d0
commit c85aa5dab2
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pyrogram/client/methods/download_media.py
View File

@ -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,