Alas/module/gg_handler/gg_handler.py

from module.logger import logger
from module.gg_handler.assets import *
from module.gg_handler.gg_data import gg_data
from module.base.base import ModuleBase as base


class gg_handler(base):

    def __init__(self, config, device, switch=True, factor=200):
        self.s = switch
        self.f = factor
        self.device = device
        self.config = config
        
    def skip_error(self):
        """
        Page: 
            in: Game down error
            out: restart
        """
        skip_first_screenshot = False
        count=0
        for i in range(10):
            skipped = 0
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.3)
                self.device.screenshot()
            if base.appear(self, BUTTON_GG_RESTART_ERROR, offset=(50, 50)):
                logger.hr('Game died with GG panel')
                logger.info('Close GG restart error')
                self.device.click(BUTTON_GG_RESTART_ERROR)
                skipped = 1
                count += 1
                if count>=2: break
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.3)
                self.device.screenshot()
            if base.appear(self, BUTTON_GG_RESTART_ERROR, offset=(50, 50)):
                logger.hr('Game died with GG panel')
                logger.info('Close GG restart error')
                skipped = 1
                self.device.click(BUTTON_GG_RESTART_ERROR)
            elif base.appear(self, BUTTON_GG_SCRIPT_END, offset=(50, 50)):
                logger.info('Close previous script')
                skipped = 1
                self.device.click(BUTTON_GG_SCRIPT_END)
            elif base.appear(self, BUTTON_GG_SCRIPT_FATAL, offset=(50, 50)):
                logger.info('Restart previous script')
                skipped = 1
                self.device.click(BUTTON_GG_SCRIPT_FATAL)
            elif base.appear(self, BUTTON_GG_APP_CHOOSE, offset=(150, 500)):
                logger.info('APP choose')
                skipped = 1
                self.device.click(BUTTON_GG_APP_CHOOSE)
            elif base.appear(self, BUTTON_GG_SCRIPT_MENU_A, offset=(50, 50)):
                skipped = 1
                logger.info('Close previous script')
                self.device.click(BUTTON_GG_EXIT_POS)
            elif not base.appear(self, BUTTON_GG_CONFIRM, offset=(50, 50)):
                logger.hr('GG Panel Disappearance Confirmed')
                break
            elif base.appear(self, BUTTON_GG_SEARCH_MODE_CONFIRM, offset=(10, 10), threshold=0.999):
                logger.info('At GG main panel, click GG exit')
                skipped = 1
                self.device.click(BUTTON_GG_EXIT_POS)
            elif base.appear(self, BUTTON_GG_CONFIRM, offset=(50, 50)) and \
                    not base.appear(self, BUTTON_GG_CONFIRM, offset=10):
                self.device.click(BUTTON_GG_TAB_SEARCH_POS)
                skipped = 1
                logger.info('Enter search mode')
            elif base.appear(self, BUTTON_GG_CONFIRM, offset=10):
                logger.info('Unexpected GG page, Try GG exit')
                self.device.click(BUTTON_GG_EXIT_POS)
                skipped = 1
        return skipped
    
    def _enter_gg(self):
        """
        Page:
            in: any
            out: any GG
        """
        self.device.click(BUTTON_GG_ENTER_POS)
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.3)
                self.device.screenshot()
            if base.appear(self, BUTTON_GG_CONFIRM, offset=(50, 50)):
                logger.info('Entered GG')
                break
            self.device.click(BUTTON_GG_ENTER_POS)
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.3)
                self.device.screenshot()
            if not base.appear(self, BUTTON_GG_APP_CHOOSE, offset=(150, 500)):
                from module.ui.assets import BACK_ARROW
                self.device.click(BACK_ARROW)
                logger.info('Actually APP choosing button')
            else:
                base.appear_then_click(self, BUTTON_GG_APP_CHOOSE, offset=(150, 500))
                logger.info('APP Choose')
                break

    def _gg_enter_script(self):
        """
        Page:
            in: any GG
            out: GG ready to start script
        """
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.5)
                self.device.screenshot()
            if base.appear(self, BUTTON_GG_SCRIPT_ENTER_CONFIRM, offset=(50, 50)):
                logger.info('GG script ready to start')
                break
            elif base.appear(self, BUTTON_GG_SCRIPT_END, offset=(50, 50)):
                logger.info('Close previous script')
                self.device.click(BUTTON_GG_SCRIPT_END)
            elif base.appear(self, BUTTON_GG_SCRIPT_FATAL, offset=(50, 50)):
                logger.info('Stop previous script')
                self.device.click(BUTTON_GG_SCRIPT_FATAL)
            elif base.appear(self, BUTTON_GG_APP_CHOOSE, offset=(150, 500)):
                logger.info('APP choose')
                self.device.click(BUTTON_GG_APP_CHOOSE)
            elif base.appear(self, BUTTON_GG_SEARCH_MODE_CONFIRM, offset=(10, 10), threshold=0.95):
                self.device.click(BUTTON_GG_SCRIPT_ENTER_POS)
                logger.info('Enter script choose')
            else:
                self.device.click(BUTTON_GG_TAB_SEARCH_POS)
                logger.info('Enter search mode')
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.5)
                self.device.screenshot()
            if base.appear(self, BUTTON_GG_SCRIPT_START, offset=(50, 50)):
                self.device.click(BUTTON_GG_SCRIPT_START)
                return 1

    def _gg_mode(self):
        """
        Page:
            in: GG Script Menu
            out: GG GG input panel
        """
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.5)
                self.device.screenshot()
            if base.appear(self, BUTTON_GG_SCRIPT_MENU_A, offset=(50, 50), threshold=0.8):
                method = [BUTTON_GG_SCRIPT_MENU_B, BUTTON_GG_SCRIPT_MENU_A]
                self.device.click(method[int(self.s)])
                break
        
    def _gg_handle_factor(self):
        """
        Page:
            in: GG input panel
            out:factor set(Not ensured yet)
        """
        base.wait_until_appear(self, BUTTON_GG_SCRIPT_START_PROCESS, skip_first_screenshot=True)
        logger.info(f'Factor={self.f}')
        if self.f == 200:
            logger.info('Skip factor input')
            return 0
        method = [
                BUTTON_GG_SCRIPT_PANEL_NUM0,
                BUTTON_GG_SCRIPT_PANEL_NUM1,
                BUTTON_GG_SCRIPT_PANEL_NUM2,
                BUTTON_GG_SCRIPT_PANEL_NUM3,
                BUTTON_GG_SCRIPT_PANEL_NUM4,
                BUTTON_GG_SCRIPT_PANEL_NUM5,
                BUTTON_GG_SCRIPT_PANEL_NUM6,
                BUTTON_GG_SCRIPT_PANEL_NUM7,
                BUTTON_GG_SCRIPT_PANEL_NUM8,
                BUTTON_GG_SCRIPT_PANEL_NUM9,
               ]
        for i in str(self.f):
            base.appear_then_click(self, method[int(i)], offset=(50, 50))
            self.device.sleep(0.5)
        logger.info('Input success')

    def _gg_script_run(self):
        """
        Page:
            in: GG factor set
            out: GG Menu
        """
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.5)
                self.device.screenshot()
            if base.appear_then_click(self, BUTTON_GG_SCRIPT_START_PROCESS, offset=(50, 50), threshold=0.9):
                break
        
        logger.info('Waiting for end')
        skip_first_screenshot = False
        while 1:
            if skip_first_screenshot:
                skip_first_screenshot = False
            else:
                self.device.sleep(0.5)
                self.device.screenshot()
            if base.appear_then_click(self, BUTTON_GG_SCRIPT_END, offset=(50, 50), threshold=0.9):
                return 1
    
    # def _gg_exit(self):
    #     self.device.click(BUTTON_GG_EXIT_POS)
    #     skip_first_screenshot = False
    #     while 1:
    #         if skip_first_screenshot:
    #             skip_first_screenshot = False
    #         else:
    #             self.device.sleep(0.5)
    #             self.device.screenshot()
    #         if not base.appear(self, BUTTON_GG_CONFIRM, offset=(10,10)):
    #             logger.hr('GG Panel Exited')
    #             return 1
    #         self.device.click(BUTTON_GG_EXIT_POS)
    #         logger.info('Click GG exit')

    def gg_run(self):
        self._enter_gg()
        self._gg_enter_script()
        self._gg_mode()
        self._gg_handle_factor()
        self._gg_script_run()
        gg_data(self.config, target='gg_on', value=self.s).set_data()
        self.skip_error()
        logger.attr('GG', 'Enabled')