Pyqt5 Model/view框架 3.可编辑项

Pyqt5 Model/view框架 3.可编辑项
https://www.cnblogs.com/hangxin1940/archive/2012/12/07/2806450.html

# -*- coding: utf-8 -*-

import sys
import sys
from PyQt5.QtWidgets import QWidget, QToolTip, QApplication
from PyQt5.QtGui import QFont
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
import sys


class MyListModel(QAbstractListModel):
    """
    我的第一个模型
    """
    def __init__(self,parent=None):
        super(MyListModel,self).__init__(parent)

        #这是数据
        self._data=[70,90,20,50]


        pass

    def createEditor(self, parent, option, index):
        ret =  QStyledItemDelegate.createEditor(self, parent, option, index)
        print(ret)
        #<PyQt5.QtWidgets.QSpinBox object at 0x0000000002936828>
        return ret

    def rowCount(self, parent=QModelIndex()):
        """
        这个方法返回了数据的行数
        也就是有多少个条目得数据
        """

        return len(self._data)

    def data(self,index,role=Qt.DisplayRole):
        """
        根据当前index索引，返回当前的数据
        然后再由Qt进行渲染显示
        """

        #如果当前得索引是不活动得
        if not index.isValid() or not 0 <= index.row() < self.rowCount():
            #亦或者当前的索引值不在合理范围，即小于等于0，超出总行数
            return QVariant() #返回一个QVariant，相当与空条目

        #从索引取得当前的航号
        row=index.row()

        #如果当前角色是DisplayRole
        if role==Qt.DisplayRole:
            #返回当前行的数据
            return self._data[row]

        #当前角色为编辑模式，显示原本数据
        #这样，当我们双击单元项时，不至于什么都不显示
        if role==Qt.EditRole:
            return self._data[row]

        #如果角色不满足需求，则返回QVariant
        return QVariant()

    def flags(self, index):
        """
        flag描述了view中数据项的状态信息
        """

        #首先获取超类的flags返回值
        flag=super(MyListModel,self).flags(index)

        #或运算，将ItemIsEditable（可编辑）标志叠加上去
        return flag | Qt.ItemIsEditable

    def setData(self,index,value,role=Qt.EditRole):
        """
        设置数据
        """
        print('MyListModel setData', index.column(), index.row())
        #如果当前为编辑角色
        if role == Qt.EditRole:
            # print("setData Qt.EditRole")
            # print(type(value))
            # print(value)

            #QVariant的这个方法，返回的bool类型表示这个值是否可以被转为int类型
            value_int = value

            #保存数据
            self._data[index.row()]=value_int
            #发射数据更改信号，以便让view更新
            self.dataChanged.emit(index,index)

            #数据是否成功更新
            return True



class MyDelegate(QStyledItemDelegate):
    """
    自定义的委托
    用来在Model获取后，view显示前，再将数据渲染一次
    """
    def __init__(self, parent=None):
        QStyledItemDelegate.__init__(self, parent)
        self.opts=QStyleOptionProgressBar()

    def paint(self,painter,option,index):
        """
        paint，有了画布画笔，想怎么显示就怎么显示，画什么按自己的想法来
        """


        if True:
            #print('MyDelegate paint', index.column(), index.row())
            # print(type(index))
            # print(type(index.data(Qt.DisplayRole)))
            # print(index.data(Qt.DisplayRole))
            #首先，从索引获取数据，这里获取当前索引角色为DisplayQole的数据
            # item_var=index.data(Qt.DisplayRole) #[QVariant]
            # #数据是C格式，我们再转为Python格式，记住这点
            # item_str=item_var.toPyObject() #[QVariant] -&gt; str

            #我们将数据以进度条的方式显现


            self.opts.rect = option.rect #进度条所占的矩形大小
            self.opts.minimum = 0
            self.opts.maximum = 100
            self.opts.text = str(index.data(Qt.DisplayRole)) #显示的内容
            self.opts.textAlignment = Qt.AlignCenter
            self.opts.textVisible = True
            self.opts.progress=int(index.data(Qt.DisplayRole)) #设置当前进度

            #这是关键
            #让QApplication根据当前的风格渲染控件并画出来
            QApplication.style().drawControl(QStyle.CE_ProgressBar,self.opts,painter)
        else:
            QStyledItemDelegate.paint(self, painter, option, index)


def main():
    app=QApplication(sys.argv)


    #新建一个自定义Model
    model=MyListModel()
    #新建一个委托(Delagate)
    delegate=MyDelegate()

    #新建一个ListView
    view=QListView()

    #设置view的model
    view.setModel(model)
    #设置view的delegate
    view.setItemDelegate(delegate)

    view.show()

    sys.exit(app.exec_())


if __name__ == "__main__":
    main()

运行效果: