Alas/module/support_build/ui.py

from module.base.button import Button
from module.base.timer import Timer
from module.gacha.ui import GachaUI
from module.logger import logger
from module.ui.page import page_build


class SupportBuildUI(GachaUI):
    def support_build_load_ensure(self, skip_first_screenshot=True):
        """
        Ensure support build page is loaded.

        Args:
            skip_first_screenshot (bool):

        Returns:
            bool: if page loaded
        """
        from module.gacha.assets import BUILD_SUBMIT_ORDERS

        confirm_timer = Timer(1, count=2).start()
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.screenshot()

            # End
            if self.appear(BUILD_SUBMIT_ORDERS):
                if confirm_timer.reached():
                    break

        return True

    def ui_goto_support_build(self):
        """
        Enter support build page.

        Pages:
            in: page_build
            out: page_support_build
        """
        from module.support_build.assets import SUPPORT_BUILD_CHECK

        self.ui_ensure(page_build)

        # Click at (53, 370) to enter support build
        self.device.click(Button(area=(), color=(), button=(53, 370, 53, 370), name='SUPPORT_BUILD_ENTER'))

        confirm_timer = Timer(1, count=2).start()
        for _ in range(10):
            if self.appear(SUPPORT_BUILD_CHECK, offset=(20, 20)):
                if confirm_timer.reached():
                    break
                self.device.screenshot()
                continue
            confirm_timer.reset()
            self.device.screenshot()

        logger.info('Enter support build page')