From 16f7626ceb10027aa270260868b1e8d71626cc2c Mon Sep 17 00:00:00 2001 From: Lonami Exo Date: Thu, 2 Aug 2018 14:53:26 +0200 Subject: [PATCH] Fix custom.Button._is_inline failing for MessageButton --- telethon/tl/custom/button.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/telethon/tl/custom/button.py b/telethon/tl/custom/button.py index 689d7b51..2029531d 100644 --- a/telethon/tl/custom/button.py +++ b/telethon/tl/custom/button.py @@ -1,4 +1,5 @@ from .. import types +from .messagebutton import MessageButton class Button: @@ -41,12 +42,14 @@ class Button: """ if isinstance(button, cls): return button.is_inline - else: - return isinstance(button, ( - types.KeyboardButtonCallback, - types.KeyboardButtonSwitchInline, - types.KeyboardButtonUrl - )) + elif isinstance(button, MessageButton): + button = button.button + + return isinstance(button, ( + types.KeyboardButtonCallback, + types.KeyboardButtonSwitchInline, + types.KeyboardButtonUrl + )) @classmethod def inline(cls, text, callback=None, data=None):