PyQt5 tableView显示底片为红色和(x.xx)

时间:2019-10-23 13:54:31

标签: python csv tableview pyqt5

首先感谢您为我提供的帮助。我在网上找到了一些代码,这些代码与我需要执行的代码非常接近,但是我无法做最后一件事。我希望csv(tableView)中的任何负数都显示为红色并格式化为(x.xx),正数为x.xx。我在线上找到了一些有关更改单元格背景的代码,但是我想更改字体和示例使用的抽象模型,是否需要使用抽象方法而不是标准方法来进行所有这些操作?如果我需要抽象地做,您可以提供示例(我对所有这些都还很陌生)。

#!/usr/bin/env python
#-*- coding:utf-8 -*-
import csv
import locale

from PyQt5 import QtCore, QtGui, QtWidgets


class MyWindow(QtWidgets.QWidget):
    def __init__(self, fileName, parent=None):
        super(MyWindow, self).__init__(parent)

        locale.setlocale(locale.LC_ALL, '')

        self.fileName = fileName

        self.model = QtGui.QStandardItemModel(self)

        self.tableView = QtWidgets.QTableView(self)
        self.tableView.setModel(self.model)
        self.tableView.horizontalHeader().setStretchLastSection(True)

        self.pushButtonLoad = QtWidgets.QPushButton(self)
        self.pushButtonLoad.setText("Load Csv File!")
        self.pushButtonLoad.clicked.connect(self.on_pushButtonLoad_clicked)

        self.pushButtonWrite = QtWidgets.QPushButton(self)
        self.pushButtonWrite.setText("Write Csv File!")
        self.pushButtonWrite.clicked.connect(self.on_pushButtonWrite_clicked)

        self.layoutVertical = QtWidgets.QVBoxLayout(self)
        self.layoutVertical.addWidget(self.tableView)
        self.layoutVertical.addWidget(self.pushButtonLoad)
        self.layoutVertical.addWidget(self.pushButtonWrite)

    def loadCsv(self, fileName):
        with open(fileName, "r") as fileInput:
            # skip header
            next(fileInput)
            for row in csv.reader(fileInput):
                # convert to $x.xx and ($x.xx)
                row[-1] = float(row[-1])
                row[-1] = locale.currency(row[-1], grouping=True)

                items = [
                    QtGui.QStandardItem(field)
                    for field in row
                ]

                self.model.appendRow(items)

    def writeCsv(self, fileName):
        with open(fileName, "w", newline='') as fileOutput:
            writer = csv.writer(fileOutput)
            for rowNumber in range(self.model.rowCount()):
                fields = [
                    self.model.data(
                        self.model.index(rowNumber, columnNumber),
                        QtCore.Qt.DisplayRole
                    )
                    for columnNumber in range(self.model.columnCount())
                ]
                writer.writerow(fields)

    @QtCore.pyqtSlot()
    def on_pushButtonWrite_clicked(self):
        self.writeCsv(self.fileName)

    @QtCore.pyqtSlot()
    def on_pushButtonLoad_clicked(self):
        self.loadCsv(self.fileName)

if __name__ == "__main__":
    import sys

    app = QtWidgets.QApplication(sys.argv)
    app.setApplicationName('MyWindow')

    main = MyWindow("data.csv")
    main.show()

    sys.exit(app.exec_())

1 个答案:

答案 0 :(得分:1)

要更改信息显示方式的格式,有几个选项:

  • 重写模型的数据方法,以便它返回与标准角色相关联的必要值,例如在这种情况下Qt :: DisplayRole和Qt :: ForegroundRole,

  • 使用诸如QIdentityProxyModel或QSortFilterProxyModel之类的代理,方法是将数据方法覆盖为以前的方法,或者使用

  • 创建委托以自定义绘画。

在这种情况下,我将使用last方法,因为它更灵活,因为如果要在多个视图中显示模型的信息以不同的形式显示,并且还可以修改不是由模型确定的其他属性,而是由其他属性来修改元素,例如样式。

考虑到上述情况,解决方案是:

class CustomDelegate(QtWidgets.QStyledItemDelegate):
    def initStyleOption(self, option, index):
        super(CustomDelegate, self).initStyleOption(option, index)
        try:
            value = float(option.text)
        except ValueError:
            return
        option.text = "{0:.2f}".format(value)
        brush = (
            option.palette.brush(QtGui.QPalette.Text)
            if value >= 0
            else QtGui.QBrush(QtGui.QColor("red"))
        )
        option.palette.setBrush(QtGui.QPalette.Text, brush)
self.tableView = QtWidgets.QTableView(self)
# ...
delegate = CustomDelegate(self.tableView)
self.tableView.setItemDelegateForColumn(1, delegate)