M-AzurLaneAutoScript/module/regular_inspect/research_inspect.py

import sys
sys.path.append(r'C:/Users/W1NDe/Documents/GitHub/M-AzurLaneAutoScript')
from module.log_res.log_res import LogRes
from module.base.utils import color_bar_percentage
from module.config.deep import deep_get
from module.logger import logger
from module.base.base import ModuleBase
from module.ui.ui import UI
from module.ui.page import page_shipyard
from module.shipyard.shipyard_reward import RewardShipyard
from module.regular_inspect.assets import (SHIP_LAUNCHED, SHIP_EXPERIENCE_PERCENT_1, SHIP_EXPERIENCE_PERCENT_2,
                                            SHIP_EXPERIENCE_FINISHED_1, SHIP_EXPERIENCE_FINISHED_2,
                                            SHIP_EXPERIENCE_COMPLETE_1, SHIP_EXPERIENCE_COMPLETE_2,
                                            SHIP_EXPERIENCE_COMMIT_1, SHIP_EXPERIENCE_COMMIT_2)
from datetime import datetime


class ExpHasFinished(Exception):
    ...


class ExpFinished(Exception):
    ...


class ExpNotFinished(Exception):
    ...

class ShipLaunched(Exception):
    ...

class ResearchInspect(UI, ModuleBase):
    # def _DisableAllResearchFarmTask(self):
    #     for i in range(1, 7):
    #         self.config.modified[f"{self._GetResearchFarmTaskName(i)}.Scheduler.Enable"] = False
    #     self.config.modified["ResearchFarmingSetting.OpsiHazard1ResearchFarming.Enable"] = False
    #     self.config.save(self.config.config_name)

    def _Override(self, Index):
        if Index == 1:
            self.SHIP_EXPERIENCE_PERCENT = SHIP_EXPERIENCE_PERCENT_1
            self.SHIP_EXPERIENCE_FINISHED = SHIP_EXPERIENCE_FINISHED_1
            self.SHIP_EXPERIENCE_COMPLETE = SHIP_EXPERIENCE_COMPLETE_1
            self.SHIP_EXPERIENCE_COMMIT = SHIP_EXPERIENCE_COMMIT_1
        elif Index == 2:
            self.SHIP_EXPERIENCE_PERCENT = SHIP_EXPERIENCE_PERCENT_2
            self.SHIP_EXPERIENCE_FINISHED = SHIP_EXPERIENCE_FINISHED_2
            self.SHIP_EXPERIENCE_COMPLETE = SHIP_EXPERIENCE_COMPLETE_2
            self.SHIP_EXPERIENCE_COMMIT = SHIP_EXPERIENCE_COMMIT_2

    def _UiGotoTargetShip(self):
        self.ui_goto(page_shipyard)
        self.device.sleep(1.2)#wait for the expBar load completely
        # Series = deep_get(self.config.data, "ResearchFarmingSetting.ResearchFarmingSetting.ResearchSeries")
        # Index = deep_get(self.config.data, "ResearchFarmingSetting.ResearchFarmingSetting.ShipIndex")
        # RewardShipyard(config=self.config, device=self.device).shipyard_set_focus(series=Series, index=Index)

    def _IsSingleFinished(self, Index):
        self._Override(Index)
        self.device.screenshot()
        if self.appear(SHIP_LAUNCHED,offset=(10,10)):
            logger.info(f"the ship has already launched")
            raise ShipLaunched
        
        if self.appear(self.SHIP_EXPERIENCE_COMPLETE,offset=(10,15)):
            logger.info(f"ship's exp {Index} has already completed")
            raise ExpHasFinished

        CurrentPercent = color_bar_percentage(self.device.image, self.SHIP_EXPERIENCE_PERCENT.area, prev_color=(255, 239, 82))
        if CurrentPercent > 0.99 and self.appear(self.SHIP_EXPERIENCE_FINISHED,offset=(15,15)):
            logger.info(f"commit ship's exp {Index}")
            self.ui_click(self.SHIP_EXPERIENCE_FINISHED, check_button=self.SHIP_EXPERIENCE_COMMIT)
            self.device.sleep(0.5)
            self.device.click(self.SHIP_EXPERIENCE_COMMIT)
            logger.info(f"ship's exp {Index} completed")
            raise ExpFinished
        LogRes(self.config).ResearchPercent = int((Index-1+CurrentPercent) * 100)
        logger.info(f"ship's exp {Index} not finished,now {(Index-1+CurrentPercent)*100:.2f}%")
        raise ExpNotFinished

    def _Notify(self, Index):
        IsPush = deep_get(self.config.data, "Main2.RegularInspections.ResearchInspectNotify")
        if IsPush:
            from module.notify import handle_notify
            handle_notify(self.config.Error_OnePushConfig,
                          title=f"Alas <{self.config.config_name}>: Research ship's experience finished",
                          content=f"Ship's experience {Index} has finished")

    def CheckResearchShipExperience(self):
        self.device.screenshot()
        self._UiGotoTargetShip()

        try:
            self._IsSingleFinished(1)
        except ExpHasFinished:
            pass
        except ExpFinished:
            self._Notify(1)
            LogRes(self.config).ResearchPercent =100
            return
        except ExpNotFinished:
            return
        except ShipLaunched:
            LogRes(self.config).ResearchPercent =999
            return

        try:
            self._IsSingleFinished(2)
        except (ExpHasFinished, ExpFinished):
            # self._DisableAllResearchFarmTask()
            self._Notify(2)
            LogRes(self.config).ResearchPercent =200
        except ExpNotFinished:
            return
        except ShipLaunched:
            LogRes(self.config).ResearchPercent =999
            return

    def run(self):
        self.config.task_stop()

if __name__ == "__main__":
    self = ResearchInspect('zTTT')
    # self.CheckResearchShipExperience()
    from PIL import Image
    import numpy as np
    # 截图文件
    file = r'C:\Users\W1NDe\Documents\GitHub\M-AzurLaneAutoScript\module\regular_inspect\test.png'

    self.device.image = np.array(Image.open(file).convert('RGB'))

    def test(self):
        try:
            self._IsSingleFinished(1)
        except ExpHasFinished:
            pass
        except ExpFinished:
            self._Notify(1)
            return
        except ExpNotFinished:
            return

        try:
            self._IsSingleFinished(2)
        except (ExpHasFinished, ExpFinished):
            # self._DisableAllResearchFarmTask()
            self._Notify(2)
        except ExpNotFinished:
            return
    test(self)