Add cover and start_timestamp in send_video and InputPaidMediaVideo

Author: SpEcHiDe

docs/source/releases/changes-in-this-fork.rst

@@ -26,6 +26,7 @@ Changes in this Fork
 | Scheme layer used: 198 |
 +------------------------+
 
+- Added the ``cover`` and ``start_timestamp`` parameters in :meth:`~pyrogram.Client.send_video` and :obj:`~pyrogram.types.InputPaidMediaVideo`.
 - Added the ``new_video_start_timestamp`` and renamed the ``send_copy`` and ``remove_caption`` parameters in :meth:`~pyrogram.Client.forward_messages` and :meth:`~pyrogram.types.Message.forward`.
 - Added the ``gift_count`` to the :obj:`~pyrogram.types.Chat`.
 - Added the :meth:`~pyrogram.Client.get_similar_bots`.

pyrogram/methods/messages/send_paid_media.py

@@ -179,9 +179,7 @@ async def send_paid_media(
                             file_reference=media.photo.file_reference
                         )
                     )
-            elif (
-                isinstance(i, types.InputPaidMediaVideo)
-            ):
+            elif isinstance(i, types.InputPaidMediaVideo):
                 if isinstance(i.media, str):
                     if os.path.isfile(i.media):
                         attributes = [
@@ -201,7 +199,9 @@ async def send_paid_media(
                                     thumb=await self.save_file(i.thumbnail),
                                     mime_type=self.guess_mime_type(i.media) or "video/mp4",
                                     nosound_video=True,
-                                    attributes=attributes
+                                    attributes=attributes,
+                                    video_cover=await self.save_file(i.cover),
+                                    video_timestamp=i.start_timestamp
                                 )
                             )
                         )
@@ -218,7 +218,9 @@ async def send_paid_media(
                             raw.functions.messages.UploadMedia(
                                 peer=await self.resolve_peer(chat_id),
                                 media=raw.types.InputMediaDocumentExternal(
-                                    url=i.media
+                                    url=i.media,
+                                    video_cover=await self.save_file(i.cover),
+                                    video_timestamp=i.start_timestamp
                                 )
                             )
                         )
@@ -248,7 +250,9 @@ async def send_paid_media(
                                         h=i.height
                                     ),
                                     raw.types.DocumentAttributeFilename(file_name=getattr(i.media, "name", "video.mp4"))
-                                ]
+                                ],
+                                video_cover=await self.save_file(i.cover),
+                                video_timestamp=i.start_timestamp
                             )
                         )
                     )

pyrogram/methods/messages/send_video.py

@@ -65,6 +65,8 @@ async def send_video(
         view_once: bool = None,
         file_name: str = None,
         mime_type: str = None,
+        cover: Optional[Union[str, "io.BytesIO"]] = None,
+        start_timestamp: int = None,
         schedule_date: datetime = None,
         reply_to_message_id: int = None,
         progress: Callable = None,
@@ -173,6 +175,12 @@ async def send_video(
             mime_type (``str``, *optional*):
                 no docs!
 
+            cover (``str`` | :obj:`io.BytesIO`, *optional*):
+                Cover of the video; pass None to skip cover uploading.
+            
+            start_timestamp (``int``, *optional*):
+                Timestamp from which the video playing must start, in seconds.
+
             schedule_date (:py:obj:`~datetime.datetime`, *optional*):
                 Date when the message will be automatically sent.
 
@@ -260,13 +268,17 @@ async def progress(current, total):
                                 h=height
                             ),
                             raw.types.DocumentAttributeFilename(file_name=file_name or os.path.basename(video))
-                        ]
+                        ],
+                        video_cover=await self.save_file(cover) if cover else None,
+                        video_timestamp=start_timestamp
                     )
                 elif re.match("^https?://", video):
                     media = raw.types.InputMediaDocumentExternal(
                         url=video,
                         ttl_seconds=ttl_seconds,
-                        spoiler=has_spoiler
+                        spoiler=has_spoiler,
+                        video_cover=await self.save_file(cover) if cover else None,
+                        video_timestamp=start_timestamp
                     )
                 else:
                     media = utils.get_input_media_from_file_id(
@@ -292,7 +304,9 @@ async def progress(current, total):
                             h=height
                         ),
                         raw.types.DocumentAttributeFilename(file_name=file_name or video.name)
-                    ]
+                    ],
+                    video_cover=await self.save_file(cover) if cover else None,
+                    video_timestamp=start_timestamp
                 )
 
             reply_to = await utils._get_reply_message_parameters(

pyrogram/types/input_paid_media/input_paid_media_video.py

@@ -54,6 +54,12 @@ class InputPaidMediaVideo(InputPaidMedia):
         supports_streaming (``bool``, *optional*):
             Pass True, if the uploaded video is suitable for streaming.
 
+        cover (``str`` | :obj:`io.BytesIO`, *optional*):
+            Cover of the video; pass None to skip cover uploading.
+        
+        start_timestamp (``int``, *optional*):
+            Timestamp from which the video playing must start, in seconds.
+
     """
 
     def __init__(
@@ -63,7 +69,9 @@ def __init__(
         width: int = 0,
         height: int = 0,
         duration: int = 0,
-        supports_streaming: bool = True
+        supports_streaming: bool = True,
+        cover: Optional[Union[str, "io.BytesIO"]] = None,
+        start_timestamp: int = None
     ):
         super().__init__(media)
 
@@ -72,3 +80,5 @@ def __init__(
         self.height = height
         self.duration = duration
         self.supports_streaming = supports_streaming
+        self.cover = cover
+        self.start_timestamp = start_timestamp

pyrogram/types/input_paid_media/paid_media.py

@@ -91,5 +91,12 @@ def _parse(
                             video = types.Video._parse(client, doc, video_attributes, file_name, ttl_seconds)
 
                             return types.PaidMediaVideo(
-                                video=video
+                                video=video,
+                                cover=types.Photo._parse(
+                                    client,
+                                    media.video_cover,
+                                    media.ttl_seconds,
+                                    media.spoiler
+                                ) if media.video_cover else None,
+                                start_timestamp=media.video_timestamp
                             )

pyrogram/types/input_paid_media/paid_media_video.py

@@ -16,6 +16,8 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+from typing import Optional
+
 import pyrogram
 from pyrogram import types
 
@@ -30,13 +32,23 @@ class PaidMediaVideo(PaidMedia):
         video (:obj:`~pyrogram.types.Video`):
             The video.
 
+        cover (:obj:`~pyrogram.types.Photo`, *optional*):
+            Cover of the video.
+        
+        start_timestamp (``int``, *optional*):
+            Timestamp from which the video playing must start, in seconds.
+
     """
 
     def __init__(
         self,
         *,
-        video: "types.Video" = None
+        video: "types.Video" = None,
+        cover: Optional["types.Photo"] = None,
+        start_timestamp: Optional[int] = None
     ):
         super().__init__()
 
         self.video = video
+        self.cover = cover
+        self.start_timestamp = start_timestamp