From 8d652c35a17cb698c7ff262f71095467454cbca2 Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Tue, 27 Mar 2018 11:22:31 +0200 Subject: [PATCH] Add missing Photo/Document cases to get_input_media --- telethon/utils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/telethon/utils.py b/telethon/utils.py index faa1537a..2427ce13 100644 --- a/telethon/utils.py +++ b/telethon/utils.py @@ -261,12 +261,22 @@ def get_input_media(media, is_photo=False): ttl_seconds=media.ttl_seconds ) + if isinstance(media, (Photo, photos.Photo, PhotoEmpty)): + return InputMediaPhoto( + id=get_input_photo(media) + ) + if isinstance(media, MessageMediaDocument): return InputMediaDocument( id=get_input_document(media.document), ttl_seconds=media.ttl_seconds ) + if isinstance(media, (Document, DocumentEmpty)): + return InputMediaDocument( + id=get_input_document(media) + ) + if isinstance(media, FileLocation): if is_photo: return InputMediaUploadedPhoto(file=media)