|
|
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
基础按钮委托类
|
|
|
|
|
|
提供通用的按钮索引获取和数据访问方法
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|
|
from PyQt5.QtWidgets import QItemDelegate, QMessageBox
|
|
|
|
|
|
from PyQt5.QtCore import Qt
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class BaseButtonDelegate(QItemDelegate):
|
|
|
|
|
|
"""基础按钮委托类，提供通用的索引处理方法"""
|
|
|
|
|
|
|
|
|
|
|
|
def __init__(self, parent=None):
|
|
|
|
|
|
super(BaseButtonDelegate, self).__init__(parent)
|
|
|
|
|
|
|
|
|
|
|
|
def _get_button_row_indices(self, button):
|
|
|
|
|
|
"""
|
|
|
|
|
|
获取按钮所在的视图行和源模型行索引
|
|
|
|
|
|
返回: (view_row, source_row) 或 (None, None) 如果找不到
|
|
|
|
|
|
"""
|
|
|
|
|
|
table_view = self.parent()
|
|
|
|
|
|
|
|
|
|
|
|
# 获取操作列索引（最后一列）
|
|
|
|
|
|
if hasattr(table_view, 'proxy') and table_view.proxy:
|
|
|
|
|
|
operation_col = table_view.proxy.columnCount() - 1
|
|
|
|
|
|
row_count = table_view.proxy.rowCount()
|
|
|
|
|
|
else:
|
|
|
|
|
|
operation_col = table_view.model.columnCount() - 1
|
|
|
|
|
|
row_count = table_view.model.rowCount()
|
|
|
|
|
|
|
|
|
|
|
|
# 查找按钮所在的行
|
|
|
|
|
|
for row in range(row_count):
|
|
|
|
|
|
if hasattr(table_view, 'proxy') and table_view.proxy:
|
|
|
|
|
|
index = table_view.proxy.index(row, operation_col)
|
|
|
|
|
|
else:
|
|
|
|
|
|
index = table_view.model.index(row, operation_col)
|
|
|
|
|
|
|
|
|
|
|
|
widget = table_view.indexWidget(index)
|
|
|
|
|
|
if widget and widget.layout():
|
|
|
|
|
|
for i in range(widget.layout().count()):
|
|
|
|
|
|
if widget.layout().itemAt(i).widget() == button:
|
|
|
|
|
|
view_row = row
|
|
|
|
|
|
|
|
|
|
|
|
# 转换为源模型索引
|
|
|
|
|
|
if hasattr(table_view, 'proxy') and table_view.proxy:
|
|
|
|
|
|
proxy_index = table_view.proxy.index(view_row, 0)
|
|
|
|
|
|
source_index = table_view.proxy.mapToSource(proxy_index)
|
|
|
|
|
|
source_row = source_index.row()
|
|
|
|
|
|
else:
|
|
|
|
|
|
source_row = view_row
|
|
|
|
|
|
|
|
|
|
|
|
return view_row, source_row
|
|
|
|
|
|
|
|
|
|
|
|
return None, None
|
|
|
|
|
|
|
|
|
|
|
|
def _get_view_data(self, view_row, column):
|
|
|
|
|
|
"""
|
|
|
|
|
|
从视图中获取指定行列的数据
|
|
|
|
|
|
"""
|
|
|
|
|
|
table_view = self.parent()
|
|
|
|
|
|
if hasattr(table_view, 'proxy') and table_view.proxy:
|
|
|
|
|
|
index = table_view.proxy.index(view_row, column)
|
|
|
|
|
|
return table_view.proxy.data(index, Qt.DisplayRole)
|
|
|
|
|
|
else:
|
|
|
|
|
|
return table_view.model.datas[view_row][column]
|
|
|
|
|
|
|
|
|
|
|
|
def _validate_button_indices(self, button):
|
|
|
|
|
|
"""
|
|
|
|
|
|
验证并获取按钮索引，如果失败则显示错误消息
|
|
|
|
|
|
返回: (view_row, source_row) 或 (None, None)
|
|
|
|
|
|
"""
|
|
|
|
|
|
view_row, source_row = self._get_button_row_indices(button)
|
|
|
|
|
|
if view_row is None:
|
|
|
|
|
|
QMessageBox.warning(self.parent(), '错误', '无法确定按钮所在行', QMessageBox.Yes)
|
|
|
|
|
|
return None, None
|
|
|
|
|
|
return view_row, source_row
|
