From 728b020db680a3d7c42a4f1f055b436fa01b27f8 Mon Sep 17 00:00:00 2001
From: marques <20172333133@m.scnu.edu.cn>
Date: Sun, 31 Aug 2025 20:00:48 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96SA=E6=89=93=E6=A0=87=E7=95=8C?=
 =?UTF-8?q?=E9=9D=A2=E5=89=8D=E6=99=AF=E8=89=B2=E8=AE=BE=E7=BD=AE=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91=EF=BC=8C=E7=AE=80=E5=8C=96=E6=9D=A1=E4=BB=B6=E5=88=A4?=
 =?UTF-8?q?=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 func/Module_SA_label_v2.py | 64 ++++++++++++++++++--------------------
 1 file changed, 31 insertions(+), 33 deletions(-)

diff --git a/func/Module_SA_label_v2.py b/func/Module_SA_label_v2.py
index ff4752d..558c945 100644
--- a/func/Module_SA_label_v2.py
+++ b/func/Module_SA_label_v2.py
@@ -526,41 +526,39 @@ class DataFrameModel(QAbstractTableModel):
             return str(value)
 
         # 当isLabeled列为1时，整行前景变色
-        if role == Qt.ForegroundRole:
-            if "isLabeled" in self._dataframe.columns:
-                try:
-                    print(self._showdata.iloc[row]["Index"])
+        if role == Qt.ForegroundRole and "isLabeled" in self._dataframe.columns:
+            try:
+                print(self._showdata.iloc[row]["Index"])
 
-                    event_index = self._showdata.iloc[row]["Index"]
-                    is_labeled_value = self._dataframe[self._dataframe["Index"] == event_index]["isLabeled"]
-                    if is_labeled_value.empty:
-                        return None
-                    is_labeled_value = is_labeled_value.values[0]
-
-                    if is_labeled_value == 1:
-                        # remark 返回蓝色 score 1 返回绿色， 2返回橙色  3返回灰色 -1返回红色
-                        # 获取此行的remark
-                        remark = not self._dataframe[self._dataframe["Index"] == event_index]["remark"].values[0]
-                        score = self._dataframe[self._dataframe["Index"] == event_index]["score"].values[0]
-                        if remark == "待讨论":
-                            return QColor(0, 0, 255)
-                        elif score == 1:
-                            return QColor(0, 255, 0)
-                        elif score == 2:
-                            return QColor(255, 165, 0)
-                        elif score == 3:
-                            return QColor(128, 128, 128)
-                        elif score == -1:
-                            return QColor(255, 0, 0)
-                        else:
-                            # 返回黑色
-                            return QColor(0, 0, 0)
-
-                    else:
-                        # 返回灰色
-                        return QColor(0, 0, 0)
-                except Exception as e:
+                event_index = self._showdata.iloc[row]["Index"]
+                is_labeled_value = self._dataframe[self._dataframe["Index"] == event_index]["isLabeled"]
+                if is_labeled_value.empty:
                     return None
+                is_labeled_value = is_labeled_value.values[0]
+
+                if is_labeled_value != 1:
+                    return QColor(0, 0, 0)
+
+                # remark 返回蓝色 score 1 返回绿色， 2返回橙色  3返回灰色 -1返回红色
+                # 获取此行的remark
+                remark = not self._dataframe[self._dataframe["Index"] == event_index]["remark"].values[0]
+                score = self._dataframe[self._dataframe["Index"] == event_index]["score"].values[0]
+                if remark == "待讨论":
+                    return QColor(0, 0, 255)
+                elif score == 1:
+                    return QColor(0, 255, 0)
+                elif score == 2:
+                    return QColor(255, 165, 0)
+                elif score == 3:
+                    return QColor(128, 128, 128)
+                elif score == -1:
+                    return QColor(255, 0, 0)
+                else:
+                    # 返回黑色
+                        return QColor(0, 0, 0)
+
+            except Exception as e:
+                return None
 
         return None