From 74cf112cbc4429044f0e28889d1da971246397df Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Fri, 17 Nov 2023 21:54:35 +0800
Subject: [PATCH] Fix: Entered _claim_rewards() but no rewards to claim

---
 tasks/battle_pass/battle_pass.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tasks/battle_pass/battle_pass.py b/tasks/battle_pass/battle_pass.py
index 0bafbc421..2f5c53d51 100644
--- a/tasks/battle_pass/battle_pass.py
+++ b/tasks/battle_pass/battle_pass.py
@@ -180,6 +180,7 @@ class BattlePassUI(UI):
     def _claim_rewards(self, skip_first_screenshot=True):
         logger.hr('Claim rewards', level=1)
         self.battle_pass_goto(KEYWORD_BATTLE_PASS_TAB.Rewards)
+        timeout = Timer(5, count=15).start()
         while 1:
             if skip_first_screenshot:
                 skip_first_screenshot = False
@@ -192,9 +193,14 @@ class BattlePassUI(UI):
             if self.appear(CLOSE_CHOOSE_GIFT):
                 logger.info('Got reward but have gift to choose')
                 break
+            if timeout.reached():
+                logger.warning('Claim reward timeout, no rewards to claim')
+                break
             if self.appear_then_click(REWARDS_CLAIM_ALL, interval=2):
+                timeout.reset()
                 continue
 
+        logger.info('Close reward popup')
         skip_first_screenshot = True
         while 1:
             if skip_first_screenshot: