ArisuAutoSweeper/tasks/circle/circle.py

from enum import Enum

from module.base.timer import Timer
from module.logger import logger
from tasks.base.page import page_circle
from tasks.base.ui import UI
from tasks.circle.assets.assets_circle import *


class CircleStatus(Enum):
    """
    Circle status
    """
    REWARD = 0
    GOT = 1
    FINISHED = -1


class Circle(UI):
    def _handle_circle(self, status):
        match status:
            case CircleStatus.REWARD:
                if self.appear_then_click(GET_REWARD_AP):
                    logger.info("Get circle AP reward")
                    return CircleStatus.FINISHED
            case CircleStatus.GOT:
                logger.info("Circle AP reward have been got")
                return CircleStatus.FINISHED
            case _:
                logger.warning(f"Invalid status: {status}")
        return status

    def run(self):
        self.ui_ensure(page_circle)

        status = CircleStatus.REWARD
        action_timer = Timer(0.5)
        ap_timer = Timer(2).start()

        while 1:
            self.device.screenshot()

            if self.ui_additional():
                continue

            if ap_timer.reached() and status == CircleStatus.REWARD:
                status = CircleStatus.GOT

            if action_timer.reached_and_reset():
                logger.attr('Status', status)
                status = self._handle_circle(status)

            if status is CircleStatus.FINISHED:
                break

        self.config.task_delay(server_update=True)