文件分离函数类，增加提示信息

QT5_Project/Shared_CODE/FaceRecognitionProtocol.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import struct
 from dataclasses import dataclass
 from typing import Optional, Tuple, List
+from QT5_Project.Shared_CODE.DialogInform import DialogInform
 
 SYNC = b"\xEF\xAA"
 
@@ -296,71 +297,86 @@ def build_enroll_with_photo_chunk(seq: int, chunk: bytes) -> bytes:
 # ---- Parsers (Reply / Note / Image) ----
 def parse_reply(data: bytes) -> dict:
     if len(data) < 2:
-        return {"type":"REPLY","error":"short"}
+        return {"type": "REPLY", "error": "short"}
+
     mid = data[0]
     result = data[1]
     rest = data[2:]
 
-    # 初始化 info 字典
+    # 通用字段
     info = {
-        "type":"REPLY",
+        "type": "REPLY",
         "mid": mid,
         "mid_name": CMD_NAMES.get(mid, f"0x{mid:02X}"),
         "result": result,
-        "result_name": RESULT_NAMES.get(result, f"0x{result:02X}")
+        "result_name": RESULT_NAMES.get(result, f"0x{result:02X}"),
+        "ok": (result == 0),  # 成功标志
     }
 
+    # ========== 分支解析 ==========
     if mid == CMD_VERIFY and len(rest) >= 36:
-        uid = (rest[0]<<8)|rest[1]
+        uid = (rest[0] << 8) | rest[1]
         name = rest[2:34].rstrip(b"\x00").decode("utf-8", errors="ignore")
         admin = rest[34]
         unlock = rest[35]
-        info.update({"user_id": uid, "user_name": name, "admin": admin, "unlock_status": unlock})
+        info.update({
+            "user_id": uid,
+            "user_name": name,
+            "admin": admin,
+            "unlock_status": unlock,
+        })
 
     elif mid in (CMD_ENROLL, 0x1D, CMD_ENROLL_ITG) and len(rest) >= 3:
-        uid = (rest[0]<<8)|rest[1]
+        uid = (rest[0] << 8) | rest[1]
         face_dir = rest[2]
         info.update({"user_id": uid, "face_direction": face_dir})
 
     elif mid == CMD_GET_STATUS and len(rest) >= 1:
         status = rest[0]
-        info.update({"status": status, "status_name": {
-            0: "空闲",
-            1: "录入中",
-            2: "验证中"
-        }.get(status, f"0x{status:02X}")})
+        status_map = {0: "空闲", 1: "录入中", 2: "验证中"}
+        info.update({"status": status, "status_name": status_map.get(status, f"0x{status:02X}")})
 
     elif mid == CMD_GET_USER_INFO and len(rest) >= 35:
-        uid = (rest[0]<<8)|rest[1]
+        uid = (rest[0] << 8) | rest[1]
         name = rest[2:34].decode("ascii", errors="ignore")
         admin = rest[34]
         info.update({"user_id": uid, "user_name": name, "admin": admin})
 
     elif mid == CMD_GET_ALL_USERID and len(rest) >= 1:
         n = rest[0]
-        ids = [(rest[i]<<8)|rest[i+1] for i in range(1, 1+2*n, 2) if i+1 < len(rest)]
+        ids = [(rest[i] << 8) | rest[i + 1] for i in range(1, 1 + 2 * n, 2) if i + 1 < len(rest)]
         info.update({"count": n, "user_ids": ids})
 
     elif mid == CMD_GET_VERSION:
         info["version_str"] = rest.decode("ascii", errors="ignore")
-    
+
     elif mid == CMD_LED_CONTROL and len(rest) >= 1:
         led_state = rest[0]
-        info.update({"led_state": led_state, "led_state_name": {
-            0: "灭",
-            1: "亮"
-        }.get(led_state, f"0x{led_state:02X}")})
+        led_map = {0: "灭", 1: "亮"}
+        info.update({"led_state": led_state, "led_state_name": led_map.get(led_state, f"0x{led_state:02X}")})
 
     elif mid == CMD_ENROLL_WITH_PHOTO:
         if len(rest) >= 2:
-            seq = (rest[0]<<8)|rest[1]
+            seq = (rest[0] << 8) | rest[1]
             info["seq"] = seq
         if len(rest) >= 6:
-            uid = (rest[2]<<8)|rest[3]
+            uid = (rest[2] << 8) | rest[3]
             info["user_id"] = uid
 
+    # 自动生成提示消息
+    if info.get("ok"):
+        info["message"] = f"{info['mid_name']} 成功"
+        inform_box : DialogInform = DialogInform()
+        inform_box.information("提示", f"{info['mid_name']} 成功")
+    else:
+        info["message"] = f"{info['mid_name']} 失败: {info['result_name']}"
+        inform_box : DialogInform = DialogInform()
+        inform_box.information("提示", f"{info['mid_name']} 失败: {info['result_name']}")
+
+
     return info
 
+
 def parse_note(data: bytes) -> dict:
     if len(data) < 1:
         return {"type":"NOTE","error":"short"}