LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-10 19:46:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix .download_media() not handling Photo (closes #473)

Browse Source
This commit is contained in:
Lonami Exo 2017-12-28 12:11:31 +01:00
parent bfff1567af
commit 75a342e24b
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon/telegram_client.py
View File

@ -52,7 +52,7 @@ from .tl.types import (
InputUserSelf, UserProfilePhoto, ChatPhoto, UpdateMessageID,
UpdateNewChannelMessage, UpdateNewMessage, UpdateShortSentMessage,
PeerUser, InputPeerUser, InputPeerChat, InputPeerChannel, MessageEmpty,
ChatInvite, ChatInviteAlready, PeerChannel
ChatInvite, ChatInviteAlready, PeerChannel, Photo
)
from .tl.types.messages import DialogsSlice
from .extensions import markdown
@ -848,7 +848,7 @@ class TelegramClient(TelegramBareClient):
date = datetime.now()
media = message
if isinstance(media, MessageMediaPhoto):
if isinstance(media, (MessageMediaPhoto, Photo)):
return self._download_photo(
media, file, date, progress_callback
)
@ -861,11 +861,15 @@ class TelegramClient(TelegramBareClient):
media, file
)
def _download_photo(self, mm_photo, file, date, progress_callback):
def _download_photo(self, photo, file, date, progress_callback):
"""Specialized version of .download_media() for photos"""
# Determine the photo and its largest size
photo = getattr(mm_photo, 'photo', mm_photo)
if isinstance(photo, MessageMediaPhoto):
photo = photo.photo
if not isinstance(photo, Photo):
return
largest_size = photo.sizes[-1]
file_size = largest_size.size
largest_size = largest_size.location