refactor(tc): separate ui operation

2026-06-10 02:34:51 +00:00 · 2023-11-20 22:10:40 +08:00
parent 8e29d7d2c0
commit e929a1efb1
2 changed files with 60 additions and 54 deletions
@@ -0,0 +1,54 @@
+from module.logger import logger
+from module.ocr.ocr import DigitCounter
+from module.ui.switch import Switch
+from tasks.base.ui import UI
+from tasks.tactical_challenge.assets.assets_tactical_challenge import *
+
+SWITCH_SKIP = Switch('Skip_switch')
+SWITCH_SKIP.add_state('on', SKIP_ON)
+SWITCH_SKIP.add_state('off', SKIP_OFF)
+
+
+class TacticalChallengeUI(UI):
+    def get_ticket(self):
+        """
+        Page:
+            in: page_tactical_challenge
+        """
+        ocr = DigitCounter(OCR_TICKET).ocr_single_line(self.device.image)
+        # number of tickets remaining
+        ticket, _, total = ocr
+        if total == 0:
+            logger.warning('Invalid ticket')
+            return False, 5
+        logger.attr('Ticket', ticket)
+
+        return True, ticket
+
+    def get_reward(self):
+        if self.match_color(GET_REWARD_DAILY):
+            self.device.click(GET_REWARD_DAILY)
+            logger.info('Get tc daily reward')
+            return True
+        if self.match_color(GET_REWARD_CREDIT):
+            self.device.click(GET_REWARD_CREDIT)
+            logger.info('Get tc credit reward')
+            return True
+        if self.match_color(GOT_REWARD_DAILY) and self.match_color(GOT_REWARD_CREDIT):
+            logger.info('Both tc reward got')
+            return True
+
+        return False
+
+    def set_skip(self):
+        """
+        Set skip switch to on
+        Returns:
+            True if switch is set, False if switch not found
+        """
+        if not SWITCH_SKIP.appear(main=self):
+            logger.info('Skip switch not found')
+            return False
+        SWITCH_SKIP.set('on', main=self)
+
+        return True