81 lines
3.5 KiB
Python
81 lines
3.5 KiB
Python
# StatusStyleDelegate.py
|
|
from PySide6.QtWidgets import QStyledItemDelegate
|
|
from PySide6.QtGui import QColor, QFont, QBrush
|
|
from PySide6.QtCore import Qt, QTimer
|
|
from PySide6.QtGui import QPalette
|
|
|
|
class StatusStyleDelegate(QStyledItemDelegate):
|
|
def __init__(self, parent=None, table_view=None):
|
|
super().__init__(parent)
|
|
self.flash_cells = {} # 存储需要闪烁的单元格 {(row, col): timer}
|
|
self.flash_duration = 500 # 闪烁持续时间(毫秒)
|
|
self.updated_cells = set() # 存储需要高亮的单元格 (row, col)
|
|
self.table_view = table_view # 保存 QTableView 实例
|
|
|
|
def flash_cell(self, row, column):
|
|
# 如果单元格已经在闪烁，重置计时器
|
|
if (row, column) in self.flash_cells:
|
|
self.flash_cells[(row, column)].stop()
|
|
|
|
# 创建新的计时器
|
|
timer = QTimer()
|
|
timer.setSingleShot(True)
|
|
timer.timeout.connect(lambda: self.end_flash(row, column))
|
|
self.flash_cells[(row, column)] = timer
|
|
timer.start(500)
|
|
|
|
self.updated_cells.add((row, column))
|
|
if self.table_view:
|
|
self.table_view.viewport().update()
|
|
|
|
def end_flash(self, row, column):
|
|
"""1秒后清除高亮"""
|
|
if (row, column) in self.flash_cells:
|
|
del self.flash_cells[(row, column)]
|
|
self.updated_cells.discard((row, column))
|
|
if self.table_view:
|
|
self.table_view.viewport().update()
|
|
|
|
def initStyleOption(self, option, index):
|
|
super().initStyleOption(option, index)
|
|
|
|
# 检查是否需要闪烁
|
|
row = index.row()
|
|
col = index.column()
|
|
if (row, col) in self.updated_cells:
|
|
option.backgroundBrush = QBrush(QColor("yellow"))
|
|
|
|
model = index.model()
|
|
status_index = model.index(index.row(), 7)
|
|
status = model.data(status_index, Qt.ItemDataRole.DisplayRole)
|
|
|
|
if status == "已触发":
|
|
option.font = QFont("Arial", 14, QFont.Weight.Bold)
|
|
option.palette.setColor(QPalette.ColorRole.Text, QColor('red'))
|
|
option.backgroundBrush = QBrush(QColor('lightcoral')) # 背景色
|
|
elif status == "错买":
|
|
option.font = QFont("Arial", 14, QFont.Weight.Bold)
|
|
option.palette.setColor(QPalette.ColorRole.Text, QColor('green')) # 文字颜色
|
|
option.backgroundBrush = QBrush(QColor('lightgreen')) # 背景色
|
|
elif status == "监控中":
|
|
option.font = QFont("Arial", 12, QFont.Weight.Normal)
|
|
option.palette.setColor(QPalette.ColorRole.Text, QColor('black'))
|
|
option.backgroundBrush = QBrush(QColor('white'))
|
|
else:
|
|
option.font = QFont("Arial", 12, QFont.Weight.Normal)
|
|
option.palette.setColor(QPalette.ColorRole.Text, QColor('black'))
|
|
option.backgroundBrush = QBrush(QColor('white'))
|
|
|
|
# 新增对获利列的处理
|
|
if index.column() == 6: # 检查是否为获利列
|
|
option.font = QFont(option.font().family(), option.font().pointSize(), QFont.Weight.Bold)
|
|
profit_text = model.data(index, Qt.ItemDataRole.DisplayRole)
|
|
try:
|
|
profit_value = float(profit_text.replace('%', '')) # 移除百分号再转换
|
|
if profit_value > 0:
|
|
option.palette.setColor(QPalette.ColorRole.Text, QColor('red'))
|
|
elif profit_value < 0:
|
|
option.palette.setColor(QPalette.ColorRole.Text, QColor('green'))
|
|
except ValueError:
|
|
pass # 如果转换失败，保持默认颜色
|