使用pyqt重写官方c++示例—FreezeTableWidget

最近使用pyqt快速搭建界面，实现一个重写的TableView，直接上代码：

feezeTableWidget.py

import typing
from PyQt5 import QtCore, QtGui
from PyQt5.QtWidgets import QTableView, QAbstractItemView, QHeaderView
from PyQt5.QtCore import Qt, QModelIndexclass FreezeTableView(QTableView):def __init__(self, model):super(FreezeTableView, self).__init__()self.model = modelself.frozenTableView = QTableView(self)def init(self):self.setModel(self.model)self.frozenTableInit()self.horizontalHeader().sectionResized.connect(self.updateSectionWidth)self.verticalHeader().sectionResized.connect(self.updateSectionHeight)self.verticalScrollBar().valueChanged.connect(self.frozenTableView.verticalScrollBar().setValue)self.frozenTableView.verticalScrollBar().valueChanged.connect(self.verticalScrollBar().setValue)def frozenTableInit(self):self.frozenTableView.setModel(self.model)self.frozenTableView.verticalHeader().hide()self.frozenTableView.setFocusPolicy(Qt.NoFocus)self.frozenTableView.horizontalHeader().setSectionResizeMode(QHeaderView.Fixed)self.frozenTableView.horizontalHeader().setFixedHeight(self.horizontalHeader().height())self.viewport().stackUnder(self.frozenTableView)self.frozenTableView.setStyleSheet('QTableView {''border: none;''background-color: #8EDE21;''selection-background-color: #999}')self.frozenTableView.setSelectionModel(self.selectionModel())[self.frozenTableView.setColumnHidden(col, True) for col in range(1, self.model.columnCount())]self.frozenTableView.setColumnWidth(0, self.columnWidth(0))self.frozenTableView.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)self.frozenTableView.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)self.frozenTableView.show()self.updateFrozenTableGeometry()self.frozenTableView.setVerticalScrollMode(self.ScrollPerPixel)self.setVerticalScrollMode(self.ScrollPerPixel)self.setHorizontalScrollMode(self.ScrollPerPixel)def updateFrozenTableGeometry(self):self.frozenTableView.setGeometry(self.verticalHeader().width() + self.frameWidth(),self.frameWidth(), self.columnWidth(0),self.viewport().height() + self.horizontalHeader().height())def updateSectionWidth(self, logicalIndex, oldSize, newSize):if not logicalIndex == 0:self.frozenTableView.setColumnWidth(0, newSize)self.updateFrozenTableGeometry()def updateSectionHeight(self, logicalIndex, oldSize, newSize):self.frozenTableView.setRowHeight(logicalIndex, newSize)def resizeEvent(self, e: QtGui.QResizeEvent) -> None:QTableView.resizeEvent(self, e)self.updateFrozenTableGeometry()def scrollTo(self, index: QtCore.QModelIndex, hint: QAbstractItemView.ScrollHint = ...) -> None:if index.column() > 0:QTableView.scrollTo(self, index, hint)def moveCursor(self, cursorAction: QAbstractItemView.CursorAction,modifiers: typing.Union[QtCore.Qt.KeyboardModifiers,QtCore.Qt.KeyboardModifier]) -> QtCore.QModelIndex:current = QTableView.moveCursor(self, cursorAction, modifiers)if cursorAction == QAbstractItemView.MoveLeft and current.column() > 0 \and self.visualRect(current).topLeft().x() < self.frozenTableView.columnWidth(0):newValue = self.verticalScrollBar().value() + self.visualRect(current).topLeft().x() \- self.frozenTableView.columnWidth(0)self.horizontalScrollBar().setValue(newValue)return current

main.py
这里分别使用了python的文件操作和Qt的文件操作接口，其中使用python接口界面表头显示有些问题，暂时没搞定，菜鸡如我。

Qt 接口

def qtRead(file, model):fd = QFile(file)try:if fd.open(QFile.ReadOnly):stream = QTextStream(fd)line = stream.readLine()con = line.split(',')model.setHorizontalHeaderLabels(con)row = 0while not stream.atEnd():line = stream.readLine()if not line.startswith('#') and line.__contains__(','):con = line.split(',')[model.setItem(row, col, QStandardItem(con[col])) for col in range(len(con))]row += 1except Exception as e:print(str(e))fd.close()

python接口读取数据

def readData(file, model):with open(file, 'r', encoding='utf-8') as fd:line = fd.readline()content = line.split(',')model.setHorizontalHeaderLabels(content)row = 0while True:line = fd.readline()if not line:breakif not line.startswith('#') and line.__contains__(','):content = line.split(',')[model.setItem(row, col, QStandardItem(content[col])) for col in range(len(content))]# for col in range(len(content)):#     item = QStandardItem(content[col])#     # yield row, col, item#     model.setItem(row, col, item)row += 1fd.close()

总的代码如下：

from freezeWidget.freezeTableWidget import FreezeTableView
from PyQt5.QtGui import QStandardItem, QStandardItemModel
from PyQt5.QtWidgets import QApplication
from PyQt5.QtCore import QFile, QTextStream
import sysdef qtRead(file, model):fd = QFile(file)try:if fd.open(QFile.ReadOnly):stream = QTextStream(fd)line = stream.readLine()con = line.split(',')model.setHorizontalHeaderLabels(con)row = 0while not stream.atEnd():line = stream.readLine()if not line.startswith('#') and line.__contains__(','):con = line.split(',')[model.setItem(row, col, QStandardItem(con[col])) for col in range(len(con))]row += 1except Exception as e:print(str(e))fd.close()def readData(file, model):with open(file, 'r', encoding='utf-8') as fd:line = fd.readline()content = line.split(',')model.setHorizontalHeaderLabels(content)row = 0while True:line = fd.readline()if not line:breakif not line.startswith('#') and line.__contains__(','):content = line.split(',')[model.setItem(row, col, QStandardItem(content[col])) for col in range(len(content))]# for col in range(len(content)):#     item = QStandardItem(content[col])#     # yield row, col, item#     model.setItem(row, col, item)row += 1fd.close()if __name__ == '__main__':app = QApplication([])model = QStandardItemModel()# 添加数据# readData('grades.txt', model)qtRead('grades.txt', model)tableView = FreezeTableView(model)tableView.init()tableView.setWindowTitle('Frozen Widget')tableView.resize(560, 680)tableView.show()sys.exit(app.exec_())

实现效果图如下: