From edb55c6f648add42e6cf5b523762539cb4e3fa5c Mon Sep 17 00:00:00 2001
From: LmeSzinc <37934724+LmeSzinc@users.noreply.github.com>
Date: Mon, 20 May 2024 15:19:34 +0800
Subject: [PATCH] Fix: Close weekly reward popup in _rogue_theme_set()

---
 tasks/rogue/entry/entry.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/tasks/rogue/entry/entry.py b/tasks/rogue/entry/entry.py
index 448e33888..acba8caf1 100644
--- a/tasks/rogue/entry/entry.py
+++ b/tasks/rogue/entry/entry.py
@@ -164,6 +164,11 @@ class RogueEntry(RouteBase, RogueRewardHandler, RoguePathHandler, DungeonUI):
             if interval.reached() and self.is_page_rogue_main():
                 self.device.click(THEME_SWITCH)
                 interval.reset()
+            # Weekly refresh popup
+            if self.appear_then_click(REWARD_CLOSE, interval=2):
+                continue
+            if self.handle_reward():
+                continue
 
     def _rogue_world_set(self, world: int | DungeonList, skip_first_screenshot=True):
         """