Merge pull request pyrogram#556 from drizzt/video-ttl_seconds

delivrance · web-flow · commit 1835b62a4077 · 2020-12-11T18:34:49.000+01:00
Add ttl_seconds support for send_video and reply_video
diff --git a/pyrogram/methods/messages/send_video.py b/pyrogram/methods/messages/send_video.py
@@ -37,6 +37,7 @@ async def send_video(
         caption: str = "",
         parse_mode: Union[str, None] = object,
         caption_entities: List["types.MessageEntity"] = None,
+        ttl_seconds: int = None,
         duration: int = 0,
         width: int = 0,
         height: int = 0,
@@ -83,6 +84,11 @@ async def send_video(
             caption_entities (List of :obj:`~pyrogram.types.MessageEntity`):
                 List of special entities that appear in the caption, which can be specified instead of __parse_mode__.
 
+            ttl_seconds (``int``, *optional*):
+                Self-Destruct Timer.
+                If you set a timer, the video will self-destruct in *ttl_seconds*
+                seconds after it was viewed.
+
             duration (``int``, *optional*):
                 Duration of sent video in seconds.
 
@@ -155,6 +161,9 @@ async def send_video(
                 # Add caption to the video
                 app.send_video("me", "video.mp4", caption="recording")
 
+                # Send self-destructing video
+                app.send_photo("me", "video.mp4", ttl_seconds=10)
+
                 # Keep track of the progress while uploading
                 def progress(current, total):
                     print(f"{current * 100 / total:.1f}%")
@@ -171,6 +180,7 @@ def progress(current, total):
                     media = raw.types.InputMediaUploadedDocument(
                         mime_type=self.guess_mime_type(video) or "video/mp4",
                         file=file,
+                        ttl_seconds=ttl_seconds,
                         thumb=thumb,
                         attributes=[
                             raw.types.DocumentAttributeVideo(
@@ -184,7 +194,8 @@ def progress(current, total):
                     )
                 elif re.match("^https?://", video):
                     media = raw.types.InputMediaDocumentExternal(
-                        url=video
+                        url=video,
+                        ttl_seconds=ttl_seconds
                     )
                 else:
                     media = utils.get_input_media_from_file_id(video, FileType.VIDEO)
@@ -194,6 +205,7 @@ def progress(current, total):
                 media = raw.types.InputMediaUploadedDocument(
                     mime_type=self.guess_mime_type(video.name) or "video/mp4",
                     file=file,
+                    ttl_seconds=ttl_seconds,
                     thumb=thumb,
                     attributes=[
                         raw.types.DocumentAttributeVideo(
diff --git a/pyrogram/types/messages_and_media/message.py b/pyrogram/types/messages_and_media/message.py
@@ -2124,6 +2124,7 @@ async def reply_video(
         caption: str = "",
         parse_mode: Union[str, None] = object,
         caption_entities: List["types.MessageEntity"] = None,
+        ttl_seconds: int = None,
         duration: int = 0,
         width: int = 0,
         height: int = 0,
@@ -2181,6 +2182,11 @@ async def reply_video(
             caption_entities (List of :obj:`~pyrogram.types.MessageEntity`):
                 List of special entities that appear in the caption, which can be specified instead of __parse_mode__.
 
+            ttl_seconds (``int``, *optional*):
+                Self-Destruct Timer.
+                If you set a timer, the video will self-destruct in *ttl_seconds*
+                seconds after it was viewed.
+
             duration (``int``, *optional*):
                 Duration of sent video in seconds.
 
@@ -2252,6 +2258,7 @@ async def reply_video(
             caption=caption,
             parse_mode=parse_mode,
             caption_entities=caption_entities,
+            ttl_seconds=ttl_seconds,
             duration=duration,
             width=width,
             height=height,
