From 896e1927a5b4862034afdfa94b6582595a8d2273 Mon Sep 17 00:00:00 2001 From: Midgard Date: Thu, 10 Dec 2020 12:12:30 +0100 Subject: [PATCH] Also recognize numX to award more pancake points --- read_mattermost.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/read_mattermost.py b/read_mattermost.py index 9210f2c..a5bf68f 100755 --- a/read_mattermost.py +++ b/read_mattermost.py @@ -9,6 +9,7 @@ import json from typing import Dict, Set import mattermost import mattermost.ws +import re SERVER = "mattermost.zeus.gent" TEAM_NAME = "zeus" @@ -64,6 +65,7 @@ def first(iterable, default=None): return x return default +NUMBER_EMOJI_NAMES = "zero,one,two,three,four,five,six,seven,eight,nine,keycap_ten".split(",") ################################## # Log in @@ -141,6 +143,14 @@ def reaction_qualifies(reaction): return 1 if reaction["emoji_name"] == DOUBLE_EMOJI_NAME: return 2 + m = re.fullmatch(r"num([0-9]+)", reaction["emoji_name"]) + if m: + return int(m.group(1)) + try: + i = NUMBER_EMOJI_NAMES.index(reaction["emoji_name"]) + return i + except ValueError: + pass return 0