Fix custom.Button._is_inline failing for MessageButton

This commit is contained in:
Lonami Exo 2018-08-02 14:53:26 +02:00
parent 3612652f39
commit 16f7626ceb

View File

@ -1,4 +1,5 @@
from .. import types from .. import types
from .messagebutton import MessageButton
class Button: class Button:
@ -41,7 +42,9 @@ class Button:
""" """
if isinstance(button, cls): if isinstance(button, cls):
return button.is_inline return button.is_inline
else: elif isinstance(button, MessageButton):
button = button.button
return isinstance(button, ( return isinstance(button, (
types.KeyboardButtonCallback, types.KeyboardButtonCallback,
types.KeyboardButtonSwitchInline, types.KeyboardButtonSwitchInline,