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