fix: login

tasks/base/ui.py

@@ -1,18 +1,29 @@
 from module.base.button import ButtonWrapper
-from module.base.decorator import run_once
+from module.base.decorator import run_once, Config
 from module.base.timer import Timer
-from module.exception import GameNotRunningError, GamePageUnknownError
+from module.exception import GameNotRunningError, GamePageUnknownError, RequestHumanTakeover
 from module.logger import logger
-from module.ocr.ocr import Ocr
+from module.ocr.ocr import Ocr, Digit
 from tasks.base.main_page import MainPage
 from tasks.base.page import Page, page_main
-from tasks.base.assets.assets_base_page import BACK
+from tasks.login.assets.assets_login import LOGIN_LOADING, OCR_YEAR
+#from tasks.base.assets.assets_base_page import BACK
 
 
 class UI(MainPage):
     ui_current: Page
     ui_main_confirm_timer = Timer(0.2, count=2)
 
+    @Config.when(Emulator_GameLanguage='zhs')
+    def appear_trademark_year(self):
+        ocr_year = Digit(OCR_YEAR).ocr_single_line(self.device.image)
+        return ocr_year == 2023
+
+    @Config.when(Emulator_GameLanguage=None)
+    def appear_trademark_year(self):
+        ocr_year = Digit(OCR_YEAR).ocr_single_line(self.device.image)
+        return ocr_year == 2021
+    
     def ui_page_appear(self, page):
         """
         Args:
@@ -77,9 +88,15 @@ class UI(MainPage):
                 logger.info("Additional ui page handled")
                 timeout.reset()
                 continue
+            # this might be bad but it works
+            if self.match_color(LOGIN_LOADING, interval=5, threshold=80) or self.appear_trademark_year():
+                from tasks.login.login import Login
+                Login(self.config, self.device).handle_app_login()
+                continue
             if back_timer.reached_and_reset():
                 logger.info("Unknown page, try to back")
-                self.device.click(BACK)
+                #self.device.click(BACK)
+                self.device.u2.press("back")
 
             app_check()
             minicap_check()
@@ -176,10 +193,12 @@ class UI(MainPage):
 
         if self.ui_current == destination:
             logger.info("Already at %s" % destination)
+            self.close_popup(destination.check_button)
             return False
         else:
             logger.info("Goto %s" % destination)
             self.ui_goto(destination, skip_first_screenshot=True)
+            self.close_popup(destination.check_button)
             return True
 
     def ui_ensure_index(
@@ -399,3 +418,18 @@ class UI(MainPage):
             button (Button):
         """
         pass
+
+    def close_popup(self, check_button):
+        if not self.match_color(check_button):
+            timer = Timer(5, 5).start()
+            wait = Timer(1).start()
+            while 1:
+                self.device.screenshot()
+                if self.match_color(check_button):
+                    break
+                self.device.u2.press("back")
+                if timer.reached():
+                    logger.error("Failed to close popup")
+                    raise RequestHumanTakeover
+                while not wait.reached():
+                    pass    

tasks/login/assets/assets_login.py

@@ -51,6 +51,24 @@ LOGIN_LOADING = ButtonWrapper(
         button=(34, 682, 60, 707),
     ),
 )
+OCR_YEAR = ButtonWrapper(
+    name='OCR_YEAR',
+    jp=None,
+    en=Button(
+        file='./assets/en/login/OCR_YEAR.png',
+        area=(377, 689, 414, 701),
+        search=(357, 669, 434, 720),
+        color=(160, 157, 158),
+        button=(377, 689, 414, 701),
+    ),
+    zht=Button(
+        file='./assets/zht/login/OCR_YEAR.png',
+        area=(377, 689, 414, 701),
+        search=(357, 669, 434, 720),
+        color=(160, 157, 158),
+        button=(377, 689, 414, 701),
+    ),
+)
 SURVEY = ButtonWrapper(
     name='SURVEY',
     jp=None,

tasks/login/login.py

@@ -1,11 +1,11 @@
 from module.base.timer import Timer
 from module.base.decorator import Config
+from module.ocr.ocr import Digit
 from module.exception import GameNotRunningError
 from module.logger import logger
 from tasks.base.page import page_main
 from tasks.base.ui import UI
-from tasks.login.assets.assets_login import LOGIN_CONFIRM, LOGIN_LOADING, UPDATE, SURVEY
-from tasks.base.assets.assets_base_page import MAIN_GO_TO_MAIL
+from tasks.login.assets.assets_login import LOGIN_CONFIRM, LOGIN_LOADING, UPDATE, SURVEY, OCR_YEAR
 
 
 class Login(UI):
@@ -94,10 +94,8 @@ class Login(UI):
                 continue
             if self.ui_additional():
                 continue
-            # press emulator back button when random popup in main
-            if self.appear(MAIN_GO_TO_MAIL) and not self.match_color(MAIN_GO_TO_MAIL):
+            if not self.appear_trademark_year():
                 self.device.u2.press("back")
-                continue
 
         return True