support str and KeyboardButton for reply_markup

This commit is contained in:
Rahiel Kasim 2016-04-29 17:42:45 +02:00
parent 998040da92
commit a327e9d6ff

View file

@ -73,7 +73,12 @@ class ReplyKeyboardMarkup(ReplyMarkup):
data = super(ReplyKeyboardMarkup, self).to_dict()
data['keyboard'] = []
for keyboard in self.keyboard:
data['keyboard'].append([x.to_dict() for x in keyboard])
for row in self.keyboard:
r = []
for button in row:
if hasattr(button, 'to_dict'):
r.append(button.to_dict()) # telegram.KeyboardButton
else:
r.append(button) # str
data['keyboard'].append(r)
return data