yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-08 16:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

pyrofork: fix ExtendedMediaPreview parser

Browse source 
Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
wulan17 2024-07-01 21:19:43 +07:00
parent 240d3bfde5
commit 6f8dde7ad0
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pyrogram/types/messages_and_media/extended_media_preview.py
View file

@ -31,7 +31,7 @@ class ExtendedMediaPreview(Object):
height (``int``, *optional*): height (``int``, *optional*):
Media Height. Media Height.
thumb (:obj:`~pyrogram.types.Thumbnail`, *optional*): thumb (:obj:`~pyrogram.types.StrippedThumbnail`, *optional*):
Media Thumbnail. Media Thumbnail.
video_duration (``int``, *optional*): video_duration (``int``, *optional*):
@ -53,10 +53,10 @@ class ExtendedMediaPreview(Object):
self.video_duration = video_duration self.video_duration = video_duration
@staticmethod @staticmethod
async def _parse(client, media: "raw.types.MessageExtendedMediaPreview") -> "ExtendedMediaPreview": def _parse(client, media: "raw.types.MessageExtendedMediaPreview") -> "ExtendedMediaPreview":
thumb = None thumb = None
if media.thumb: if media.thumb:
thumb = await types.Thumbnail._parse(client, media.thumb) thumb = types.StrippedThumbnail._parse(client, media.thumb)
return ExtendedMediaPreview( return ExtendedMediaPreview(
width=media.w, width=media.w,

6
pyrogram/types/messages_and_media/paid_media.py
View file

@ -55,7 +55,9 @@ class PaidMedia(Object):
def _parse(client, media: "raw.types.MessageMediaPaidMedia") -> "PaidMedia": def _parse(client, media: "raw.types.MessageMediaPaidMedia") -> "PaidMedia":
extended_media = [] extended_media = []
for m in media.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)) extended_media.append(types.Photo._parse(client, m.media.photo, m.media.ttl_seconds))
elif isinstance(m.media, raw.types.MessageMediaDocument): elif isinstance(m.media, raw.types.MessageMediaDocument):
attributes = {type(i): i for i in m.media.document.attributes} attributes = {type(i): i for i in m.media.document.attributes}
@ -70,8 +72,6 @@ class PaidMedia(Object):
else: else:
video_attributes = attributes[raw.types.DocumentAttributeVideo] video_attributes = attributes[raw.types.DocumentAttributeVideo]
extended_media.append(types.Video._parse(client, m.media.document, video_attributes, file_name, m.media.ttl_seconds)) 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( return PaidMedia(
stars_amount=media.stars_amount, stars_amount=media.stars_amount,
extended_media=extended_media extended_media=extended_media