LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-03 05:04:33 +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, InputUserSelf, UserProfilePhoto, ChatPhoto, UpdateMessageID,
UpdateNewChannelMessage, UpdateNewMessage, UpdateShortSentMessage, UpdateNewChannelMessage, UpdateNewMessage, UpdateShortSentMessage,
PeerUser, InputPeerUser, InputPeerChat, InputPeerChannel, MessageEmpty, PeerUser, InputPeerUser, InputPeerChat, InputPeerChannel, MessageEmpty,
ChatInvite, ChatInviteAlready, PeerChannel ChatInvite, ChatInviteAlready, PeerChannel, Photo
) )
from .tl.types.messages import DialogsSlice from .tl.types.messages import DialogsSlice
from .extensions import markdown from .extensions import markdown
@ -848,7 +848,7 @@ class TelegramClient(TelegramBareClient):
date = datetime.now() date = datetime.now()
media = message media = message
if isinstance(media, MessageMediaPhoto): if isinstance(media, (MessageMediaPhoto, Photo)):
return self._download_photo( return self._download_photo(
media, file, date, progress_callback media, file, date, progress_callback
) )
@ -861,11 +861,15 @@ class TelegramClient(TelegramBareClient):
media, file 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""" """Specialized version of .download_media() for photos"""
# Determine the photo and its largest size # 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] largest_size = photo.sizes[-1]
file_size = largest_size.size file_size = largest_size.size
largest_size = largest_size.location largest_size = largest_size.location