Merge a766829469 into 2cad7ebbac

tasks/base/assets/assets_base_page.py

@@ -573,10 +573,10 @@ WORK_CHECK = ButtonWrapper(
     ),
     zht=Button(
         file='./assets/zht/base/page/WORK_CHECK.png',
-        area=(102, 6, 169, 41),
+        area=(668, 402, 774, 442),
-        search=(82, 0, 189, 61),
+        search=(648, 382, 794, 462),
-        color=(184, 194, 203),
+        color=(182, 199, 224),
-        button=(102, 6, 169, 41),
+        button=(668, 402, 774, 442),
     ),
 )
 WORK_GO_TO_BOUNTY = ButtonWrapper(

tasks/base/ui.py

@@ -1,6 +1,7 @@
 from module.base.button import ButtonWrapper
 from module.base.decorator import run_once, Config
 from module.base.timer import Timer
+from module.base.utils import get_color
 from module.exception import GameNotRunningError, GamePageUnknownError, RequestHumanTakeover
 from module.logger import logger
 from module.ocr.ocr import Ocr, Digit
@@ -97,6 +98,7 @@ class UI(MainPage):
                     timeout.reset()
                     continue
                 logger.info("Unknown page, try to back")
+                # allows TooManyClicks to be triggered in case something goes wrong
                 if u2_back:
                     self.device.back()
                     u2_back = False
@@ -130,7 +132,7 @@ class UI(MainPage):
         self.interval_clear(list(Page.iter_check_buttons()))
 
         # loading_timer = Timer(0.5)
-        #back_timer = Timer(10, 10)
+        back_timer = Timer(15,15)
         logger.hr(f"UI goto {destination}")
         while 1:
             if skip_first_screenshot:
@@ -177,14 +179,18 @@ class UI(MainPage):
             if self.ui_additional():
                 continue
 
-            #back_timer.start()
+            back_timer.start()
-            #if back_timer.reached_and_reset():
+            if back_timer.reached():
-            #    if self.match_color(LOGIN_LOADING, interval=5, threshold=80) or self.appear_trademark_year():
+                if self.match_color(LOGIN_LOADING, interval=5, threshold=80) or self.appear_trademark_year():
-            #        from tasks.login.login import Login
+                    from tasks.login.login import Login
-            #        Login(self.config, self.device).handle_app_login()
+                    Login(self.config, self.device).handle_app_login()
-            #        continue
+                # don't click back when screen is black. 
-            #    self.device.back()
+                # Useful for loading screen after switching between pages
-            #    logger.info("Unknown page, try to back")
+                elif [x for x in get_color(self.device.image, BACK.area) if x > 50]:
+                    self.device.back()
+                    logger.info("Unknown page, try to back")
+                back_timer.reset()
+                
 
         # Reset connection
         Page.clear_connection()
@@ -380,12 +386,14 @@ class UI(MainPage):
             return True
         if self.handle_new_student():
             return True
-        if self.handle_ap_exceed():
+        # disabled because will exit the game if quit appears
-            return True
+
-        if self.handle_insufficient_inventory():
+        #if self.handle_ap_exceed():
-            return True
+        #    return True
-        if self.handle_item_expired():
+        #if self.handle_insufficient_inventory():
-            return True
+        #    return True
+        #if self.handle_item_expired():
+        #    return True
 
         return False