mirror of
https://github.com/Mayuri-Chan/pyrofork.git
synced 2026-01-12 00:54:51 +00:00
90 lines
No EOL
2.1 KiB
Python
90 lines
No EOL
2.1 KiB
Python
from pyrogram.types import (
|
|
InlineKeyboardButton,
|
|
InlineKeyboardMarkup,
|
|
KeyboardButton,
|
|
ReplyKeyboardMarkup,
|
|
ForceReply,
|
|
)
|
|
|
|
|
|
def ikb(rows=None):
|
|
if rows is None:
|
|
rows = []
|
|
|
|
lines = []
|
|
for row in rows:
|
|
line = []
|
|
for button in row:
|
|
button = (
|
|
btn(button, button) if isinstance(button, str) else btn(*button)
|
|
) # InlineKeyboardButton
|
|
line.append(button)
|
|
lines.append(line)
|
|
return InlineKeyboardMarkup(inline_keyboard=lines)
|
|
# return {'inline_keyboard': lines}
|
|
|
|
|
|
def btn(text, value, type="callback_data"):
|
|
return InlineKeyboardButton(text, **{type: value})
|
|
# return {'text': text, type: value}
|
|
|
|
|
|
# The inverse of above
|
|
def bki(keyboard):
|
|
lines = []
|
|
for row in keyboard.inline_keyboard:
|
|
line = []
|
|
for button in row:
|
|
button = ntb(button) # btn() format
|
|
line.append(button)
|
|
lines.append(line)
|
|
return lines
|
|
# return ikb() format
|
|
|
|
|
|
def ntb(button):
|
|
for btn_type in [
|
|
"callback_data",
|
|
"url",
|
|
"switch_inline_query",
|
|
"switch_inline_query_current_chat",
|
|
"callback_game",
|
|
]:
|
|
value = getattr(button, btn_type)
|
|
if value:
|
|
break
|
|
button = [button.text, value]
|
|
if btn_type != "callback_data":
|
|
button.append(btn_type)
|
|
return button
|
|
# return {'text': text, type: value}
|
|
|
|
|
|
def kb(rows=None, **kwargs):
|
|
if rows is None:
|
|
rows = []
|
|
|
|
lines = []
|
|
for row in rows:
|
|
line = []
|
|
for button in row:
|
|
button_type = type(button)
|
|
if button_type == str:
|
|
button = KeyboardButton(button)
|
|
elif button_type == dict:
|
|
button = KeyboardButton(**button)
|
|
|
|
line.append(button)
|
|
lines.append(line)
|
|
return ReplyKeyboardMarkup(keyboard=lines, **kwargs)
|
|
|
|
|
|
kbtn = KeyboardButton
|
|
|
|
|
|
def force_reply(selective=True):
|
|
return ForceReply(selective=selective)
|
|
|
|
|
|
def array_chunk(input_array, size):
|
|
return [input_array[i : i + size] for i in range(0, len(input_array), size)] |