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

Support autocast of polls into input media when possible

Browse Source 
Closes #1516.
This commit is contained in:
Lonami Exo 2020-08-07 16:03:50 +02:00
parent 241c6c4ac8
commit 1d71cdc9e0
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
telethon/utils.py
View File

@ -525,6 +525,27 @@ def get_input_media(
if isinstance(media, types.Message):
return get_input_media(media.media, is_photo=is_photo)
if isinstance(media, types.MessageMediaPoll):
if media.poll.quiz:
if not media.results.results:
# A quiz has correct answers, which we don't know until answered.
# If the quiz hasn't been answered we can't reconstruct it properly.
raise TypeError('Cannot cast unanswered quiz to any kind of InputMedia.')
correct_answers = [r.option for r in media.results.results if r.correct]
else:
correct_answers = None
return types.InputMediaPoll(
poll=media.poll,
correct_answers=correct_answers,
solution=media.results.solution,
solution_entities=media.results.solution_entities,
)
if isinstance(media, types.Poll):
return types.InputMediaPoll(media)
_raise_cast_fail(media, 'InputMedia')