diff --git a/commands/react_baltics/consts.py b/commands/react_baltics/consts.py index e2613d0..fbcd0ab 100644 --- a/commands/react_baltics/consts.py +++ b/commands/react_baltics/consts.py @@ -3,7 +3,7 @@ POSSIBLE_REACTS = ( "<:tw_baltics_heart:1132524940587962388>", "<:bubblemilktea:1132632348651966596>", - "<:chaos:1140296704516706315>", + # "<:chaos:1140296704516706315>", ) LT_REGEX_STR = r"Taivan(?:as|a|e|o|ui|ie(?:tiškas|tis|tė|či(?:ų|u|ai|ui|ams)))" @@ -33,6 +33,9 @@ LV_REGEX = compile( rf"\b(?:{LV_REGEX_STR})\b|{LV_REGEX_STR_TW}", flags=IGNORECASE | UNICODE ) +ET_REGEX = compile( + rf"\b(?:{ET_REGEX_STR})\b|{ET_REGEX_STR_TW}", flags=IGNORECASE | UNICODE +) BALTIC_REGEX = compile( rf"\b(?:{'|'.join(KEYWORDS)})\b|{LV_REGEX_STR_TW}|{LT_REGEX_STR_TW}|{ET_REGEX_STR_TW}", diff --git a/commands/react_baltics/react_baltics.py b/commands/react_baltics/react_baltics.py index 513d2cd..4e1c108 100644 --- a/commands/react_baltics/react_baltics.py +++ b/commands/react_baltics/react_baltics.py @@ -1,6 +1,6 @@ import discord import random -from .consts import POSSIBLE_REACTS, BALTIC_REGEX, LT_REGEX, LV_REGEX +from .consts import POSSIBLE_REACTS, BALTIC_REGEX, LT_REGEX, LV_REGEX, ET_REGEX from ..modules.probability import mock_bernoulli @@ -12,7 +12,9 @@ async def send_react_baltic(message: discord.Message): for react in POSSIBLE_REACTS: if mock_bernoulli(0.40): await message.add_reaction(react) - if LV_REGEX.search(message.content) and mock_bernoulli(0.60): + if LV_REGEX.search(message.content) and mock_bernoulli(0.69): await message.add_reaction("<:lv_tw_hgs:1139660598574055514>") - if LT_REGEX.search(message.content) and mock_bernoulli(0.60): + elif LT_REGEX.search(message.content) and mock_bernoulli(0.69): await message.add_reaction("<:lt_tw_hgs:1139647918442283038>") + elif ET_REGEX.search(message.content) and mock_bernoulli(0.69): + await message.add_reaction("<:et_tw_hgs:1139660507167608882>")