Auto balancing ReplyKeyboardMarkup

This commit is contained in:
Forden 2019-09-01 16:19:14 +03:00
parent e56be672a3
commit 1460c07e48

View file

@ -78,6 +78,16 @@ class ReplyKeyboardMarkup(base.TelegramObject):
self.add(button)
return self
def arrange_schema(self, buttons: typing.List[str], count: typing.List[int]):
btns = list(map(KeyboardButton, buttons))
self.row_width = max(count)
tmplist = [[KeyboardButton('') for _ in range(count[i])] for i in range(len(count))]
for a in range(len(tmplist)):
for b in range(len(tmplist[a])):
tmplist[a][b] = btns.pop(0)
self.keyboard = tmplist
return self
class KeyboardButton(base.TelegramObject):
"""