MissKatyPyro/misskaty/core/keyboard.py
2023-08-03 11:01:15 +07:00

31 lines
932 B
Python

from pykeyboard import InlineKeyboard
from pyrogram.types import InlineKeyboardButton as Ikb
from misskaty.helper.functions import get_urls_from_text as is_url
def keyboard(buttons_list, row_width: int = 2):
"""
Buttons builder, pass buttons in a list and it will
return pyrogram.types.IKB object
Ex: keyboard([["click here", "https://google.com"]])
if theres, a url, it will make url button, else callback button
"""
buttons = InlineKeyboard(row_width=row_width)
data = [
Ikb(text=str(i[0]), url=str(i[1]))
if is_url(i[1])
else Ikb(text=str(i[0]), callback_data=str(i[1]))
for i in buttons_list
]
buttons.add(*data)
return buttons
def ikb(data: dict, row_width: int = 2):
"""
Converts a dict to pyrogram buttons
Ex: dict_to_keyboard({"click here": "this is callback data"})
"""
return keyboard(data.items(), row_width=row_width)