Fix: Wait current team shown up

2024-11-25 10:01:10 +00:00 · 2024-05-13 12:38:55 +08:00 · 2024-05-13 12:38:55 +08:00 · ee3cb4f5ef
commit ee3cb4f5ef
parent 3018449c9c
1 changed files with 7 additions and 6 deletions
--- a/tasks/combat/team.py
+++ b/tasks/combat/team.py
@ -60,12 +60,13 @@ class CombatTeam(UI):
                logger.warning('Wait current team timeout')
                break
            current = self._get_team()
-            if current == index:
-                logger.attr('Team', current)
-                logger.info(f'Already selected to the correct team')
-                return False
-            else:
-                break
+            if current:
+                if current == index:
+                    logger.attr('Team', current)
+                    logger.info(f'Already selected to the correct team')
+                    return False
+                else:
+                    break

        # Set team
        retry = Timer(2, count=10)