AlasGG

campaign/event_20200917_cn/campaign_base.py

@@ -0,0 +1,129 @@
+from module.base.button import Button
+from module.base.utils import *
+from module.campaign.campaign_base import CampaignBase as CampaignBase_
+from module.exception import CampaignNameError
+from module.logger import logger
+
+# Here manually type coordinates, because the ball appears in event Dreamwaker's Butterfly only.
+BALL = Button(area=(571, 283, 696, 387), color=(), button=(597, 274, 671, 343))
+
+
+class CampaignBase(CampaignBase_):
+    STAGE_INCREASE = [
+        'TS1 > T1 > T2 > T3 > T4 > TS2 > T5 > T6',
+        'HTS1 > HT1 > HT2 > HT3',
+        'HT4 > HTS2 > HT5 > HT6',
+    ]
+
+    def campaign_set_chapter(self, name, mode='normal'):
+        """
+        Args:
+            name (str): Campaign name, such as '7-2', 'd3', 'sp3'.
+            mode (str): 'normal' or 'hard'.
+        """
+        chapter, stage = self._campaign_separate_name(name)
+        name = chapter + stage
+
+        if chapter.isdigit():
+            self.ui_goto_campaign()
+            self.campaign_ensure_mode('normal')
+            self.campaign_ensure_chapter(index=chapter)
+            if mode == 'hard':
+                self.campaign_ensure_mode('hard')
+
+        elif chapter in 'abcd' or chapter == 'ex_sp':
+            self.ui_goto_event()
+            if chapter in 'ab':
+                self.campaign_ensure_mode('normal')
+            elif chapter in 'cd':
+                self.campaign_ensure_mode('hard')
+            elif chapter == 'ex_sp':
+                self.campaign_ensure_mode('ex')
+            self.campaign_ensure_chapter(index=chapter)
+
+        elif chapter == 'sp':
+            self.ui_goto_sp()
+            self.campaign_ensure_chapter(index=chapter)
+
+        elif chapter in ['t', 'ts', 'ht', 'hts']:
+            self.ui_goto_event()
+            # Campaign ball
+            if stage in ['1', '6']:
+                self._campaign_ball_set('blue')
+            else:
+                self._campaign_ball_set('red')
+            # Campaign mode
+            if chapter in ['t', 'ts']:
+                self.campaign_ensure_mode('normal')
+            if chapter in ['ht', 'hts']:
+                self.campaign_ensure_mode('hard')
+            if chapter == 'ex_sp':
+                self.campaign_ensure_mode('ex')
+            # Get stage
+            self.campaign_ensure_chapter(index=1)
+        else:
+            logger.warning(f'Unknown campaign chapter: {name}')
+
+    @staticmethod
+    def _campaign_get_chapter_index(name):
+        """
+        Args:
+            name (str, int):
+
+        Returns:
+            int
+        """
+        if isinstance(name, int):
+            return name
+        else:
+            if name.isdigit():
+                return int(name)
+            elif name in ['a', 'c', 'sp', 'ex_sp', 'ts', 't', 'ht', 'hts']:
+                return 1
+            elif name in ['b', 'd', 'ex_ex']:
+                return 2
+            else:
+                raise CampaignNameError
+
+    def _campaign_ball_get(self):
+        """
+        Returns:
+            str: 'blue' or 'red'.
+        """
+        color = get_color(self.device.image, BALL.area)
+        # Blue: (93, 127, 182), Red: (186, 116, 124)
+        index = np.argmax(color)
+        if index == 0:
+            return 'red'
+        elif index == 2:
+            return 'blue'
+        else:
+            logger.warning(f'Unknown campaign ball color: {color}')
+            return 'unknown'
+
+    def _campaign_ball_set(self, status):
+        """
+        Args:
+            status (str): 'blue' or 'red'.
+        """
+        skip_first_screenshot = True
+        while 1:
+            if skip_first_screenshot:
+                skip_first_screenshot = False
+            else:
+                self.device.screenshot()
+
+            current = self._campaign_ball_get()
+            logger.attr('Campaign_ball', current)
+
+            if current == status:
+                break
+            else:
+                if self.is_in_stage():
+                    self.device.click(BALL)
+                    self.device.sleep(3)
+                    # wait until is_in_stage
+                    while 1:
+                        self.device.screenshot()
+                        if self.is_in_stage():
+                            break