3 Commits

Author SHA1 Message Date
AlberLC
f402293c99 Upgrade _on_choose 2022-06-27 04:22:47 +02:00
AlberLC
99effad710 Fix constants accents 2022-06-23 04:51:49 +02:00
AlberLC
54cada0649 Fix _on_choose 2022-06-23 04:31:45 +02:00
2 changed files with 23 additions and 10 deletions

View File

@@ -303,21 +303,34 @@ class FlanaBot(MultiBot, ABC):
if message.chat.is_group and not self.is_bot_mentioned(message):
return
discarded_words = {*constants.KEYWORDS['choose'], *constants.KEYWORDS['random'], self.name, f'<@{self.id}>'}
discarded_words = {
*constants.KEYWORDS['choose'],
*constants.KEYWORDS['random'],
self.name.lower(), f'<@{self.id}>',
*flanautils.CommonWords.get('conjunctions'),
'entre', 'between'
}
if final_words := [word for word in message.text.split() if not flanautils.cartesian_product_string_matching(word.lower(), discarded_words, min_ratio=multibot_constants.PARSE_CALLBACKS_MIN_RATIO_DEFAULT)]:
for i in range(1, len(final_words) - 1):
if final_words[i] in ('al', 'del', 'to'):
n1 = final_words[i - 1]
n2 = final_words[i + 1]
await self.send(random.randint(n1, n2), message)
return
await self.send(random.choice(final_words), message)
else:
await self.send(random.choice(('¿Que elija el qué?', '¿Y las opciones?', '?', '🤔')), message)
async def _on_config_button_press(self, message: Message):
await self._accept_button_event(message)
await self.accept_button_event(message)
if message.buttons_info.presser_user.is_admin is False:
return
config = message.buttons_info.pressed_text.split()[1]
message.chat.config[config] = not message.chat.config[config]
pressed_button = message.buttons_info[message.buttons_info.pressed_text]
pressed_button = message.buttons_info.pressed_button
pressed_button.is_checked = not pressed_button.is_checked
pressed_button.text = f"{'' if pressed_button.is_checked else ''} {config}"
@@ -451,7 +464,7 @@ class FlanaBot(MultiBot, ABC):
await self.send(random.choice(('¿Y las opciones?', '?', '🤔')), message)
async def _on_poll_button_press(self, message: Message):
await self._accept_button_event(message)
await self.accept_button_event(message)
if not message.contents['poll']['is_active']:
return
@@ -636,7 +649,7 @@ class FlanaBot(MultiBot, ABC):
await self.unpunish(user, message, message)
async def _on_weather_button_press(self, message: Message):
await self._accept_button_event(message)
await self.accept_button_event(message)
match message.buttons_info.pressed_text:
case WeatherEmoji.ZOOM_IN.value:
@@ -699,7 +712,7 @@ class FlanaBot(MultiBot, ABC):
- flanautils.cartesian_product_string_matching(original_text_words, multibot_constants.KEYWORDS['date'], min_ratio=0.85).keys()
- flanautils.cartesian_product_string_matching(original_text_words, multibot_constants.KEYWORDS['thanks'], min_ratio=0.85).keys()
- possible_mentioned_ids
- flanautils.CommonWords.all_words
- flanautils.CommonWords.get()
)
if not place_words:
if not message.is_inline:

View File

@@ -28,8 +28,8 @@ INSULTS = (
'Baneito pa ti en breve.',
'Calla noob.',
'Cansino.',
'Cuentame menos.',
'Cuentame más.',
'Cuéntame menos.',
'Cuéntame más.',
'Cállate ya anda.',
'Cállate.',
'Das penilla.',
@@ -51,7 +51,7 @@ INSULTS = (
'Más tonto y no naces.',
'No eres muy avispado tú...',
'Pesado.',
'Que bien, ¿eh?',
'Qué bien, ¿eh?',
'Que me dejes en paz.',
'Qué pesado.',
'Quita bicho.',
@@ -85,7 +85,7 @@ INSULTS = (
)
KEYWORDS = {
'choose': ('choose', 'elige'),
'choose': ('choose', 'elige', 'escoge'),
'covid_chart': ('case', 'caso', 'contagiado', 'contagio', 'corona', 'coronavirus', 'covid', 'covid19', 'death',
'disease', 'enfermedad', 'enfermos', 'fallecido', 'incidencia', 'jacovid', 'mascarilla', 'muerte',
'muerto', 'pandemia', 'sick', 'virus'),