From 6ef18ed8c0c4cac8c814c843dabb146c764dbfc5 Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Sat, 18 Nov 2023 19:00:24 +0800
Subject: [PATCH] feat(login): add survey page handle

---
 tasks/login/login.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/tasks/login/login.py b/tasks/login/login.py
index 987c4bf..fa3f867 100644
--- a/tasks/login/login.py
+++ b/tasks/login/login.py
@@ -1,9 +1,10 @@
 from module.base.timer import Timer
+from module.base.decorator import Config
 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
+from tasks.login.assets.assets_login import LOGIN_CONFIRM, LOGIN_LOADING, UPDATE, SURVEY
 
 
 class Login(UI):
@@ -29,6 +30,16 @@ class Login(UI):
                             return True
                         return False
 
+        @Config.when(Emulator_GameLanguage='jp')
+        def _handle_survey():
+            pass
+
+        @Config.when(Emulator_GameLanguage='en')
+        def _handle_survey():
+            if self.appear_then_click(SURVEY):
+                return True
+            return False
+
         logger.hr('App login')
         orientation_timer = Timer(5)
         startup_timer = Timer(5).start()
@@ -78,6 +89,8 @@ class Login(UI):
             #     continue
             if self.appear_then_click(UPDATE):
                 continue
+            if _handle_survey():
+                continue
             if self.ui_additional():
                 continue