feat: 支持富士奇石的表演秀模式 (#173)

Author: ZJChe

module/umamusume/asset/template.py

@@ -4,6 +4,8 @@
 UMAMUSUME_BTN_TEMPLATE_PATH = "/umamusume/btn"
 UMAMUSUME_REF_TEMPLATE_PATH = "/umamusume/ref"
 UMAMUSUME_UMA_ICON_TEMPLATE_PATH = "/umamusume/uma_icon"
+# 限时: 富士奇石的表演秀
+UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH = "/umamusume/ui/fujikiseki_show"
 
 UI_INFO = Template("INFO", UMAMUSUME_UI_TEMPLATE_PATH)
 UI_MAIN_MENU = Template("MAIN_MENU", UMAMUSUME_UI_TEMPLATE_PATH)
@@ -124,4 +126,12 @@
 UMA_ICON_URA_NPC_2 = Template("URA_NPC_2", UMAMUSUME_UMA_ICON_TEMPLATE_PATH)
 UMA_ICON_FRIEND_1 = Template("FRIEND_1", UMAMUSUME_UMA_ICON_TEMPLATE_PATH)
 
-
+# 限时: 富士奇石的表演秀
+FUJIKISEKI_SHOW_DIFFICULTY_LOCKED = Template("DIFFICULTY_LOCKED", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH)
+FUJIKISEKI_SHOW_DIFFICULTY = [Template("DIFFICULTY_1", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH),
+                              Template("DIFFICULTY_2", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH),
+                              Template("DIFFICULTY_3", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH),
+                              Template("DIFFICULTY_4", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH),
+                              Template("DIFFICULTY_5", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH)]
+UI_FUJIKISEKI_SHOW_MODE_RESULT_1 = Template("SHOW_MODE_RESULT_1", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH)
+UI_FUJIKISEKI_SHOW_MODE_RESULT_2 = Template("SHOW_MODE_RESULT_2", UMAMUSUME_UI_FUJIKISEKI_SHOW_TEMPLATE_PATH)

module/umamusume/asset/ui.py

@@ -58,6 +58,10 @@
 ACTIVITY_RESULT = UI("ACTIVITY_RESULT", [template.UI_ACTIVITY_RESULT], [])
 ACTIVITY_REWARD = UI("ACTIVITY_REWARD", [template.UI_ACTIVITY_REWARD], [])
 
+# 限时: 富士奇石的表演秀
+FUJIKISEKI_SHOW_MODE_RESULT_1 = UI("FUJIKISEKI_SHOW_MODE_RESULT_1", [template.UI_FUJIKISEKI_SHOW_MODE_RESULT_1], [])
+FUJIKISEKI_SHOW_MODE_RESULT_2 = UI("FUJIKISEKI_SHOW_MODE_RESULT_2", [template.UI_FUJIKISEKI_SHOW_MODE_RESULT_2], [])
+
 scan_ui_list = [MAIN_MENU,
                 CULTIVATE_SCENARIO_SELECT, CULTIVATE_UMAMUSUME_SELECT, CULTIVATE_EXTEND_UMAMUSUME_SELECT,
                 CULTIVATE_SUPPORT_CARD_SELECT, CULTIVATE_FOLLOW_SUPPORT_CARD_SELECT, CULTIVATE_FINAL_CHECK, INFO,
@@ -70,4 +74,4 @@
                 GOAL_ACHIEVED, NEXT_GOAL, ALL_GOAL_ACHIEVED, CULTIVATE_FINISH, GOAL_FAILED,
                 FACTOR_RECEIVE, HISTORICAL_RATING_UPDATE, SCENARIO_RATING_UPDATE,
                 CULTIVATE_EVENT_UMAMUSUME, CULTIVATE_EVENT_SUPPORT_CARD, CULTIVATE_EVENT_SCENARIO, ACTIVITY_RESULT,
-                ACTIVITY_REWARD]
+                ACTIVITY_REWARD, FUJIKISEKI_SHOW_MODE_RESULT_1, FUJIKISEKI_SHOW_MODE_RESULT_2]

module/umamusume/manifest.py

@@ -56,7 +56,9 @@
         CULTIVATE_URA_RACE_2: script_cultivate_goal_race,
         CULTIVATE_URA_RACE_3: script_cultivate_goal_race,
         ACTIVITY_RESULT: script_cultivate_result,
-        ACTIVITY_REWARD: script_cultivate_result
+        ACTIVITY_REWARD: script_cultivate_result,
+        FUJIKISEKI_SHOW_MODE_RESULT_1: script_fujikiseki_show_result_1,
+        FUJIKISEKI_SHOW_MODE_RESULT_2: script_fujikiseki_show_result_2
     }
 }
 

module/umamusume/script/cultivate_task/cultivate.py

@@ -1,5 +1,6 @@
 import json
 import time
+import threading
 
 import numpy as np
 
@@ -107,6 +108,14 @@ def script_cultivate_training_select(ctx: UmamusumeContext):
         parse_training_result(ctx, img, train_type)
         parse_training_support_card(ctx, img, train_type)
         viewed = train_type.value
+
+        def _parse_training_in_thread(ctx, img, train_type):
+            """Helper function to run parsing in a separate thread."""
+            parse_training_result(ctx, img, train_type)
+            parse_training_support_card(ctx, img, train_type)
+        
+        threads :list[threading.Thread] = []
+
         for i in range(5):
             if i != (viewed - 1):
                 retry = 0
@@ -121,8 +130,15 @@ def script_cultivate_training_select(ctx: UmamusumeContext):
                     retry += 1
                 if retry == max_retry:
                     return
-                parse_training_result(ctx, img, TrainingType(i + 1))
-                parse_training_support_card(ctx, img, TrainingType(i + 1))
+                
+                thread = threading.Thread(target=_parse_training_in_thread,
+                                          args=(ctx, img, TrainingType(i + 1)))
+                threads.append(thread)
+                thread.start()
+
+        for thread in threads:
+            thread.join()
+
         ctx.cultivate_detail.turn_info.parse_train_info_finish = True
     if not ctx.cultivate_detail.turn_info.parse_main_menu_finish:
         ctx.ctrl.click_by_point(RETURN_TO_CULTIVATE_MAIN_MENU)
@@ -319,6 +335,12 @@ def script_cultivate_extend(ctx: UmamusumeContext):
 def script_cultivate_result(ctx: UmamusumeContext):
     ctx.ctrl.click_by_point(CULTIVATE_RESULT_CONFIRM)
 
+# 限时: 富士奇石的表演秀
+def script_fujikiseki_show_result_1(ctx: UmamusumeContext):
+    ctx.ctrl.click(360, 1180, "确认富士奇石表演秀模式结果")
+
+def script_fujikiseki_show_result_2(ctx: UmamusumeContext):
+    ctx.ctrl.click(360, 1120, "确认富士奇石表演秀模式结果")
 
 # 1.878s 2s 0.649s
 def script_cultivate_catch_doll(ctx: UmamusumeContext):

module/umamusume/script/cultivate_task/info.py

@@ -1,4 +1,6 @@
 import time
+from datetime import datetime
+import pytz
 
 import cv2
 
@@ -44,6 +46,8 @@
     "回复训练值",
     "选择养成难度",
     "确定因子确认",
+    # 限时: 富士奇石的表演秀
+    "解锁新难度"
 ]
 
 
@@ -143,7 +147,39 @@ def script_info(ctx: UmamusumeContext):
                 ctx.ctrl.click_by_point(USE_TP_DRINK_CONFIRM)
             elif image_match(ctx.ctrl.get_screen(to_gray=True), REF_RECOVER_TP_3).find_match:
                 ctx.ctrl.click_by_point(USE_TP_DRINK_RESULT_CLOSE)
+        if title_text == TITLE[28]:
+            # 限时: 富士奇石的表演秀
+            # 目前似乎只有这里用到了 "选择养成难度，如果以后有别的的话需要调整代码结构"
+            beijing_tz = pytz.timezone('Asia/Shanghai')
+            cutoff_time = beijing_tz.localize(datetime(2025, 7, 13, 11, 59))
+            current_time_beijing = datetime.now(beijing_tz)
+
+            if current_time_beijing <= cutoff_time:
+                if ctx.task.detail.fujikiseki_show_mode == False:
+                    ctx.ctrl.click(360, 300, "选择普通模式")
+                else :
+                    ctx.ctrl.click(360, 500, "选择富士奇石的表演秀模式")
+                    match = False
+                    for i in range(5):
+                        screen = ctx.ctrl.get_screen(to_gray=True)
+                        if ((not image_match(screen, FUJIKISEKI_SHOW_DIFFICULTY_LOCKED).find_match) 
+                            and image_match(screen, FUJIKISEKI_SHOW_DIFFICULTY[ctx.task.detail.fujikiseki_show_difficulty-1]).find_match):
+                            log.info(f"选择难度{ctx.task.detail.fujikiseki_show_difficulty}")
+                            match = True
+                            break
+                        ctx.ctrl.click(675, 800, "切换至下一个难度")
+                        time.sleep(1)
+                    if not match:
+                        log.error(f"选择的难度{ctx.task.detail.fujikiseki_show_difficulty}尚未解锁, 请先游玩低难度模式!")
+                        ctx.task.end_task(TaskStatus.TASK_STATUS_FAILED, UEndTaskReason.DIFFICULTY_LOCKED)
+                        return
+                        
+                ctx.ctrl.click(520, 1180, "")
+            
         if title_text == TITLE[29]:
             ctx.ctrl.click_by_point(CULTIVATE_RESULT_DIVISOR_CONFIRM)
+        if title_text == TITLE[30]:
+            # 限时: 富士奇石的表演秀
+            ctx.ctrl.click(360, 850, "确认解锁新难度")
         time.sleep(1)
 

module/umamusume/task.py

@@ -17,10 +17,14 @@ class TaskDetail:
     allow_recover_tp: bool
     cultivate_progress_info: dict
     extra_weight: list
+    # 限时: 富士奇石的表演秀
+    fujikiseki_show_mode: bool
+    fujikiseki_show_difficulty: int
 
 
 class EndTaskReason(Enum):
     TP_NOT_ENOUGH = "训练值不足"
+    DIFFICULTY_LOCKED = "难度未解锁"
 
 
 class UmamusumeTask(Task):
@@ -58,6 +62,9 @@ def build_task(task_execute_mode: TaskExecuteMode, task_type: int,
     td.extra_weight = attachment_data['extra_weight']
     td.cultivate_result = {}
     # td.scenario_name = attachment_data['scenario_name']
+    # 限时: 富士奇石的表演秀
+    td.fujikiseki_show_mode = attachment_data['fujikiseki_show_mode']
+    td.fujikiseki_show_difficulty = attachment_data['fujikiseki_show_difficulty']
     ut.detail = td
     return ut