add InlineQueryResultVideo

LёNya · delivrance · commit a56b1a3287c4 · 2021-05-12T08:56:44.000+02:00
diff --git a/pyrogram/types/inline_mode/__init__.py b/pyrogram/types/inline_mode/__init__.py
@@ -22,9 +22,10 @@
 from .inline_query_result_animation import InlineQueryResultAnimation
 from .inline_query_result_article import InlineQueryResultArticle
 from .inline_query_result_photo import InlineQueryResultPhoto
+from .inline_query_result_video import InlineQueryResultVideo
 from .inline_query_result_audio import InlineQueryResultAudio
 
 __all__ = [
     "InlineQuery", "InlineQueryResult", "InlineQueryResultArticle", "InlineQueryResultPhoto",
-    "InlineQueryResultAnimation", "InlineQueryResultAudio", "ChosenInlineResult"
+    "InlineQueryResultAnimation", "InlineQueryResultAudio", "InlineQueryResultVideo", "ChosenInlineResult"
 ]
diff --git a/pyrogram/types/inline_mode/inline_query_result_video.py b/pyrogram/types/inline_mode/inline_query_result_video.py
@@ -0,0 +1,136 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-2021 Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Lesser General Public License as published
+#  by the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Pyrogram is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Lesser General Public License for more details.
+#
+#  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
+
+from pyrogram import raw
+from pyrogram import types
+from pyrogram.parser import Parser
+from .inline_query_result import InlineQueryResult
+
+
+class InlineQueryResultVideo(InlineQueryResult):
+    """Link to a video.
+
+    By default, this video will be sent by the user with optional caption.
+    Alternatively, you can use *input_message_content* to send a message with the specified content instead of the
+    video.
+
+    Parameters:
+        video_url (``str``):
+            A valid URL for the embedded video player or video file.
+
+        thumb_url (``str``):
+            URL of the thumbnail (jpeg only) for the video.
+
+        id (``str``, *optional*):
+            Unique identifier for this result, 1-64 bytes.
+            Defaults to a randomly generated UUID4.
+
+        title (``str``, *optional*):
+            Title for the result.
+
+        mime_type (``str``, *optional*):
+            Mime type of the content of video url, “text/html” or “video/mp4”.
+
+        description (``str``, *optional*):
+            Short description of the result.
+
+        caption (``str``, *optional*):
+            Caption of the video to be sent, 0-1024 characters.
+
+        parse_mode (``str``, *optional*):
+            By default, texts are parsed using both Markdown and HTML styles.
+            You can combine both syntaxes together.
+            Pass "markdown" or "md" to enable Markdown-style parsing only.
+            Pass "html" to enable HTML-style parsing only.
+            Pass None to completely disable style parsing.
+
+        reply_markup (:obj:`~pyrogram.types.InlineKeyboardMarkup`, *optional*):
+            An InlineKeyboardMarkup object.
+
+        input_message_content (:obj:`~pyrogram.types.InputMessageContent`):
+            Content of the message to be sent instead of the video. This field is required if InlineQueryResultVideo is
+            used to send an HTML-page as a result (e.g., a YouTube video).
+    """
+
+    def __init__(
+        self,
+        video_url: str,
+        thumb_url: str,
+        id: str = None,
+        title: str = None,
+        mime_type: str = None,
+        description: str = None,
+        caption: str = "",
+        parse_mode: Optional[str] = object,
+        reply_markup: "types.InlineKeyboardMarkup" = None,
+        input_message_content: "types.InputMessageContent" = None
+    ):
+        super().__init__("video", id, input_message_content, reply_markup)
+
+        self.video_url = video_url
+
+        self.thumb_url = thumb_url
+        self.title = title
+
+        if mime_type != "text/html" and mime_type != "video/mp4":
+            raise ValueError("Invalid mime type")
+
+        self.mime_type = mime_type
+        self.description = description
+        self.caption = caption
+        self.parse_mode = parse_mode
+        self.reply_markup = reply_markup
+
+        if mime_type == "text/html" and input_message_content is None:
+            raise ValueError("input_message_content is required for videos with `text/html` mime type")
+
+        self.input_message_content = input_message_content
+
+    async def write(self):
+        video = raw.types.InputWebDocument(
+            url=self.video_url,
+            size=0,
+            mime_type=self.mime_type,
+            attributes=[]
+        )
+
+        thumb = raw.types.InputWebDocument(
+            url=self.thumb_url,
+            size=0,
+            mime_type="image/jpeg",
+            attributes=[]
+        )
+
+        return raw.types.InputBotInlineResult(
+            id=self.id,
+            type=self.type,
+            title=self.title,
+            description=self.description,
+            thumb=thumb,
+            content=video,
+            send_message=(
+                await self.input_message_content.write(self.reply_markup)
+                if self.input_message_content
+                else raw.types.InputBotInlineMessageMediaAuto(
+                    reply_markup=self.reply_markup.write() if self.reply_markup else None,
+                    **await(Parser(None)).parse(self.caption, self.parse_mode)
+                )
+            )
+        )
