1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00

Readd legacy keyboard registration

This commit is contained in:
Steffo 2020-04-08 20:21:52 +02:00
parent 81cc77dd66
commit 1ff3d6edd4
2 changed files with 18 additions and 2 deletions

View file

@ -80,3 +80,11 @@ class CommandData:
async def keyboard(self, text, keys: List["KeyboardKey"]):
yield
raise UnsupportedError(f"{self.keyboard.__name__} is not supported")
@classmethod
def register_keyboard_key(cls, identifier: str, key: "KeyboardKey"):
raise UnsupportedError(f"{cls.register_keyboard_key.__name__} is not supported")
@classmethod
def unregister_keyboard_key(cls, identifier: str):
raise UnsupportedError(f"{cls.unregister_keyboard_key.__name__} is not supported")

View file

@ -147,7 +147,7 @@ class TelegramSerf(Serf):
for key in keys:
uid: str = str(uuid.uuid4())
key_uids.append(uid)
self.key_callbacks[uid] = key
data.register_keyboard_key(uid, key)
tg_button: telegram.InlineKeyboardButton = telegram.InlineKeyboardButton(key.text,
callback_data=uid)
tg_row: List[telegram.InlineKeyboardButton] = [tg_button]
@ -161,7 +161,15 @@ class TelegramSerf(Serf):
yield message
await self.api_call(message.edit_reply_markup, reply_markup=None)
for uid in key_uids:
del self.key_callbacks[uid]
data.unregister_keyboard_key(uid)
@classmethod
def register_keyboard_key(cls, identifier: str, key: rc.KeyboardKey):
self.key_callbacks[identifier] = key
@classmethod
def unregister_keyboard_key(cls, identifier: str):
del self.key_callbacks[identifier]
return TelegramMessageData