ducklet/hotdog
1
0
Fork 0
Code Issues 1 Pull requests Releases Activity

prost: harden code against unicode input problems

Browse source
This commit is contained in:
ducklet 2020-11-07 19:53:36 +01:00
parent 798b339b4b
commit e7009406c6
1 changed files with 20 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

37
hotdog/command/prost.py
View file

@ -8,28 +8,31 @@ def init(bot):
bot.on_message(handle)
vs16 = "\ufe0f" # see https://en.wikipedia.org/wiki/Variation_Selectors_%28Unicode_block%29
emojis = {
"☕️",
"🍵",
"🍶",
"🍷",
"🍸",
"🍹",
"🍺",
"🍻",
"🍼",
"🍾",
"🥂",
"🥃",
"🥛",
"🥤",
"🧃",
"🧉",
e.rstrip(vs16)
for e in {
"☕",
"🍵",
"🍶",
"🍷",
"🍸",
"🍹",
"🍺",
"🍻",
"🍼",
"🍾",
"🥂",
"🥃",
"🥛",
"🥤",
"🧃",
"🧉",
}
}
async def handle(message: Message):
vs16 = "\ufe0f" # see https://en.wikipedia.org/wiki/Variation_Selectors_%28Unicode_block%29
is_drinkmoji = message.text.rstrip(vs16) in emojis
if not (message.tokens.str(0).startswith("PR!") or is_drinkmoji):
return