From e00b3ff2bf3dad5e2f3f9d3578255614ab1e6153 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Thu, 19 Dec 2024 22:40:09 +0300
Subject: [PATCH] detection theme changes:

Try parse invisible labs
Full matching theme instead only one index, this prevent some errors
---
 src/autobioebbing.py | 34 +++++++++++++++++++---------------
 1 file changed, 19 insertions(+), 15 deletions(-)

diff --git a/src/autobioebbing.py b/src/autobioebbing.py
index 61ba731..06bb25f 100644
--- a/src/autobioebbing.py
+++ b/src/autobioebbing.py
@@ -21,62 +21,67 @@ def find_infect_theme(text):
         # current order in theme:
         # ('infected', 'bio_expr', 'infected days', 'pathogen remaining')
         # UA theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> йобнув.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> йобнув.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"([0-9\.\,k]+) біо-ресурса",
          default_infected_days_theme,
          default_pathogen_remaining_theme),
         # RU theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> подверг.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> подверг.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          default_bioexpr_theme,
          default_infected_days_theme,
          default_pathogen_remaining_theme),
         # EN theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> infected.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> infected.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"([0-9\.\,k]+) pcs\.",
          r' for ([0-9\ ]+) d.*',
          r'Remaining: ([0-9\ ]+)'),
         # AZ theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> сикди.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> сикди.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"верир: ([0-9\.\,k]+)",
          default_infected_days_theme,
          default_pathogen_remaining_theme),
         # "ПК гик" theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> насрал.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> насрал.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"потеряет: ([0-9\.\,k]+)",
          default_infected_days_theme,
          default_pathogen_remaining_theme),
         # "Новогодняя" theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> подверг заморозке.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> подверг заморозке.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          default_bioexpr_theme,
          default_infected_days_theme,
          default_pathogen_remaining_theme),
         # "Сексуальная индустрия" theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a>.+выебал.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a>.+выебал.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"кончила ([0-9\.\,k]+)",
          r' ещё ([0-9\ ]+) д.*',
          default_pathogen_remaining_theme),
         # "Аферисты в сетях" theme
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> атаковал.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> атаковал.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"приносит: ([0-9\.\,k]+)",
          default_infected_days_theme,
          default_pathogen_remaining_theme),
+        # "Квадробер" theme
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a>.+бешенству.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+         r"корма: ([0-9\.\,k]+)",
+         default_infected_days_theme,
+         default_pathogen_remaining_theme),
         # UA theme [via trust]
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> за допомогою довіреності зазнала зараження.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> за допомогою довіреності зазнала зараження.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"([0-9\.\,k]+) біо-ресурса",
          default_infected_days_theme,
          default_pathogen_remaining_theme),
         # RU theme [via trust]
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> при помощи доверенности подвергла заражению.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> при помощи доверенности подвергла заражению.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          default_bioexpr_theme,
          default_infected_days_theme,
          default_pathogen_remaining_theme),
         # EN theme [via trust]
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> by authorization infected.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> by authorization infected.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r"([0-9\.\,k]+) pcs\.",
          r' for ([0-9\ ]+) d.*',
          r'Remaining: ([0-9\ ]+)'),
         # idk what is theme [via trust]
-        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> при помощи анонимуса атаковала.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
+        (r'<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)?">.*</a> при помощи анонимуса атаковала.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',
          r'приносит: ([0-9\.\,k]+)',
          default_infected_days_theme,
          default_pathogen_remaining_theme),
@@ -84,14 +89,13 @@ def find_infect_theme(text):
     for theme in bio_attack_themes:
         trying_theme_index = bio_attack_themes.index(theme)
         logger.debug(f'trying theme {trying_theme_index}...')
-        r = re.findall(theme[0], text)
-        if r:
+        r, bioexp, days, remaining = re.findall(theme[0], text), re.findall(theme[1], text), re.findall(theme[2], text), re.findall(theme[3], text)
+        if r and bioexp and days and remaining:
             logger.debug(f'found theme {trying_theme_index}')
             return r, bio_attack_themes[trying_theme_index]
     return r, None
 
 
-
 async def eb(client, c, conn, con, d, get_id, my_id, message_q):
     @client.on(events.NewMessage(
                pattern='.*йобнув.*|.*подверг(ла)?.*|.*infected.*|.*сикди.*|.*насрал.*|.*выебал.*|.*за допомогою довіреності.*|.*by authorization infected.*|.*при помощи анонимуса атаковала.*',