Add {get,set}_chat_menu_button

compiler/docs/compiler.py

@@ -297,6 +297,8 @@ def pyrogram_api():
             delete_bot_commands
             set_bot_default_privileges
             get_bot_default_privileges
+            set_chat_menu_button
+            get_chat_menu_button
         """,
         authorization="""
         Authorization

pyrogram/methods/bots/__init__.py

@@ -21,6 +21,7 @@ from .answer_inline_query import AnswerInlineQuery
 from .delete_bot_commands import DeleteBotCommands
 from .get_bot_commands import GetBotCommands
 from .get_bot_default_privileges import GetBotDefaultPrivileges
+from .get_chat_menu_button import GetChatMenuButton
 from .get_game_high_scores import GetGameHighScores
 from .get_inline_bot_results import GetInlineBotResults
 from .request_callback_answer import RequestCallbackAnswer
@@ -28,6 +29,7 @@ from .send_game import SendGame
 from .send_inline_bot_result import SendInlineBotResult
 from .set_bot_commands import SetBotCommands
 from .set_bot_default_privileges import SetBotDefaultPrivileges
+from .set_chat_menu_button import SetChatMenuButton
 from .set_game_score import SetGameScore
 
 
@@ -44,6 +46,8 @@ class Bots(
     GetBotCommands,
     DeleteBotCommands,
     SetBotDefaultPrivileges,
-    GetBotDefaultPrivileges
+    GetBotDefaultPrivileges,
+    SetChatMenuButton,
+    GetChatMenuButton
 ):
     pass

pyrogram/methods/bots/get_chat_menu_button.py

@@ -0,0 +1,63 @@
+#  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 Union
+
+import pyrogram
+from pyrogram import raw
+from pyrogram import types
+
+
+class GetChatMenuButton:
+    async def get_chat_menu_button(
+        self: "pyrogram.Client",
+        chat_id: Union[int, str] = None,
+    ) -> "types.MenuButton":
+        """Get the current value of the bot's menu button in a private chat, or the default menu button.
+
+        chat_id (``int`` | ``str``):
+            Unique identifier (int) or username (str) of the target chat.
+            If not specified, default bot's menu button will be returned.
+        """
+
+        if chat_id:
+            r = await self.invoke(
+                raw.functions.bots.GetBotMenuButton(
+                    user_id=await self.resolve_peer(chat_id),
+                )
+            )
+        else:
+            r = (await self.invoke(
+                raw.functions.users.GetFullUser(
+                    id=raw.types.InputUserSelf()
+                )
+            )).full_user.bot_info.menu_button
+
+        if isinstance(r, raw.types.BotMenuButtonCommands):
+            return types.MenuButtonCommands()
+
+        if isinstance(r, raw.types.BotMenuButtonDefault):
+            return types.MenuButtonDefault()
+
+        if isinstance(r, raw.types.BotMenuButton):
+            return types.MenuButtonWebApp(
+                text=r.text,
+                web_app=types.WebAppInfo(
+                    url=r.url
+                )
+            )

pyrogram/methods/bots/set_chat_menu_button.py

@@ -0,0 +1,53 @@
+#  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 Union
+
+import pyrogram
+from pyrogram import raw
+from pyrogram import types
+
+
+class SetChatMenuButton:
+    async def set_chat_menu_button(
+        self: "pyrogram.Client",
+        chat_id: Union[int, str] = None,
+        menu_button: "types.MenuButton" = None
+    ) -> bool:
+        """Change the bot's menu button in a private chat, or the default menu button.
+
+        chat_id (``int`` | ``str``, *optional*):
+            Unique identifier (int) or username (str) of the target chat.
+            If not specified, default bot's menu button will be changed.
+
+        menu_button (:obj:`~pyrogram.types.MenuButton`, *optional*):
+            The new bot's menu button.
+            Defaults to :obj:`~pyrogram.types.MenuButtonDefault`.
+        """
+
+        await self.invoke(
+            raw.functions.bots.SetBotMenuButton(
+                user_id=await self.resolve_peer(chat_id or "me"),
+                button=(
+                    (await menu_button.write(self)) if menu_button
+                    else (await types.MenuButtonDefault().write(self))
+                )
+            )
+        )
+
+        return True