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

This commit is contained in:
marques
2025-08-31 20:00:48 +08:00
parent a7efcd0a85
commit 728b020db6

View File

@ -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