feat: Add send_message_draft method

Author: KurimuzonAkuma

compiler/docs/compiler.py

@@ -243,6 +243,7 @@ def get_title_list(s: str) -> list:
             send_voice
             send_video_note
             send_media_group
+            send_message_draft
             send_location
             send_venue
             send_contact

pyrogram/methods/messages/__init__.py

@@ -74,6 +74,7 @@
 from .send_document import SendDocument
 from .send_location import SendLocation
 from .send_media_group import SendMediaGroup
+from .send_message_draft import SendMessageDraft
 from .send_message import SendMessage
 from .send_paid_media import SendPaidMedia
 from .send_paid_reaction import SendPaidReaction
@@ -128,6 +129,7 @@ class Messages(
     SendAnimation,
     SendLocation,
     SendMediaGroup,
+    SendMessageDraft,
     SendMessage,
     SendPaidMedia,
     SendPaidReaction,

pyrogram/methods/messages/send_message_draft.py

@@ -0,0 +1,76 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present 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 List, Optional, Union
+
+import pyrogram
+from pyrogram import enums, raw, types
+
+
+class SendMessageDraft:
+    async def send_message_draft(
+        self: "pyrogram.Client",
+        chat_id: Union[int, str],
+        draft_id: int,
+        text: str,
+        message_thread_id: Optional[int] = None,
+        parse_mode: Optional["enums.ParseMode"] = None,
+        entities: Optional[List["types.MessageEntity"]] = None,
+    ) -> bool:
+        """Use this method to stream a partial message to a user while the message is being generated.
+
+        .. include:: /_includes/usable-by/bots.rst
+
+        Parameters:
+            chat_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target chat.
+                For your personal cloud (Saved Messages) you can simply use "me" or "self".
+                For a contact that exists in your Telegram address book you can use his phone number (str).
+
+            draft_id (``int``):
+                Unique identifier of the message draft, must be non-zero.
+                Changes of drafts with the same identifier are animated.
+
+            text (``str``):
+                Text of the message to be sent.
+
+            message_thread_id (``int``, *optional*):
+                Unique identifier for the target message thread.
+
+            parse_mode (:obj:`~pyrogram.enums.ParseMode`, *optional*):
+                By default, texts are parsed using both Markdown and HTML styles.
+                You can combine both syntaxes together.
+
+            entities (List of :obj:`~pyrogram.types.MessageEntity`):
+                List of special entities that appear in message text, which can be specified instead of *parse_mode*.
+
+        Returns:
+            ``bool``: On success, True is returned.
+        """
+        return await self.invoke(
+            raw.functions.messages.SetTyping(
+                peer=await self.resolve_peer(chat_id),
+                action=raw.types.SendMessageTextDraftAction(
+                    random_id=draft_id,
+                    text=await types.FormattedText(
+                        text=text, parse_mode=parse_mode, entities=entities
+                    ).write(),
+                ),
+                top_msg_id=message_thread_id,
+            )
+        )