python tableview绑定字典_PyQt5之 QTableView 添加复选框(自定义委托)

import sys

from untitled import Ui_Form

from PyQt5.QtWidgets import QApplication, QWidget, QStyleOptionButton, QAbstractItemDelegate, QStyle, QCheckBox, QStyledItemDelegate, QStyleOptionViewItem, QItemDelegate

from PyQt5.QtCore import Qt, QAbstractTableModel, QModelIndex, QVariant, QThread, pyqtSignal, QEvent, QRect

from PyQt5.QtGui import QColor, QMouseEvent

class WorkThread(QThread):

scrollBottomSignal = pyqtSignal()

addDataSignal = pyqtSignal()

def __init__(self, model):

super(WorkThread, self).__init__()

self.model = model

self.run_flag = True

def run(self):

while self.run_flag:

self.addDataSignal.emit()

self.scrollBottomSignal.emit()

self.usleep(1) # 不加延迟界面会卡顿。

def stop(self):

self.run_flag = False

class MyTableModel(QAbstractTableModel):

def __init__(self):

super(MyTableModel, self).__init__()

self._data = [] # 要显示的数据

self._headers = ['选项', '姓名', '年龄', '性别'] # 表头

def rowCount(self, parent=QModelIndex()):

"""

返回行数量。

"""

return len(self._data)

def columnCount(self, parent=QModelIndex()):

"""

返回列数量。

"""

return len(self._headers)

def insertRows(self, row, count, parent):

"""

插入行。

:param row: 插入起始行。

:param count: 插入行数量。

:param parent:

:return:

"""

self.beginInsertRows(QModelIndex(), row, row + count - 1)

for i in range(count):

self._data.insert(row, [Qt.Checked, 'CZ', '25', '男'])

self.endInsertRows()

return True

def removeRows(self, row, count, parent):

self.beginRemoveRows(QModelIndex(), 0, row + count - 1)

for i in range(count):

self._data.pop(row + count - 1 - i) # 倒着删

self.endRemoveRows()

def clearView(self):

self.removeRows(0, self.rowCount(), QModelIndex())

def addData(self):

self.insertRows(self.rowCount(), 1, QModelIndex())

def headerData(self, section, orientation, role):

"""

设置表头。

"""

if role == Qt.DisplayRole and orientation == Qt.Horizontal: # 限定只更改行表头

return self._headers[section]

def data(self, index, role=Qt.DisplayRole):

if not index.isValid() or not 0 <= index.row() < self.rowCount():

return QVariant()

row = index.row()

col = index.column()

if role == Qt.DisplayRole:

if col != 0:

return str(self._data[row][col]) # 数据

elif role == Qt.TextColorRole:

return QColor(Qt.red)

elif Qt.UserRole:

if col == 0:

return self._data[row][col]

return QVariant()

def setData(self, index, value, role):

if not index.isValid() or not 0 <= index.row() < self.rowCount():

return False

col = index.column()

row = index.row()

if role == Qt.UserRole or role == Qt.CheckStateRole:

if col == 0:

self._data[row][col] = value

self.layoutChanged.emit()

return True

return False

def flags(self, index):

flags = Qt.ItemFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)

return flags

class MyDelegate(QStyledItemDelegate):

def __init__(self):

super(MyDelegate, self).__init__()

def paint(self, painter, option, index):

if index.column() == 0:

data = index.model().data(index, Qt.UserRole)

ckbtn = QStyleOptionButton()

ckbtn.state = QStyle.State_On if data == Qt.Checked else QStyle.State_Off

ckbtn.state |= QStyle.State_Enabled

ckbtn.rect = option.rect

QApplication.style().drawControl(QStyle.CE_CheckBox, ckbtn, painter)

else:

QStyledItemDelegate.paint(self, painter, option, index)

def editorEvent(self, event, model, option, index):

if event.type() == QEvent.KeyPress:

return False

if event.type() == QEvent.MouseButtonRelease:

event = QMouseEvent(event) # 类型强转

if event.button() == Qt.LeftButton:

value = index.data(Qt.UserRole)

if value == Qt.Checked:

rev_value = Qt.Unchecked

else:

rev_value = Qt.Checked

return model.setData(index, rev_value, Qt.UserRole)

else:

return False

else:

return False

class MainUI(QWidget, Ui_Form):

def __init__(self):

super(MainUI, self).__init__()

self.setupUi(self)

self.workThread = None

self.pushButton.clicked.connect(self.buttonClickedStart)

self.pushButton_2.clicked.connect(self.buttonClickedStop)

self.pushButton_3.clicked.connect(self.buttonClickedClear)

self.model = MyTableModel()

self.myDelegate = MyDelegate()

self.tableView.setItemDelegate(self.myDelegate)

self.tableView.setModel(self.model)

self.tableView.show()

def buttonClickedStart(self):

"""开启线程，向表中插入数据。"""

self.workThread = WorkThread(self.model)

self.workThread.addDataSignal.connect(self.model.addData)

self.workThread.scrollBottomSignal.connect(self.scrollBottom)

self.workThread.start()

def buttonClickedStop(self):

"""停止线程向表中插入数据。"""

self.workThread.stop()

def buttonClickedClear(self):

"""清空表。"""

self.model.clearView()

def scrollBottom(self):

"""右侧滑动条保持在最下面。"""

self.tableView.scrollToBottom()

if __name__ == '__main__':

app = QApplication(sys.argv)

ui = MainUI()

ui.show()

sys.exit(app.exec_())

程序运行界面图：