AzurLaneAutoScript/campaign/event_20210722_cn/campaign_base.py

from module.base.button import Button
from module.campaign.campaign_base import CampaignBase as CampaignBase_
from module.exception import CampaignNameError
from module.logger import logger

ANIMATION_PINK = Button(
    area=(1186, 446, 1272, 493), color=(255, 153, 172), button=(1186, 446, 1272, 493), name='ANIMATION_PINK')
ANIMATION_ORANGE = Button(
    area=(1186, 446, 1272, 493), color=(255, 177, 123), button=(1186, 446, 1272, 493), name='ANIMATION_ORANGE')
ANIMATION_BLUE = Button(
    area=(1186, 446, 1272, 493), color=(176, 192, 251), button=(1186, 446, 1272, 493), name='ANIMATION_BLUE')


class CampaignBase(CampaignBase_):
    """
    In event Azur Anthem (event_20210722_cn), The Idol Master Collaboration, maps are:
    Chapter 1: SP1, SP2, SP3, SP4.
    Chapter 2: VSP.
    """

    @staticmethod
    def _campaign_separate_name(name):
        """
        Args:
            name (str): Stage name in lowercase, such as 7-2, d3, sp3.

        Returns:
            tuple[str]: Campaign_name and stage index in lowercase, Such as ['7', '2'], ['d', '3'], ['sp', '3'].
        """
        if name == 'vsp' or name == 'sp':  # Difference
            return 'ex_sp', '1'
        elif name.startswith('extra'):
            return 'ex_ex', '1'
        elif '-' in name:
            return name.split('-')
        elif name.startswith('sp'):
            return 'sp', name[-1]
        elif name[-1].isdigit():
            return name[:-1], name[-1]

        return CampaignBase_._campaign_separate_name(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', 'as', 'cs', 'sp']:
                return 1
            elif name in ['b', 'd', 'bs', 'ds', 'ex_ex', 'ex_sp']:  # Difference
                return 2
            else:
                raise CampaignNameError

    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, _ = self._campaign_separate_name(name)

        if chapter.isdigit():
            self.ui_goto_campaign()
            self.campaign_ensure_mode('normal')
            self.campaign_ensure_chapter(chapter)
            if mode == 'hard':
                self.campaign_ensure_mode('hard')
                self.campaign_ensure_chapter(chapter)

        elif chapter in ['a', 'b', 'c', 'd', 'ex_sp', 'as', 'bs', 'cs', 'ds']:
            self.ui_goto_event()
            if chapter in ['a', 'b', 'as', 'bs']:
                self.campaign_ensure_mode('normal')
            elif chapter in ['c', 'd', 'cs', 'ds']:
                self.campaign_ensure_mode('hard')
            elif chapter == 'ex_sp':
                # self.campaign_ensure_mode('ex')
                pass  # Difference
            self.campaign_ensure_chapter(chapter)

        elif chapter == 'sp':
            # self.ui_goto_sp()
            self.ui_goto_event()  # Difference
            self.campaign_ensure_chapter(chapter)

        else:
            logger.warning(f'Unknown campaign chapter: {name}')

    def campaign_get_entrance(self, name):
        if name == 'sp':
            name = 'vsp'
        return super().campaign_get_entrance(name)

    def is_event_animation(self):
        """
        Animation in events after cleared an enemy.

        Returns:
            bool: If animation appearing.
        """
        for button in [ANIMATION_PINK, ANIMATION_ORANGE, ANIMATION_BLUE]:
            if self.appear(button):
                logger.info('Idol Master animation, waiting')
                return True

        return False

    def campaign_match_multi(self, *args, **kwargs):
        # Lower campaign match threshold to 0.8, in order to detect 50% clear SP3
        kwargs['similarity'] = 0.8
        return super().campaign_match_multi(*args, **kwargs)