Fix conditions in handling show_caption_above_media

TelegramPlayGround · Nov 11, 2024 · 70b6f40 · 70b6f40
1 parent 9c0209a
commit 70b6f40
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pyrogram/methods/messages/edit_inline_media.py b/pyrogram/methods/messages/edit_inline_media.py
@@ -99,6 +99,7 @@ async def edit_inline_media(
             filename_attribute = []
 
         if isinstance(media, types.InputMediaPhoto):
+            show_caption_above_media.append(media.show_caption_above_media)
             if is_uploaded_file:
                 media = raw.types.InputMediaUploadedPhoto(
                     file=await self.save_file(media.media),
@@ -111,8 +112,8 @@ async def edit_inline_media(
                 )
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.PHOTO, has_spoiler=media.has_spoiler)
-            show_caption_above_media.append(media.show_caption_above_media)
         elif isinstance(media, types.InputMediaVideo):
+            show_caption_above_media.append(media.show_caption_above_media)
             if is_uploaded_file:
                 media = raw.types.InputMediaUploadedDocument(
                     mime_type=(None if is_bytes_io else self.guess_mime_type(media.media)) or "video/mp4",
@@ -135,7 +136,6 @@ async def edit_inline_media(
                 )
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.VIDEO, has_spoiler=media.has_spoiler)
-            show_caption_above_media.append(media.show_caption_above_media)
         elif isinstance(media, types.InputMediaAudio):
             if is_uploaded_file:
                 media = raw.types.InputMediaUploadedDocument(
@@ -157,6 +157,7 @@ async def edit_inline_media(
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.AUDIO)
         elif isinstance(media, types.InputMediaAnimation):
+            show_caption_above_media.append(media.show_caption_above_media)
             if is_uploaded_file:
                 media = raw.types.InputMediaUploadedDocument(
                     mime_type=(None if is_bytes_io else self.guess_mime_type(media.media)) or "video/mp4",
@@ -182,7 +183,6 @@ async def edit_inline_media(
                 )
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.ANIMATION)
-            show_caption_above_media.append(media.show_caption_above_media)
         elif isinstance(media, types.InputMediaDocument):
             if is_uploaded_file:
                 media = raw.types.InputMediaUploadedDocument(

diff --git a/pyrogram/methods/messages/edit_message_media.py b/pyrogram/methods/messages/edit_message_media.py
@@ -123,6 +123,7 @@ async def edit_message_media(
             filename_attribute = []
 
         if isinstance(media, types.InputMediaPhoto):
+            show_caption_above_media.append(media.show_caption_above_media)
             if is_uploaded_file:
                 uploaded_media = await self.invoke(
                     raw.functions.messages.UploadMedia(
@@ -150,8 +151,8 @@ async def edit_message_media(
                 )
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.PHOTO, has_spoiler=media.has_spoiler)
-            show_caption_above_media.append(media.show_caption_above_media)
         elif isinstance(media, types.InputMediaVideo):
+            show_caption_above_media.append(media.show_caption_above_media)
             if is_uploaded_file:
                 uploaded_media = await self.invoke(
                     raw.functions.messages.UploadMedia(
@@ -191,7 +192,6 @@ async def edit_message_media(
                 )
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.VIDEO, has_spoiler=media.has_spoiler)
-            show_caption_above_media.append(media.show_caption_above_media)
         elif isinstance(media, types.InputMediaAudio):
             if is_uploaded_file:
                 media = await self.invoke(
@@ -227,6 +227,7 @@ async def edit_message_media(
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.AUDIO)
         elif isinstance(media, types.InputMediaAnimation):
+            show_caption_above_media.append(media.show_caption_above_media)
             if is_uploaded_file:
                 uploaded_media = await self.invoke(
                     raw.functions.messages.UploadMedia(
@@ -265,7 +266,6 @@ async def edit_message_media(
                 )
             else:
                 media = utils.get_input_media_from_file_id(media.media, FileType.ANIMATION, has_spoiler=media.has_spoiler)
-            show_caption_above_media.append(media.show_caption_above_media)
         elif isinstance(media, types.InputMediaDocument):
             if is_uploaded_file:
                 media = await self.invoke(