56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
import abc
|
|
from typing import Optional, Type
|
|
|
|
from telebot import TeleBot
|
|
from telebot.types import CallbackQuery, InlineKeyboardMarkup
|
|
|
|
from ..i18n import I18N
|
|
|
|
|
|
class Markup (metaclass=abc.ABCMeta):
|
|
tag: str
|
|
|
|
def __init__(self, bot: TeleBot, i18n: I18N):
|
|
self.bot = bot
|
|
self.t = i18n
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
return self.build(*args, **kwargs)
|
|
|
|
def check(self, call: CallbackQuery) -> bool:
|
|
return call.data == self.tag
|
|
|
|
@abc.abstractmethod
|
|
def build(self, *args, **kwargs) -> Optional[InlineKeyboardMarkup]:
|
|
pass
|
|
|
|
|
|
class DummyMarkup (Markup):
|
|
tag = "__dummy"
|
|
|
|
def check(self, call: CallbackQuery) -> bool:
|
|
return True
|
|
|
|
def build(self, *args, **kwargs) -> Optional[InlineKeyboardMarkup]:
|
|
return None
|
|
|
|
|
|
class MarkupManager:
|
|
def __init__(self, bot: TeleBot, i18n: I18N):
|
|
self.bot = bot
|
|
self.i18n = i18n
|
|
self._prototypes: list[Markup] = []
|
|
self._dummy = DummyMarkup(self.bot, self.i18n)
|
|
|
|
def register_prototype(self, markup_proto_class: Type[Markup]):
|
|
self._prototypes.append(markup_proto_class(self.bot, self.i18n))
|
|
|
|
def __call__(self, tag: str, *args, **kwargs):
|
|
return self.get(tag).build(*args, **kwargs)
|
|
|
|
def get(self, tag: str) -> Optional[Markup]:
|
|
for mp in self._prototypes:
|
|
if mp.tag == tag:
|
|
return mp
|
|
return None
|