diff --git a/telegram/_bot.py b/telegram/_bot.py index f8b53211ed8..ca3d108451a 100644 --- a/telegram/_bot.py +++ b/telegram/_bot.py @@ -1218,6 +1218,7 @@ async def forward_message( disable_notification: ODVInput[bool] = DEFAULT_NONE, protect_content: ODVInput[bool] = DEFAULT_NONE, message_thread_id: Optional[int] = None, + video_start_timestamp: Optional[int] = None, *, read_timeout: ODVInput[float] = DEFAULT_NONE, write_timeout: ODVInput[float] = DEFAULT_NONE, @@ -1242,6 +1243,10 @@ async def forward_message( original message was sent (or channel username in the format ``@channelusername``). message_id (:obj:`int`): Message identifier in the chat specified in :paramref:`from_chat_id`. + video_start_timestamp (:obj:`int`, optional): New start timestamp for the + forwarded video in the message + + .. versionadded:: NEXT.VERSION disable_notification (:obj:`bool`, optional): |disable_notification| protect_content (:obj:`bool`, optional): |protect_content| @@ -1260,6 +1265,7 @@ async def forward_message( "chat_id": chat_id, "from_chat_id": from_chat_id, "message_id": message_id, + "video_start_timestamp": video_start_timestamp, } return await self._send_message( @@ -1955,6 +1961,8 @@ async def send_video( message_effect_id: Optional[str] = None, allow_paid_broadcast: Optional[bool] = None, show_caption_above_media: Optional[bool] = None, + cover: Optional[FileInput] = None, + start_timestamp: Optional[int] = None, *, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, reply_to_message_id: Optional[int] = None, @@ -2002,6 +2010,13 @@ async def send_video( |time-period-input| width (:obj:`int`, optional): Video width. height (:obj:`int`, optional): Video height. + cover (:term:`file object` | :obj:`bytes` | :class:`pathlib.Path` | :obj:`str`, \ + optional): Cover for the video in the message. |fileinputnopath| + + .. versionadded:: NEXT.VERSION + start_timestamp (:obj:`int`, optional): Start timestamp for the video in the message. + + .. versionadded:: NEXT.VERSION caption (:obj:`str`, optional): Video caption (may also be used when resending videos by file_id), 0-:tg-const:`telegram.constants.MessageLimit.CAPTION_LENGTH` characters after entities parsing. @@ -2088,6 +2103,8 @@ async def send_video( "width": width, "height": height, "supports_streaming": supports_streaming, + "cover": self._parse_file_input(cover, attach=True) if cover else None, + "start_timestamp": start_timestamp, "thumbnail": self._parse_file_input(thumbnail, attach=True) if thumbnail else None, "has_spoiler": has_spoiler, "show_caption_above_media": show_caption_above_media, @@ -7974,6 +7991,7 @@ async def copy_message( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, reply_to_message_id: Optional[int] = None, @@ -7993,6 +8011,10 @@ async def copy_message( from_chat_id (:obj:`int` | :obj:`str`): Unique identifier for the chat where the original message was sent (or channel username in the format ``@channelusername``). message_id (:obj:`int`): Message identifier in the chat specified in from_chat_id. + video_start_timestamp (:obj:`int`, optional): New start timestamp for the + copied video in the message + + .. versionadded:: NEXT.VERSION caption (:obj:`str`, optional): New caption for media, 0-:tg-const:`telegram.constants.MessageLimit.CAPTION_LENGTH` characters after entities parsing. If not specified, the original caption is kept. @@ -8083,6 +8105,7 @@ async def copy_message( "reply_parameters": reply_parameters, "show_caption_above_media": show_caption_above_media, "allow_paid_broadcast": allow_paid_broadcast, + "video_start_timestamp": video_start_timestamp, } result = await self._post( diff --git a/telegram/_callbackquery.py b/telegram/_callbackquery.py index c1ba637546f..99b4ad115b5 100644 --- a/telegram/_callbackquery.py +++ b/telegram/_callbackquery.py @@ -831,6 +831,7 @@ async def copy_message( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, reply_to_message_id: Optional[int] = None, @@ -864,6 +865,7 @@ async def copy_message( chat_id=chat_id, caption=caption, parse_mode=parse_mode, + video_start_timestamp=video_start_timestamp, caption_entities=caption_entities, disable_notification=disable_notification, reply_to_message_id=reply_to_message_id, diff --git a/telegram/_chat.py b/telegram/_chat.py index e615f1b9b6b..f5203a2e1c3 100644 --- a/telegram/_chat.py +++ b/telegram/_chat.py @@ -1940,6 +1940,8 @@ async def send_video( message_effect_id: Optional[str] = None, allow_paid_broadcast: Optional[bool] = None, show_caption_above_media: Optional[bool] = None, + cover: Optional[FileInput] = None, + start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -1978,6 +1980,8 @@ async def send_video( parse_mode=parse_mode, supports_streaming=supports_streaming, thumbnail=thumbnail, + cover=cover, + start_timestamp=start_timestamp, api_kwargs=api_kwargs, allow_sending_without_reply=allow_sending_without_reply, caption_entities=caption_entities, @@ -2199,6 +2203,7 @@ async def send_copy( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -2225,6 +2230,7 @@ async def send_copy( from_chat_id=from_chat_id, message_id=message_id, caption=caption, + video_start_timestamp=video_start_timestamp, parse_mode=parse_mode, caption_entities=caption_entities, disable_notification=disable_notification, @@ -2257,6 +2263,7 @@ async def copy_message( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -2283,6 +2290,7 @@ async def copy_message( chat_id=chat_id, message_id=message_id, caption=caption, + video_start_timestamp=video_start_timestamp, parse_mode=parse_mode, caption_entities=caption_entities, disable_notification=disable_notification, @@ -2398,6 +2406,7 @@ async def forward_from( disable_notification: ODVInput[bool] = DEFAULT_NONE, protect_content: ODVInput[bool] = DEFAULT_NONE, message_thread_id: Optional[int] = None, + video_start_timestamp: Optional[int] = None, *, read_timeout: ODVInput[float] = DEFAULT_NONE, write_timeout: ODVInput[float] = DEFAULT_NONE, @@ -2423,6 +2432,7 @@ async def forward_from( chat_id=self.id, from_chat_id=from_chat_id, message_id=message_id, + video_start_timestamp=video_start_timestamp, disable_notification=disable_notification, read_timeout=read_timeout, write_timeout=write_timeout, @@ -2440,6 +2450,7 @@ async def forward_to( disable_notification: ODVInput[bool] = DEFAULT_NONE, protect_content: ODVInput[bool] = DEFAULT_NONE, message_thread_id: Optional[int] = None, + video_start_timestamp: Optional[int] = None, *, read_timeout: ODVInput[float] = DEFAULT_NONE, write_timeout: ODVInput[float] = DEFAULT_NONE, @@ -2466,6 +2477,7 @@ async def forward_to( from_chat_id=self.id, chat_id=chat_id, message_id=message_id, + video_start_timestamp=video_start_timestamp, disable_notification=disable_notification, read_timeout=read_timeout, write_timeout=write_timeout, diff --git a/telegram/_files/inputmedia.py b/telegram/_files/inputmedia.py index 0ac17c2d55d..912769758e7 100644 --- a/telegram/_files/inputmedia.py +++ b/telegram/_files/inputmedia.py @@ -214,6 +214,13 @@ class InputPaidMediaVideo(InputPaidMedia): Lastly you can pass an existing :class:`telegram.Video` object to send. thumbnail (:term:`file object` | :obj:`bytes` | :class:`pathlib.Path` | :obj:`str`, \ optional): |thumbdocstringnopath| + cover (:term:`file object` | :obj:`bytes` | :class:`pathlib.Path` | :obj:`str`, \ + optional): Cover for the video in the message. |fileinputnopath| + + .. versionchanged:: NEXT.VERSION + start_timestamp (:obj:`int`, optional): Start timestamp for the video in the message + + .. versionchanged:: NEXT.VERSION width (:obj:`int`, optional): Video width. height (:obj:`int`, optional): Video height. duration (:obj:`int`, optional): Video duration in seconds. @@ -225,6 +232,13 @@ class InputPaidMediaVideo(InputPaidMedia): :tg-const:`telegram.constants.InputPaidMediaType.VIDEO`. media (:obj:`str` | :class:`telegram.InputFile`): Video to send. thumbnail (:class:`telegram.InputFile`): Optional. |thumbdocstringbase| + cover (:class:`telegram.InputFile`): Optional. Cover for the video in the message. + |fileinputnopath| + + .. versionchanged:: NEXT.VERSION + start_timestamp (:obj:`int`): Optional. Start timestamp for the video in the message + + .. versionchanged:: NEXT.VERSION width (:obj:`int`): Optional. Video width. height (:obj:`int`): Optional. Video height. duration (:obj:`int`): Optional. Video duration in seconds. @@ -232,7 +246,15 @@ class InputPaidMediaVideo(InputPaidMedia): suitable for streaming. """ - __slots__ = ("duration", "height", "supports_streaming", "thumbnail", "width") + __slots__ = ( + "cover", + "duration", + "height", + "start_timestamp", + "supports_streaming", + "thumbnail", + "width", + ) def __init__( self, @@ -242,6 +264,8 @@ def __init__( height: Optional[int] = None, duration: Optional[int] = None, supports_streaming: Optional[bool] = None, + cover: Optional[FileInput] = None, + start_timestamp: Optional[int] = None, *, api_kwargs: Optional[JSONDict] = None, ): @@ -264,6 +288,10 @@ def __init__( self.height: Optional[int] = height self.duration: Optional[int] = duration self.supports_streaming: Optional[bool] = supports_streaming + self.cover: Optional[Union[InputFile, str]] = ( + parse_file_input(cover, attach=True, local_mode=True) if cover else None + ) + self.start_timestamp: Optional[int] = start_timestamp class InputMediaAnimation(InputMedia): @@ -536,6 +564,13 @@ class InputMediaVideo(InputMedia): optional): |thumbdocstringnopath| .. versionadded:: 20.2 + cover (:term:`file object` | :obj:`bytes` | :class:`pathlib.Path` | :obj:`str`, \ + optional): Cover for the video in the message. |fileinputnopath| + + .. versionchanged:: NEXT.VERSION + start_timestamp (:obj:`int`, optional): Start timestamp for the video in the message + + .. versionchanged:: NEXT.VERSION show_caption_above_media (:obj:`bool`, optional): Pass |show_cap_above_med| .. versionadded:: 21.3 @@ -568,13 +603,22 @@ class InputMediaVideo(InputMedia): show_caption_above_media (:obj:`bool`): Optional. |show_cap_above_med| .. versionadded:: 21.3 + cover (:class:`telegram.InputFile`): Optional. Cover for the video in the message. + |fileinputnopath| + + .. versionchanged:: NEXT.VERSION + start_timestamp (:obj:`int`): Optional. Start timestamp for the video in the message + + .. versionchanged:: NEXT.VERSION """ __slots__ = ( + "cover", "duration", "has_spoiler", "height", "show_caption_above_media", + "start_timestamp", "supports_streaming", "thumbnail", "width", @@ -594,6 +638,8 @@ def __init__( has_spoiler: Optional[bool] = None, thumbnail: Optional[FileInput] = None, show_caption_above_media: Optional[bool] = None, + cover: Optional[FileInput] = None, + start_timestamp: Optional[int] = None, *, api_kwargs: Optional[JSONDict] = None, ): @@ -625,6 +671,10 @@ def __init__( self.supports_streaming: Optional[bool] = supports_streaming self.has_spoiler: Optional[bool] = has_spoiler self.show_caption_above_media: Optional[bool] = show_caption_above_media + self.cover: Optional[Union[InputFile, str]] = ( + parse_file_input(cover, attach=True, local_mode=True) if cover else None + ) + self.start_timestamp: Optional[int] = start_timestamp class InputMediaAudio(InputMedia): diff --git a/telegram/_files/video.py b/telegram/_files/video.py index c5ca6f3b946..37a87e77136 100644 --- a/telegram/_files/video.py +++ b/telegram/_files/video.py @@ -17,12 +17,17 @@ # You should have received a copy of the GNU Lesser Public License # along with this program. If not, see [http://www.gnu.org/licenses/]. """This module contains an object that represents a Telegram Video.""" -from typing import Optional +from collections.abc import Sequence +from typing import TYPE_CHECKING, Optional from telegram._files._basethumbedmedium import _BaseThumbedMedium from telegram._files.photosize import PhotoSize +from telegram._utils.argumentparsing import de_list_optional, parse_sequence_arg from telegram._utils.types import JSONDict +if TYPE_CHECKING: + from telegram import Bot + class Video(_BaseThumbedMedium): """This object represents a video file. @@ -48,6 +53,13 @@ class Video(_BaseThumbedMedium): thumbnail (:class:`telegram.PhotoSize`, optional): Video thumbnail. .. versionadded:: 20.2 + cover (Sequence[:class:`telegram.PhotoSize`], optional): Available sizes of the cover of + the video in the message. + + .. versionadded:: NEXT.VERSION + start_timestamp (:obj:`int`, optional): Timestamp in seconds from which the video + will play in the message + .. versionadded:: NEXT.VERSION Attributes: file_id (:obj:`str`): Identifier for this file, which can be used to download @@ -64,9 +76,24 @@ class Video(_BaseThumbedMedium): thumbnail (:class:`telegram.PhotoSize`): Optional. Video thumbnail. .. versionadded:: 20.2 + cover (tuple[:class:`telegram.PhotoSize`]): Optional, Available sizes of the cover of + the video in the message. + + .. versionadded:: NEXT.VERSION + start_timestamp (:obj:`int`): Optional, Timestamp in seconds from which the video + will play in the message + .. versionadded:: NEXT.VERSION """ - __slots__ = ("duration", "file_name", "height", "mime_type", "width") + __slots__ = ( + "cover", + "duration", + "file_name", + "height", + "mime_type", + "start_timestamp", + "width", + ) def __init__( self, @@ -79,6 +106,8 @@ def __init__( file_size: Optional[int] = None, file_name: Optional[str] = None, thumbnail: Optional[PhotoSize] = None, + cover: Optional[Sequence[PhotoSize]] = None, + start_timestamp: Optional[int] = None, *, api_kwargs: Optional[JSONDict] = None, ): @@ -97,3 +126,14 @@ def __init__( # Optional self.mime_type: Optional[str] = mime_type self.file_name: Optional[str] = file_name + self.cover: Optional[Sequence[PhotoSize]] = parse_sequence_arg(cover) + self.start_timestamp: Optional[int] = start_timestamp + + @classmethod + def de_json(cls, data: JSONDict, bot: Optional["Bot"] = None) -> "Video": + """See :meth:`telegram.TelegramObject.de_json`.""" + data = cls._parse_data(data) + + data["cover"] = de_list_optional(data.get("cover"), PhotoSize, bot) + + return super().de_json(data=data, bot=bot) diff --git a/telegram/_message.py b/telegram/_message.py index e5e5ce7ed4f..d8b6d13db5b 100644 --- a/telegram/_message.py +++ b/telegram/_message.py @@ -2592,6 +2592,8 @@ async def reply_video( message_effect_id: Optional[str] = None, allow_paid_broadcast: Optional[bool] = None, show_caption_above_media: Optional[bool] = None, + cover: Optional[FileInput] = None, + start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -2661,6 +2663,8 @@ async def reply_video( message_thread_id=message_thread_id, has_spoiler=has_spoiler, thumbnail=thumbnail, + cover=cover, + start_timestamp=start_timestamp, business_connection_id=self.business_connection_id, message_effect_id=message_effect_id, allow_paid_broadcast=allow_paid_broadcast, @@ -3506,6 +3510,7 @@ async def forward( disable_notification: ODVInput[bool] = DEFAULT_NONE, protect_content: ODVInput[bool] = DEFAULT_NONE, message_thread_id: Optional[int] = None, + video_start_timestamp: Optional[int] = None, *, read_timeout: ODVInput[float] = DEFAULT_NONE, write_timeout: ODVInput[float] = DEFAULT_NONE, @@ -3540,6 +3545,7 @@ async def forward( chat_id=chat_id, from_chat_id=self.chat_id, message_id=self.message_id, + video_start_timestamp=video_start_timestamp, disable_notification=disable_notification, protect_content=protect_content, message_thread_id=message_thread_id, @@ -3563,6 +3569,7 @@ async def copy( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -3593,6 +3600,7 @@ async def copy( from_chat_id=self.chat_id, message_id=self.message_id, caption=caption, + video_start_timestamp=video_start_timestamp, parse_mode=parse_mode, caption_entities=caption_entities, disable_notification=disable_notification, @@ -3625,6 +3633,7 @@ async def reply_copy( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -3675,6 +3684,7 @@ async def reply_copy( from_chat_id=from_chat_id, message_id=message_id, caption=caption, + video_start_timestamp=video_start_timestamp, parse_mode=parse_mode, caption_entities=caption_entities, disable_notification=disable_notification, diff --git a/telegram/_user.py b/telegram/_user.py index d28892e4135..bff13077cf7 100644 --- a/telegram/_user.py +++ b/telegram/_user.py @@ -1328,6 +1328,8 @@ async def send_video( message_effect_id: Optional[str] = None, allow_paid_broadcast: Optional[bool] = None, show_caption_above_media: Optional[bool] = None, + cover: Optional[FileInput] = None, + start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -1369,6 +1371,8 @@ async def send_video( parse_mode=parse_mode, supports_streaming=supports_streaming, thumbnail=thumbnail, + cover=cover, + start_timestamp=start_timestamp, api_kwargs=api_kwargs, allow_sending_without_reply=allow_sending_without_reply, caption_entities=caption_entities, @@ -1708,6 +1712,7 @@ async def send_copy( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -1735,6 +1740,7 @@ async def send_copy( from_chat_id=from_chat_id, message_id=message_id, caption=caption, + video_start_timestamp=video_start_timestamp, parse_mode=parse_mode, caption_entities=caption_entities, disable_notification=disable_notification, @@ -1767,6 +1773,7 @@ async def copy_message( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -1794,6 +1801,7 @@ async def copy_message( chat_id=chat_id, message_id=message_id, caption=caption, + video_start_timestamp=video_start_timestamp, parse_mode=parse_mode, caption_entities=caption_entities, disable_notification=disable_notification, @@ -1909,6 +1917,7 @@ async def forward_from( disable_notification: ODVInput[bool] = DEFAULT_NONE, protect_content: ODVInput[bool] = DEFAULT_NONE, message_thread_id: Optional[int] = None, + video_start_timestamp: Optional[int] = None, *, read_timeout: ODVInput[float] = DEFAULT_NONE, write_timeout: ODVInput[float] = DEFAULT_NONE, @@ -1934,6 +1943,7 @@ async def forward_from( chat_id=self.id, from_chat_id=from_chat_id, message_id=message_id, + video_start_timestamp=video_start_timestamp, disable_notification=disable_notification, read_timeout=read_timeout, write_timeout=write_timeout, @@ -1951,6 +1961,7 @@ async def forward_to( disable_notification: ODVInput[bool] = DEFAULT_NONE, protect_content: ODVInput[bool] = DEFAULT_NONE, message_thread_id: Optional[int] = None, + video_start_timestamp: Optional[int] = None, *, read_timeout: ODVInput[float] = DEFAULT_NONE, write_timeout: ODVInput[float] = DEFAULT_NONE, @@ -1977,6 +1988,7 @@ async def forward_to( from_chat_id=self.id, chat_id=chat_id, message_id=message_id, + video_start_timestamp=video_start_timestamp, disable_notification=disable_notification, read_timeout=read_timeout, write_timeout=write_timeout, diff --git a/telegram/ext/_extbot.py b/telegram/ext/_extbot.py index 9b527ef2b05..b35f1b98c9e 100644 --- a/telegram/ext/_extbot.py +++ b/telegram/ext/_extbot.py @@ -815,6 +815,7 @@ async def copy_message( reply_parameters: Optional["ReplyParameters"] = None, show_caption_above_media: Optional[bool] = None, allow_paid_broadcast: Optional[bool] = None, + video_start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -831,6 +832,7 @@ async def copy_message( from_chat_id=from_chat_id, message_id=message_id, caption=caption, + video_start_timestamp=video_start_timestamp, parse_mode=parse_mode, caption_entities=caption_entities, disable_notification=disable_notification, @@ -1752,6 +1754,7 @@ async def forward_message( disable_notification: ODVInput[bool] = DEFAULT_NONE, protect_content: ODVInput[bool] = DEFAULT_NONE, message_thread_id: Optional[int] = None, + video_start_timestamp: Optional[int] = None, *, read_timeout: ODVInput[float] = DEFAULT_NONE, write_timeout: ODVInput[float] = DEFAULT_NONE, @@ -1764,6 +1767,7 @@ async def forward_message( chat_id=chat_id, from_chat_id=from_chat_id, message_id=message_id, + video_start_timestamp=video_start_timestamp, disable_notification=disable_notification, protect_content=protect_content, message_thread_id=message_thread_id, @@ -3240,6 +3244,8 @@ async def send_video( message_effect_id: Optional[str] = None, allow_paid_broadcast: Optional[bool] = None, show_caption_above_media: Optional[bool] = None, + cover: Optional[FileInput] = None, + start_timestamp: Optional[int] = None, *, reply_to_message_id: Optional[int] = None, allow_sending_without_reply: ODVInput[bool] = DEFAULT_NONE, @@ -3270,6 +3276,8 @@ async def send_video( business_connection_id=business_connection_id, has_spoiler=has_spoiler, thumbnail=thumbnail, + cover=cover, + start_timestamp=start_timestamp, filename=filename, reply_parameters=reply_parameters, read_timeout=read_timeout, diff --git a/tests/_files/test_inputmedia.py b/tests/_files/test_inputmedia.py index 5aef4e62da4..b362411cbd8 100644 --- a/tests/_files/test_inputmedia.py +++ b/tests/_files/test_inputmedia.py @@ -58,6 +58,8 @@ def input_media_video(class_thumb_file): parse_mode=InputMediaVideoTestBase.parse_mode, caption_entities=InputMediaVideoTestBase.caption_entities, thumbnail=class_thumb_file, + cover=class_thumb_file, + start_timestamp=InputMediaVideoTestBase.start_timestamp, supports_streaming=InputMediaVideoTestBase.supports_streaming, has_spoiler=InputMediaVideoTestBase.has_spoiler, show_caption_above_media=InputMediaVideoTestBase.show_caption_above_media, @@ -130,6 +132,8 @@ def input_paid_media_video(class_thumb_file): return InputPaidMediaVideo( media=InputMediaVideoTestBase.media, thumbnail=class_thumb_file, + cover=class_thumb_file, + start_timestamp=InputMediaVideoTestBase.start_timestamp, width=InputMediaVideoTestBase.width, height=InputMediaVideoTestBase.height, duration=InputMediaVideoTestBase.duration, @@ -144,6 +148,7 @@ class InputMediaVideoTestBase: width = 3 height = 4 duration = 5 + start_timestamp = 3 parse_mode = "HTML" supports_streaming = True caption_entities = [MessageEntity(MessageEntity.BOLD, 0, 2)] @@ -169,6 +174,8 @@ def test_expected_values(self, input_media_video): assert input_media_video.caption_entities == tuple(self.caption_entities) assert input_media_video.supports_streaming == self.supports_streaming assert isinstance(input_media_video.thumbnail, InputFile) + assert isinstance(input_media_video.cover, InputFile) + assert input_media_video.start_timestamp == self.start_timestamp assert input_media_video.has_spoiler == self.has_spoiler assert input_media_video.show_caption_above_media == self.show_caption_above_media @@ -194,6 +201,8 @@ def test_to_dict(self, input_media_video): input_media_video_dict["show_caption_above_media"] == input_media_video.show_caption_above_media ) + assert input_media_video_dict["cover"] == input_media_video.cover + assert input_media_video_dict["start_timestamp"] == input_media_video.start_timestamp def test_with_video(self, video): # fixture found in test_video @@ -214,10 +223,13 @@ def test_with_video_file(self, video_file): def test_with_local_files(self): input_media_video = InputMediaVideo( - data_file("telegram.mp4"), thumbnail=data_file("telegram.jpg") + data_file("telegram.mp4"), + thumbnail=data_file("telegram.jpg"), + cover=data_file("telegram.jpg"), ) assert input_media_video.media == data_file("telegram.mp4").as_uri() assert input_media_video.thumbnail == data_file("telegram.jpg").as_uri() + assert input_media_video.cover == data_file("telegram.jpg").as_uri() def test_type_enum_conversion(self): # Since we have a lot of different test classes for all the input media types, we test this @@ -565,6 +577,8 @@ def test_expected_values(self, input_paid_media_video): assert input_paid_media_video.duration == self.duration assert input_paid_media_video.supports_streaming == self.supports_streaming assert isinstance(input_paid_media_video.thumbnail, InputFile) + assert isinstance(input_paid_media_video.cover, InputFile) + assert input_paid_media_video.start_timestamp == self.start_timestamp def test_to_dict(self, input_paid_media_video): input_paid_media_video_dict = input_paid_media_video.to_dict() @@ -578,6 +592,11 @@ def test_to_dict(self, input_paid_media_video): == input_paid_media_video.supports_streaming ) assert input_paid_media_video_dict["thumbnail"] == input_paid_media_video.thumbnail + assert input_paid_media_video_dict["cover"] == input_paid_media_video.cover + assert ( + input_paid_media_video_dict["start_timestamp"] + == input_paid_media_video.start_timestamp + ) def test_with_video(self, video): # fixture found in test_video @@ -596,10 +615,13 @@ def test_with_video_file(self, video_file): def test_with_local_files(self): input_paid_media_video = InputPaidMediaVideo( - data_file("telegram.mp4"), thumbnail=data_file("telegram.jpg") + data_file("telegram.mp4"), + thumbnail=data_file("telegram.jpg"), + cover=data_file("telegram.jpg"), ) assert input_paid_media_video.media == data_file("telegram.mp4").as_uri() assert input_paid_media_video.thumbnail == data_file("telegram.jpg").as_uri() + assert input_paid_media_video.cover == data_file("telegram.jpg").as_uri() @pytest.fixture(scope="module") diff --git a/tests/_files/test_video.py b/tests/_files/test_video.py index 65d7ee8c354..d4d87122576 100644 --- a/tests/_files/test_video.py +++ b/tests/_files/test_video.py @@ -46,6 +46,8 @@ class VideoTestBase: mime_type = "video/mp4" supports_streaming = True file_name = "telegram.mp4" + start_timestamp = 3 + cover = (PhotoSize("file_id", "unique_id", 640, 360, file_size=0),) thumb_width = 180 thumb_height = 320 thumb_file_size = 1767 @@ -92,6 +94,8 @@ def test_de_json(self, offline_bot): "mime_type": self.mime_type, "file_size": self.file_size, "file_name": self.file_name, + "start_timestamp": self.start_timestamp, + "cover": [photo_size.to_dict() for photo_size in self.cover], } json_video = Video.de_json(json_dict, offline_bot) assert json_video.api_kwargs == {} @@ -104,6 +108,8 @@ def test_de_json(self, offline_bot): assert json_video.mime_type == self.mime_type assert json_video.file_size == self.file_size assert json_video.file_name == self.file_name + assert json_video.start_timestamp == self.start_timestamp + assert json_video.cover == self.cover def test_to_dict(self, video): video_dict = video.to_dict() @@ -223,7 +229,9 @@ async def make_assertion(url, request_data: RequestData, *args, **kwargs): class TestVideoWithRequest(VideoTestBase): @pytest.mark.parametrize("duration", [dtm.timedelta(seconds=5), 5]) - async def test_send_all_args(self, bot, chat_id, video_file, video, thumb_file, duration): + async def test_send_all_args( + self, bot, chat_id, video_file, video, thumb_file, photo_file, duration + ): message = await bot.send_video( chat_id, video_file, @@ -236,6 +244,8 @@ async def test_send_all_args(self, bot, chat_id, video_file, video, thumb_file, height=video.height, parse_mode="Markdown", thumbnail=thumb_file, + cover=photo_file, + start_timestamp=self.start_timestamp, has_spoiler=True, show_caption_above_media=True, ) @@ -256,6 +266,11 @@ async def test_send_all_args(self, bot, chat_id, video_file, video, thumb_file, assert message.video.thumbnail.width == self.thumb_width assert message.video.thumbnail.height == self.thumb_height + assert message.video.start_timestamp == self.start_timestamp + + assert isinstance(message.video.cover, tuple) + assert isinstance(message.video.cover[0], PhotoSize) + assert message.video.file_name == self.file_name assert message.has_protected_content assert message.has_media_spoiler diff --git a/tests/test_bot.py b/tests/test_bot.py index 2c9ba6abed0..29474ef2bed 100644 --- a/tests/test_bot.py +++ b/tests/test_bot.py @@ -1617,6 +1617,7 @@ async def post(url, request_data: RequestData, *args, **kwargs): == [MessageEntity(MessageEntity.BOLD, 0, 4).to_dict()], data["protect_content"] is True, data["message_thread_id"] == 1, + data["video_start_timestamp"] == 999, ] ): pytest.fail("I got wrong parameters in post") @@ -1628,6 +1629,7 @@ async def post(url, request_data: RequestData, *args, **kwargs): from_chat_id=chat_id, message_id=media_message.message_id, caption=caption, + video_start_timestamp=999, caption_entities=[MessageEntity(MessageEntity.BOLD, 0, 4)], parse_mode=ParseMode.HTML, reply_to_message_id=media_message.message_id,