From 16ef6e3daee64e7021d7394632d1ea6986d05c2e Mon Sep 17 00:00:00 2001
From: guoh064 <50830808+guoh064@users.noreply.github.com>
Date: Fri, 14 Aug 2026 22:09:50 +0800
Subject: [PATCH] Fix: missing ime check before equip code export

---
 module/equipment/equipment_code.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/module/equipment/equipment_code.py b/module/equipment/equipment_code.py
index ff00194e0..bd60d5ea2 100644
--- a/module/equipment/equipment_code.py
+++ b/module/equipment/equipment_code.py
@@ -162,6 +162,9 @@ class EquipmentCodeHandler(StorageHandler):
 
     def set_fastinput_ime(self):
         d = self.device.u2
+        name, _ = d.current_ime()
+        if name == self.FASTINPUT_IME:
+            return
         failed = False
         try:
             for command in ('enable', 'set'):
@@ -185,12 +188,11 @@ class EquipmentCodeHandler(StorageHandler):
         click_timer = Timer(1, count=3)
         for _ in self.loop():
             name, shown = d.current_ime()
+            if name != self.FASTINPUT_IME:
+                self.set_fastinput_ime()
+                continue
             if shown:
-                if name != self.FASTINPUT_IME:
-                    self.set_fastinput_ime()
-                    continue
-                else:
-                    break
+                break
             if click_timer.reached_and_reset():
                 self.device.click(EQUIPMENT_CODE_TEXTBOX)
         else:
@@ -244,6 +246,7 @@ class EquipmentCodeHandler(StorageHandler):
     def _code_export(self):
         self.handle_info_bar()
         d = self.device.u2
+        self.set_fastinput_ime()
         for _ in self.loop(timeout=10):
             if self.info_bar_count():
                 break