AzurLaneAutoScript/module/combat/submarine.py

from module.base.base import ModuleBase
from module.base.timer import Timer
from module.combat.assets import *
from module.logger import logger


class SubmarineCall(ModuleBase):
    submarine_call_flag = False
    submarine_call_timer = Timer(5)
    submarine_call_click_timer = Timer(1)

    def submarine_call_reset(self):
        """
        Call this method after in battle_execute.
        """
        self.submarine_call_timer.reset()
        self.submarine_call_flag = False

    def handle_submarine_call(self, call_mode='do_not_use', is_boss=False):
        """
        Args:
            call_mode (str): do_not_use, boss_only, every_combat
            is_boss (bool): True on BOSS battle

        Returns:
            bool: If clicked submarine call button.
        """
        if self.submarine_call_flag:
            return False
        if call_mode == 'do_not_use' or (call_mode == 'boss_only' and not is_boss):
            self.submarine_call_flag = True
            return False

        if self.submarine_call_timer.reached():
            logger.info('Submarine call timer reached')
            self.submarine_call_flag = True
            return False

        if not self.appear(SUBMARINE_AVAILABLE_CHECK_1) or not self.appear(SUBMARINE_AVAILABLE_CHECK_2):
            return False

        if self.appear(SUBMARINE_CALLED):
            logger.info('Submarine called')
            self.submarine_call_flag = True
            return False
        elif self.submarine_call_click_timer.reached():
            if not self.appear_then_click(SUBMARINE_READY):
                logger.info('Incorrect submarine icon')
                self.device.click(SUBMARINE_READY)
            logger.info('Call submarine')
            self.submarine_call_click_timer.reset()
            return True