From 84ff0c777d9bc3c7b16fc3cbca7bcfdc39aea852 Mon Sep 17 00:00:00 2001 From: Tulir Asokan Date: Sat, 25 Apr 2020 18:33:34 +0300 Subject: [PATCH] Allow !tg random command with text names instead of emojis --- mautrix_telegram/commands/telegram/misc.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/mautrix_telegram/commands/telegram/misc.py b/mautrix_telegram/commands/telegram/misc.py index 4a51677f..fe41cc12 100644 --- a/mautrix_telegram/commands/telegram/misc.py +++ b/mautrix_telegram/commands/telegram/misc.py @@ -315,10 +315,14 @@ async def random(evt: CommandEvent) -> EventID: if not evt.is_portal: return await evt.reply("You can only roll dice in portal rooms") portal = po.Portal.get_by_mxid(evt.room_id) + arg = evt.args[0] if len(evt.args) > 0 else "dice" + emoticon = { + "dart": "\U0001F3AF", + "dice": "\U0001F3B2", + }.get(arg, arg) try: await evt.sender.client.send_media(await portal.get_input_entity(evt.sender), - InputMediaDice(evt.args[0] if len(evt.args) > 0 - else "\U0001F3B2")) + InputMediaDice(emoticon)) except EmoticonInvalidError: return await evt.reply("Invalid emoji for randomization")