如何通过QAbstractTableModel实现checkbox复选框

liaozd

我想通过flag的方式实现第一列数据有checkbox，我尝试了一下，但是不明白是哪里出错了。

我还想做的是，得到那些被复选框选中的坐标

1. from PyQt4.QtCore import *
   
2. from PyQt4.QtGui import *
   
3. import sys
   
4. 
   
5. data = [['00','01','02'],
   
6.          ['10','11','12'],
   
7.          ['20','21','22']]
   
8. 
   
9. class MainWindow(QWidget):
   
10.     def __init__(self, parent=None, *args):
    
11.         super(MainWindow, self).__init__(parent)
    
12.         
    
13.         clipTableWidget = QTableWidget()
    
14.         self.model = TModel(data, self)
    
15.         clipTableView = QTableView()
    
16.         clipTableView.setModel(self.model)
    
17.         layout = QVBoxLayout()
    
18.         layout.addWidget(clipTableView)
    
19.         self.setLayout(layout)
    
20. 
    
21.                
    
22. class TModel(QAbstractTableModel):
    
23. 
    
24.     def __init__(self, datain, parent=None):
    
25.         super(TModel, self).__init__(parent)
    
26.         self.arraydata = datain
    
27. 
    
28.     def rowCount(self, parent=QModelIndex()):
    
29.         return len(self.arraydata)
    
30. 
    
31.     def columnCount(self, parent=QModelIndex()):
    
32.         return len(self.arraydata[0])
    
33. 
    
34.     def data(self, index, role):
    
35.         if not index.isValid():
    
36.             return QVariant()
    
37.         elif role != Qt.DisplayRole:
    
38.             return QVariant()
    
39.         return QVariant(self.arraydata[index.row()][index.column()])
    
40.    
    
41.      # 以下是我的尝试，我不知道哪里错了，第一列无法显示出有复选框   
    
42.     def flags(self, index):
    
43.         if not index.isValid():
    
44.             return QVariant()
    
45.         elif index.column() == 1:
    
46.             return Qt.ItemIsSelectable|Qt.ItemIsEnabled|Qt.ItemIsUserCheckable
    
47.         return QVariant()
    
48.    
    
49. 
    
50. 
    
51. if __name__ == '__main__':
    
52.    
    
53.     app = QApplication(sys.argv)
    
54.     mainwindow = MainWindow()
    
55.     mainwindow.show()
    
56.     sys.exit(app.exec_())

复制代码

变形金刚四

帮顶                           

liaozd

回复 1# liaozd

网上找了一下，但是还是不是很明确原理，和如何得到哪些row是checked的?

1. from PyQt4.QtCore import *
   
2. from PyQt4.QtGui import *
   
3. import sys
   
4. 
   
5. my_array = [['00','01','02'],
   
6.             ['10','11','12'],
   
7.             ['20','21','22']]
   
8. 
   
9. def main():
   
10.     app = QApplication(sys.argv)
    
11.     w = MyWindow()
    
12.     w.show()
    
13.     sys.exit(app.exec_())
    
14. 
    
15. class MyWindow(QWidget):
    
16.     def __init__(self, *args):
    
17.         QWidget.__init__(self, *args)
    
18.         tablemodel = MyTableModel(my_array, self)
    
19.         tableview = QTableView()
    
20.         tableview.setModel(tablemodel)
    
21.         layout = QVBoxLayout(self)
    
22.         layout.addWidget(tableview)
    
23.         self.setLayout(layout)
    
24. 
    
25. class MyTableModel(QAbstractTableModel):
    
26.     def __init__(self, datain, parent=None, *args):
    
27.         QAbstractTableModel.__init__(self, parent, *args)
    
28.         self.arraydata = datain
    
29.         self._checked=[False for i in  xrange(self.rowCount(self))]
    
30.         print self._checked
    
31. 
    
32.     def rowCount(self, parent):
    
33.         return len(self.arraydata)
    
34. 
    
35.     def columnCount(self, parent):
    
36.         return len(self.arraydata[0])
    
37. 
    
38.     def flags(self, index):
    
39.         if not index.isValid():
    
40.             return
    
41.         elif index.column() == 0:
    
42.             return Qt.ItemIsEnabled|Qt.ItemIsUserCheckable
    
43.         return Qt.ItemIsSelectable|Qt.ItemIsEditable|Qt.ItemIsEnabled
    
44. 
    
45.     def data(self, index, role):
    
46. 
    
47.         if not index.isValid():
    
48.             return QVariant()
    
49.         elif role == Qt.DisplayRole:
    
50.             return QVariant(self.arraydata[index.row()][index.column()])
    
51.         elif index.column() == 0 and role==Qt.CheckStateRole:
    
52.             return self._checked[index.row()]
    
53.         return None # what this for? I am not sure.
    
54. 
    
55.     def setData(self, index, value, role):
    
56.         if not index.isValid() or role!=Qt.CheckStateRole:
    
57.             return False
    
58.         elif index.column()==0:
    
59.             self._checked[index.row()]=value
    
60.             self.dataChanged.emit(index, index)
    
61.             print self._checked
    
62.         return True
    
63. 
    
64. if __name__ == "__main__":
    
65.     main()
    

复制代码