fix: treat wrong team inputs a bit better

maduck · Jul 26, 2020 · b1b91ea · b1b91ea
1 parent 758665e
commit b1b91ea
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/bot.py b/bot.py
@@ -114,7 +114,7 @@ class DiscordBot(discord.Client):
         {
             'function': 'handle_team_code',
             'pattern': re.compile(
-                r'.*?(?P<lang>en|fr|de|ru|it|es|cn)?(?P<shortened>-)?\[(?P<team_code>(\d+,?){1,13})\].*',
+                r'.*?(?P<lang>en|fr|de|ru|it|es|cn)?(?P<shortened>-)?\[(?P<team_code>(\d+,?){1,12})\].*',
                 re.IGNORECASE)
         },
         {

diff --git a/team_expando.py b/team_expando.py
@@ -201,6 +201,7 @@ def get_team_from_code(self, code, lang):
             'troops_title': _('[TROOPS]', lang),
         }
         has_weapon = False
+        has_class = False
 
         for element in code:
             troop = self.troops.get(element)
@@ -221,6 +222,7 @@ def get_team_from_code(self, code, lang):
             if _class:
                 result['class'] = _(_class['name'], lang)
                 result['class_talents'] = _class['talents']
+                has_class = True
                 continue
 
             banner = self.banners.get(element)
@@ -233,7 +235,7 @@ def get_team_from_code(self, code, lang):
                 result['talents'].append(element)
                 continue
 
-        if has_weapon:
+        if has_weapon and has_class:
             new_talents = []
             for talent_no, talent_code in enumerate(result['talents']):
                 talent = '-'