优化SA打标界面前景色设置逻辑，简化条件判断

func/Module_SA_label_v2.py

@@ -526,41 +526,39 @@ class DataFrameModel(QAbstractTableModel):
             return str(value)
 
         # 当isLabeled列为1时，整行前景变色
-        if role == Qt.ForegroundRole:
+        if role == Qt.ForegroundRole and "isLabeled" in self._dataframe.columns:
-            if "isLabeled" in self._dataframe.columns:
+            try:
-                try:
+                print(self._showdata.iloc[row]["Index"])
-                    print(self._showdata.iloc[row]["Index"])
 
-                    event_index = self._showdata.iloc[row]["Index"]
+                event_index = self._showdata.iloc[row]["Index"]
-                    is_labeled_value = self._dataframe[self._dataframe["Index"] == event_index]["isLabeled"]
+                is_labeled_value = self._dataframe[self._dataframe["Index"] == event_index]["isLabeled"]
-                    if is_labeled_value.empty:
+                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:
                     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