Add: Collect battle pass rewards

module/battle_pass/assets.py

@@ -0,0 +1,9 @@
+from module.base.button import Button
+from module.base.template import Template
+
+# This file was automatically generated by dev_tools.button_extract.
+# Don't modify it manually.
+
+BATTLE_PASS_RED_DOT = Button(area={'cn': (623, 105, 629, 113), 'en': (623, 105, 629, 113), 'jp': (623, 105, 629, 113), 'tw': (623, 105, 629, 113)}, color={'cn': (202, 86, 66), 'en': (202, 86, 66), 'jp': (202, 86, 66), 'tw': (202, 86, 66)}, button={'cn': (623, 105, 629, 113), 'en': (623, 105, 629, 113), 'jp': (623, 105, 629, 113), 'tw': (623, 105, 629, 113)}, file={'cn': './assets/cn/battle_pass/BATTLE_PASS_RED_DOT.png', 'en': './assets/cn/battle_pass/BATTLE_PASS_RED_DOT.png', 'jp': './assets/cn/battle_pass/BATTLE_PASS_RED_DOT.png', 'tw': './assets/cn/battle_pass/BATTLE_PASS_RED_DOT.png'})
+PURCHASE_POPUP = Button(area={'cn': (907, 204, 934, 229), 'en': (907, 204, 934, 229), 'jp': (907, 204, 934, 229), 'tw': (907, 204, 934, 229)}, color={'cn': (176, 130, 110), 'en': (176, 130, 110), 'jp': (176, 130, 110), 'tw': (176, 130, 110)}, button={'cn': (907, 204, 934, 229), 'en': (907, 204, 934, 229), 'jp': (907, 204, 934, 229), 'tw': (907, 204, 934, 229)}, file={'cn': './assets/cn/battle_pass/PURCHASE_POPUP.png', 'en': './assets/cn/battle_pass/PURCHASE_POPUP.png', 'jp': './assets/cn/battle_pass/PURCHASE_POPUP.png', 'tw': './assets/cn/battle_pass/PURCHASE_POPUP.png'})
+REWARD_RECEIVE = Button(area={'cn': (1192, 520, 1255, 536), 'en': (1192, 520, 1255, 536), 'jp': (1192, 520, 1255, 536), 'tw': (1192, 520, 1255, 536)}, color={'cn': (191, 178, 163), 'en': (191, 178, 163), 'jp': (191, 178, 163), 'tw': (191, 178, 163)}, button={'cn': (1192, 520, 1255, 536), 'en': (1192, 520, 1255, 536), 'jp': (1192, 520, 1255, 536), 'tw': (1192, 520, 1255, 536)}, file={'cn': './assets/cn/battle_pass/REWARD_RECEIVE.png', 'en': './assets/cn/battle_pass/REWARD_RECEIVE.png', 'jp': './assets/cn/battle_pass/REWARD_RECEIVE.png', 'tw': './assets/cn/battle_pass/REWARD_RECEIVE.png'})

module/battle_pass/battle_pass.py

@@ -0,0 +1,103 @@
+from module.base.timer import Timer
+from module.base.utils import get_color
+from module.battle_pass.assets import *
+from module.combat.combat import Combat
+from module.logger import logger
+from module.ui.assets import REWARD_GOTO_BATTLE_PASS, BATTLE_PASS_CHECK
+from module.ui.page import page_reward
+from module.ui.ui import UI
+
+
+class BattlePass(Combat, UI):
+    def battle_pass_red_dot_appear(self):
+        """
+        Returns:
+            bool: If appear.
+
+        Page:
+            in: page_reward
+        """
+        if self.appear(REWARD_GOTO_BATTLE_PASS, offset=(10, 150)):
+            # Load button offset from REWARD_GOTO_BATTLE_PASS,
+            # because entrance may not be the top one.
+            BATTLE_PASS_RED_DOT.load_offset(REWARD_GOTO_BATTLE_PASS)
+            # Not using self.appear() here, because it's transparent,
+            # color may be different depending on background.
+            r, _, _ = get_color(self.device.image, BATTLE_PASS_RED_DOT.area)
+            if r > BATTLE_PASS_RED_DOT.color[0] - 40:
+                logger.info('Found battle pass red dot')
+                return True
+            else:
+                logger.info('No battle pass red dot')
+                return False
+        else:
+            logger.warning('No battle pass entrance')
+            return False
+
+    def handle_battle_pass_popup(self):
+        return self.appear_then_click(PURCHASE_POPUP, offset=(20, 20), interval=2)
+
+    def battle_pass_enter(self):
+        """
+        Page:
+            in: page_reward
+            out: page_battle_pass
+        """
+
+        def appear_button():
+            return self.appear(REWARD_GOTO_BATTLE_PASS, offset=(10, 150))
+
+        self.ui_click(REWARD_GOTO_BATTLE_PASS, appear_button=appear_button, check_button=BATTLE_PASS_CHECK,
+                      additional=self.handle_battle_pass_popup, skip_first_screenshot=True)
+
+    def battle_pass_receive(self, skip_first_screenshot=True):
+        """
+        Returns:
+            bool: If received.
+
+        Pages:
+            in: page_battle_pass
+            out: page_battle_pass
+        """
+        logger.hr('Battle pass receive', level=1)
+        self.battle_status_click_interval = 2
+        confirm_timer = Timer(1, count=3).start()
+        received = False
+        while 1:
+            if skip_first_screenshot:
+                skip_first_screenshot = False
+            else:
+                self.device.screenshot()
+
+            if self.appear_then_click(REWARD_RECEIVE, offset=(20, 20), interval=2):
+                confirm_timer.reset()
+                continue
+            if self.handle_battle_pass_popup():
+                confirm_timer.reset()
+                continue
+            if self.handle_get_items():
+                received = True
+                confirm_timer.reset()
+                continue
+            if self.handle_get_ship():
+                received = True
+                confirm_timer.reset()
+                continue
+
+            # End
+            if self.appear(BATTLE_PASS_CHECK, offset=(20, 20)) and not self.appear(REWARD_RECEIVE, offset=(20, 20)):
+                if confirm_timer.reached():
+                    break
+            else:
+                confirm_timer.reset()
+
+        return received
+
+    def run(self):
+        self.ui_ensure(page_reward)
+
+        if self.battle_pass_red_dot_appear():
+            self.battle_pass_enter()
+            self.battle_pass_receive()
+
+        self.config.task_delay(server_update=True)