Create a custom.Button class and support send_message(buttons=...)

telethon/client/buttons.py

@@ -0,0 +1,48 @@
+from .updates import UpdateMethods
+from ..tl import types, custom
+from .. import utils
+
+
+class ButtonMethods(UpdateMethods):
+    def _build_reply_markup(self, buttons):
+        if buttons is None:
+            return None
+
+        try:
+            if buttons.SUBCLASS_OF_ID == 0xe2e10ef2:
+                return buttons  # crc32(b'ReplyMarkup'):
+        except AttributeError:
+            pass
+
+        if not utils.is_list_like(buttons):
+            buttons = [[buttons]]
+        elif not utils.is_list_like(buttons[0]):
+            buttons = [buttons]
+
+        is_inline = False
+        is_normal = False
+
+        rows = []
+        for row in buttons:
+            current = []
+            for button in row:
+                inline = custom.Button._is_inline(button)
+                is_inline |= inline
+                is_normal |= not inline
+                if isinstance(button, custom.Button):
+                    # TODO actually register callbacks
+                    button = button.button
+
+                if button.SUBCLASS_OF_ID == 0xbad74a3:
+                    # 0xbad74a3 == crc32(b'KeyboardButton')
+                    current.append(button)
+
+            if current:
+                rows.append(types.KeyboardButtonRow(current))
+
+        if is_inline == is_normal and is_normal:
+            raise ValueError('You cannot mix inline with normal buttons')
+        elif is_inline:
+            return types.ReplyInlineMarkup(rows)
+        elif is_normal:
+            return types.ReplyKeyboardMarkup(rows)