Add: Chapter SP

campaign/event_20260417_cn/campaign_base.py

@@ -0,0 +1,38 @@
+from module.base.button import Button
+from module.campaign.campaign_base import CampaignBase as CampaignBase_
+from module.logger import logger
+
+EVENT_ANIMATION = Button(area=(49, 229, 119, 400), color=(118, 215, 240), button=(49, 229, 119, 400),
+                         name='EVENT_ANIMATION')
+
+
+class CampaignBase(CampaignBase_):
+    @staticmethod
+    def _campaign_ocr_result_process(result):
+        result = CampaignBase_._campaign_ocr_result_process(result)
+        if result in ['ysp', 'usp', 'vsp']:
+            result = 'sp'
+        return result
+
+    def is_event_animation(self):
+        """
+        Animation in events after cleared an enemy.
+
+        Returns:
+            bool: If animation appearing.
+        """
+        appear = self.appear(EVENT_ANIMATION)
+        if appear:
+            logger.info('DOA animation, waiting')
+        return appear
+
+    def event_animation_end(self):
+        if not self.appear(EVENT_ANIMATION):
+            return False
+        # wait until EVENT_ANIMATION closed
+        for _ in self.loop():
+            if self.is_event_animation():
+                continue
+            break
+        # now in_map
+        return True