From 6f8dde7ad0de5eb64a7a0b3359786ecdc0bc1d85 Mon Sep 17 00:00:00 2001 From: wulan17 Date: Mon, 1 Jul 2024 21:19:43 +0700 Subject: [PATCH] pyrofork: fix ExtendedMediaPreview parser Signed-off-by: wulan17 --- pyrogram/types/messages_and_media/extended_media_preview.py | 6 +++--- pyrogram/types/messages_and_media/paid_media.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/pyrogram/types/messages_and_media/extended_media_preview.py b/pyrogram/types/messages_and_media/extended_media_preview.py index 201e90a5..d1f66cb4 100644 --- a/pyrogram/types/messages_and_media/extended_media_preview.py +++ b/pyrogram/types/messages_and_media/extended_media_preview.py @@ -31,7 +31,7 @@ class ExtendedMediaPreview(Object): height (``int``, *optional*): Media Height. - thumb (:obj:`~pyrogram.types.Thumbnail`, *optional*): + thumb (:obj:`~pyrogram.types.StrippedThumbnail`, *optional*): Media Thumbnail. video_duration (``int``, *optional*): @@ -53,10 +53,10 @@ class ExtendedMediaPreview(Object): self.video_duration = video_duration @staticmethod - async def _parse(client, media: "raw.types.MessageExtendedMediaPreview") -> "ExtendedMediaPreview": + def _parse(client, media: "raw.types.MessageExtendedMediaPreview") -> "ExtendedMediaPreview": thumb = None if media.thumb: - thumb = await types.Thumbnail._parse(client, media.thumb) + thumb = types.StrippedThumbnail._parse(client, media.thumb) return ExtendedMediaPreview( width=media.w, diff --git a/pyrogram/types/messages_and_media/paid_media.py b/pyrogram/types/messages_and_media/paid_media.py index b5dabb2a..aa7ccabd 100644 --- a/pyrogram/types/messages_and_media/paid_media.py +++ b/pyrogram/types/messages_and_media/paid_media.py @@ -55,7 +55,9 @@ class PaidMedia(Object): def _parse(client, media: "raw.types.MessageMediaPaidMedia") -> "PaidMedia": extended_media = [] for m in media.extended_media: - if isinstance(m.media, raw.types.MessageMediaPhoto): + if isinstance(m, raw.types.MessageExtendedMediaPreview): + extended_media.append(types.ExtendedMediaPreview._parse(client, m)) + elif isinstance(m.media, raw.types.MessageMediaPhoto): extended_media.append(types.Photo._parse(client, m.media.photo, m.media.ttl_seconds)) elif isinstance(m.media, raw.types.MessageMediaDocument): attributes = {type(i): i for i in m.media.document.attributes} @@ -70,8 +72,6 @@ class PaidMedia(Object): else: video_attributes = attributes[raw.types.DocumentAttributeVideo] extended_media.append(types.Video._parse(client, m.media.document, video_attributes, file_name, m.media.ttl_seconds)) - elif isinstance(m, raw.types.MessageExtendedMediaPreview): - extended_media.append(types.ExtendedMediaPreview._parse(client, m)) return PaidMedia( stars_amount=media.stars_amount, extended_media=extended_media