Vladless
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
from __future__ import annotations
|
||
|
||
import aiogram.types
|
||
|
||
from pydantic import ConfigDict
|
||
|
||
|
||
_UniqueGiftColors = getattr(aiogram.types, "UniqueGiftColors", None)
|
||
if _UniqueGiftColors is not None:
|
||
_cfg = getattr(_UniqueGiftColors, "model_config", None)
|
||
_base = dict(_cfg) if _cfg is not None else {}
|
||
_base.pop("protected_namespaces", None)
|
||
_UniqueGiftColors.model_config = ConfigDict(**_base, protected_namespaces=())
|
||
|
||
_OriginalInlineKeyboardButton = aiogram.types.InlineKeyboardButton
|
||
|
||
_button_icon_config: dict[str, dict[str, str]] = {}
|
||
|
||
|
||
def apply_button_icons_patch(config: dict[str, dict[str, str]] | None = None) -> None:
|
||
"""
|
||
Патчит InlineKeyboardButton, добавляя поддержку глобального конфига для иконок и стилей кнопок по callback_data или url.
|
||
"""
|
||
if config is not None:
|
||
_button_icon_config.clear()
|
||
_button_icon_config.update(config)
|
||
|
||
class _PatchedInlineKeyboardButton(_OriginalInlineKeyboardButton):
|
||
def __init__(self, **kwargs: object) -> None:
|
||
key = kwargs.get("callback_data") or kwargs.get("url")
|
||
if key is not None and isinstance(key, str):
|
||
config = _button_icon_config.get(key)
|
||
if config is None and "|" in key:
|
||
prefix = key.split("|", 1)[0]
|
||
config = _button_icon_config.get(prefix)
|
||
if config is not None:
|
||
kwargs = {**kwargs, **config}
|
||
super().__init__(**kwargs)
|
||
|
||
aiogram.types.InlineKeyboardButton = _PatchedInlineKeyboardButton
|
||
|
||
|
||
def set_button_icon_config(config: dict[str, dict[str, str]]) -> None:
|
||
"""Подставить конфиг кнопок (вызвать после загрузки handlers, из handlers.buttons.BUTTON_ICON_CONFIG)."""
|
||
_button_icon_config.clear()
|
||
_button_icon_config.update(config)
|
||
|
||
|
||
def inline_button(
|
||
text: str,
|
||
callback_data: str | None = None,
|
||
url: str | None = None,
|
||
web_app: object | None = None,
|
||
*,
|
||
icon_custom_emoji_id: str | None = None,
|
||
style: str | None = None,
|
||
) -> _OriginalInlineKeyboardButton:
|
||
"""Собирает InlineKeyboardButton с опциональной иконкой (custom emoji) и стилем."""
|
||
kwargs: dict = {"text": text}
|
||
if callback_data is not None:
|
||
kwargs["callback_data"] = callback_data
|
||
if url is not None:
|
||
kwargs["url"] = url
|
||
if web_app is not None:
|
||
kwargs["web_app"] = web_app
|
||
if icon_custom_emoji_id is not None:
|
||
kwargs["icon_custom_emoji_id"] = icon_custom_emoji_id
|
||
if style is not None:
|
||
kwargs["style"] = style
|
||
return aiogram.types.InlineKeyboardButton(**kwargs)
|