Skip to content

Commit 8fccf91

Browse files
Add checklist to ExternalReplyInfo
1 parent a40a58f commit 8fccf91

1 file changed

Lines changed: 12 additions & 2 deletions

File tree

pyrogram/types/messages_and_media/external_reply_info.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,9 @@ class ExternalReplyInfo(Object):
8080
has_media_spoiler (``bool``, *optional*):
8181
True, if the message media is covered by a spoiler animation.
8282
83+
checklist (:obj:`~pyrogram.types.Checklist`, *optional*):
84+
Message is a checklist.
85+
8386
contact (:obj:`~pyrogram.types.Contact`, *optional*):
8487
Message is a shared contact, information about the contact.
8588
@@ -129,6 +132,7 @@ def __init__(
129132
video_note: Optional["types.VideoNote"] = None,
130133
voice: Optional["types.Voice"] = None,
131134
has_media_spoiler: Optional[bool] = None,
135+
checklist: Optional["types.Checklist"] = None,
132136
contact: Optional["types.Contact"] = None,
133137
dice: Optional["types.Dice"] = None,
134138
game: Optional["types.Game"] = None,
@@ -157,6 +161,7 @@ def __init__(
157161
self.video_note = video_note
158162
self.voice = voice
159163
self.has_media_spoiler = has_media_spoiler
164+
self.checklist = checklist
160165
self.contact = contact
161166
self.dice = dice
162167
self.game = game
@@ -189,7 +194,9 @@ async def _parse(
189194
story = None
190195
video = None
191196
video_note = None
197+
has_media_spoiler = None
192198
voice = None
199+
checklist = None
193200
contact = None
194201
dice = None
195202
game = None
@@ -202,7 +209,6 @@ async def _parse(
202209

203210
media = reply.reply_media
204211
media_type = None
205-
has_media_spoiler = None
206212

207213
if media:
208214
if isinstance(media, raw.types.MessageMediaPhoto):
@@ -284,6 +290,9 @@ async def _parse(
284290
elif isinstance(media, raw.types.MessageMediaPaidMedia):
285291
paid_media = types.PaidMediaInfo._parse(client, media)
286292
media_type = enums.MessageMediaType.PAID_MEDIA
293+
elif isinstance(media, raw.types.MessageMediaToDo):
294+
checklist = types.Checklist._parse(client, media, users)
295+
media_type = enums.MessageMediaType.CHECKLIST
287296
else:
288297
media = None
289298

@@ -320,5 +329,6 @@ async def _parse(
320329
invoice=invoice,
321330
location=location,
322331
poll=poll,
323-
venue=venue
332+
venue=venue,
333+
checklist=checklist
324334
)

0 commit comments

Comments
 (0)