Add support for user profile buttons

delivrance · delivrance · commit a909dc12e7d2 · 2021-12-22T14:36:02.000+01:00
diff --git a/pyrogram/types/bots_and_keyboards/inline_keyboard_button.py b/pyrogram/types/bots_and_keyboards/inline_keyboard_button.py
@@ -43,6 +43,9 @@ class InlineKeyboardButton(Object):
              An HTTP URL used to automatically authorize the user. Can be used as a replacement for
              the `Telegram Login Widget <https://core.telegram.org/widgets/login>`_.
 
+        user_id (``id``, *optional*):
+            User id, for links to the user profile.
+
         switch_inline_query (``str``, *optional*):
             If set, pressing the button will prompt the user to select one of their chats, open that chat and insert
             the bot's username and the specified inline query in the input field. Can be empty, in which case just
@@ -68,16 +71,18 @@ def __init__(
         callback_data: Union[str, bytes] = None,
         url: str = None,
         login_url: "types.LoginUrl" = None,
+        user_id: int = None,
         switch_inline_query: str = None,
         switch_inline_query_current_chat: str = None,
         callback_game: "types.CallbackGame" = None
     ):
         super().__init__()
 
         self.text = str(text)
+        self.callback_data = callback_data
         self.url = url
         self.login_url = login_url
-        self.callback_data = callback_data
+        self.user_id = user_id
         self.switch_inline_query = switch_inline_query
         self.switch_inline_query_current_chat = switch_inline_query_current_chat
         self.callback_game = callback_game
@@ -110,6 +115,12 @@ def read(b: "raw.base.KeyboardButton"):
                 login_url=types.LoginUrl.read(b)
             )
 
+        if isinstance(b, raw.types.KeyboardButtonUserProfile):
+            return InlineKeyboardButton(
+                text=b.text,
+                user_id=b.user_id
+            )
+
         if isinstance(b, raw.types.KeyboardButtonSwitchInline):
             if b.same_peer:
                 return InlineKeyboardButton(
@@ -150,6 +161,12 @@ async def write(self, client: "pyrogram.Client"):
                 bot=await client.resolve_peer(self.login_url.bot_username)
             )
 
+        if self.user_id is not None:
+            return raw.types.InputKeyboardButtonUserProfile(
+                text=self.text,
+                user_id=await client.resolve_peer(self.user_id)
+            )
+
         if self.switch_inline_query is not None:
             return raw.types.KeyboardButtonSwitchInline(
                 text=self.text,
