From 628fd5c19a38c1d7388916619dbe8234b3f72736 Mon Sep 17 00:00:00 2001
From: RedDeadDepresso <94017243+RedDeadDepresso@users.noreply.github.com>
Date: Mon, 8 Jan 2024 20:22:14 +0000
Subject: [PATCH] fix: login

---
 tasks/base/ui.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/tasks/base/ui.py b/tasks/base/ui.py
index e9ee337..651b90a 100644
--- a/tasks/base/ui.py
+++ b/tasks/base/ui.py
@@ -7,7 +7,7 @@ from module.ocr.ocr import Ocr, Digit
 from tasks.base.main_page import MainPage
 from tasks.base.page import Page, page_main
 from tasks.login.assets.assets_login import LOGIN_LOADING, OCR_YEAR
-#from tasks.base.assets.assets_base_page import BACK
+from tasks.base.assets.assets_base_page import BACK
 
 
 class UI(MainPage):
@@ -61,6 +61,7 @@ class UI(MainPage):
 
         timeout = Timer(10, count=20).start()
         back_timer = Timer(0.5, count=2)
+        u2_back = True
         while 1:
             if skip_first_screenshot:
                 skip_first_screenshot = False
@@ -95,8 +96,12 @@ class UI(MainPage):
                 continue
             if back_timer.reached_and_reset():
                 logger.info("Unknown page, try to back")
-                #self.device.click(BACK)
-                self.device.u2.press("back")
+                if u2_back:
+                    self.device.u2.press("back")
+                    u2_back = False
+                else:
+                    self.device.click(BACK)
+                    u2_back = True
 
             app_check()
             minicap_check()