AzurLaneAutoScript/module/meowfficer/meowfficer.py

from module.meowfficer.buy import MeowfficerBuy
from module.meowfficer.fort import MeowfficerFort
from module.meowfficer.train import MeowfficerTrain
from module.ui.page import page_meowfficer
from module.meowfficer.assets import MEOWFFICER_BUY_ENTER


class RewardMeowfficer(MeowfficerBuy, MeowfficerFort, MeowfficerTrain):
    def wait_meowfficer_buttons(self, skip_first_screenshot=True):
        """
        MEOWFFICER_INFO and MEOWFFICER_BUY_ENTER 
        loads slowly than MEOWFFICER_CHECK
        """
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.screenshot()

            if self.appear(MEOWFFICER_BUY_ENTER, offset=(20, 20)):
                break

            # MEOWFFICER_INFO
            if self.ui_additional():
                continue

    def run(self):
        """
        Execute buy, enhance, train, and fort operations
        if enabled in configurations

        Pages:
            in: Any page
            out: page_meowfficer
        """
        if self.config.Meowfficer_BuyAmount <= 0 \
                and not self.config.Meowfficer_FortChoreMeowfficer \
                and not self.config.MeowfficerTrain_Enable:
            self.config.Scheduler_Enable = False
            self.config.task_stop()

        self.ui_ensure(page_meowfficer)
        self.wait_meowfficer_buttons()  # Wait for the ui to load fully

        if self.config.Meowfficer_BuyAmount > 0:
            self.meow_buy()
        if self.config.Meowfficer_FortChoreMeowfficer:
            self.meow_fort()

        # Train
        if self.config.MeowfficerTrain_Enable:
            self.meow_train()
            if self.config.MeowfficerTrain_Mode == 'seamlessly':
                self.meow_enhance()
            elif self.meow_is_sunday():
                self.meow_enhance()
            else:
                pass

        # Scheduler
        if self.config.MeowfficerTrain_Enable:
            # Meowfficer training duration:
            # - Blue, 2.0h ~ 2.5h
            # - Purple, 5.5h ~ 6.5h
            # - Gold, 9.5h ~ 10.5h
            # Delay 2.5h ~ 3.5h when having meowfficers under training
            self.config.task_delay(minute=(150, 210), server_update=True)
        else:
            self.config.task_delay(server_update=True)