#!/usr/bin/env python
# -*- coding: utf8 -*-
# some imports :
from __future__ import print_function
import sys
# display infos :
PYTHONVERSION = sys.version_info[0] * 10 + sys.version_info[1]
print(‘PYTHONVERSION :‘, PYTHONVERSION)
# PyQt or PySide ?
PYSIDE = False
if not("PYQT" in sys.argv):
try:
from PySide import QtCore, QtGui
PYSIDE = True
except:
pass
if PYSIDE:
import PySide
print(‘PYSIDE‘, PySide.__version__)
else:
import sip
sip.setapi(‘QString‘, 2)
sip.setapi(‘QVariant‘, 2)
from PyQt4 import QtCore, QtGui
print(‘PYQT‘, QtCore.PYQT_VERSION_STR)
# display Qt version :
QTVERSION = QtCore.qVersion()
print(‘QTVERSION :‘, QTVERSION)
class TestModel(QtGui.QStandardItemModel):
def __init__(self, parent=None):
super(TestModel, self).__init__(parent)
self.headers = [‘id‘, ‘firstname‘, ‘lastname‘]
self.datas = []
self.datas.append([‘101‘, ‘Danny‘, ‘Young‘])
self.datas.append([‘102‘, ‘Christine‘, ‘Holand‘])
self.datas.append([‘103‘, ‘Lars‘, ‘Gordon‘])
self.datas.append([‘104‘, ‘Roberto‘, ‘Robitaille‘])
self.datas.append([‘105‘, ‘Maria‘, ‘Papadopoulos‘])
self.emptyRow = []
nbCols = len(self.headers)
self.setColumnCount(nbCols)
nbRows = len(self.datas)
self.setRowCount(nbRows)
for col in range(nbCols):
self.setHeaderData(col, QtCore.Qt.Horizontal, self.headers[col])
for row in range(nbRows):
for col in range(nbCols):
data = self.datas[row][col]
self.setData(self.index(row, col, QtCore.QModelIndex()), data)
def data(self, index, role=QtCore.Qt.DisplayRole):
column, row = index.column(), index.row()
if role == QtCore.Qt.TextAlignmentRole:
if column > 0:
return QtCore.Qt.AlignLeft
else:
return QtCore.Qt.AlignRight
elif role == QtCore.Qt.DisplayRole:
return self.datas[row][column]
return
def columnCount(self, parent=QtCore.QModelIndex()):
return len(self.headers)
def rowCount(self, parent=QtCore.QModelIndex()):
return len(self.datas)
class MainForm(QtGui.QDialog):
def __init__(self, parent=None):
super(MainForm, self).__init__(parent)
self.model = TestModel()
self.tableView = QtGui.QTableView()
self.tableView.setModel(self.model)
insertRowButton = QtGui.QPushButton("InsertRow")
insertRowButton.clicked.connect(self.insertRow)
removeRowBugButton = QtGui.QPushButton("RemoveRowBug")
removeRowBugButton.clicked.connect(self.removeRowBug)
removeRowButton = QtGui.QPushButton("RemoveRow")
removeRowButton.clicked.connect(self.removeRow)
quitButton = QtGui.QPushButton("Quit")
quitButton.clicked.connect(self.accept)
buttonLayout = QtGui.QHBoxLayout()
buttonLayout.addWidget(insertRowButton)
buttonLayout.addWidget(removeRowBugButton)
buttonLayout.addWidget(removeRowButton)
buttonLayout.addStretch()
buttonLayout.addWidget(quitButton)
splitter = QtGui.QSplitter(QtCore.Qt.Horizontal)
self.vbox = QtGui.QVBoxLayout()
self.vbox.addWidget(self.tableView)
widget = QtGui.QWidget()
widget.setLayout(self.vbox)
splitter.addWidget(widget)
layout = QtGui.QVBoxLayout()
layout.addWidget(splitter)
layout.addLayout(buttonLayout)
self.setLayout(layout)
self.setMinimumWidth(400)
def insertRow(self):
index = self.tableView.currentIndex()
if index.isValid():
row = index.row()
column = index.column()
else:
row, column = 0, 0
if self.model.insertRow(row):
emptyRow = [‘999‘, ‘AAA‘, ‘Zzzzzzzzzzz‘]
row += 1
self.model.datas.insert(row, emptyRow)
index = self.model.index(row, column)
self.tableView.setFocus()
self.tableView.setCurrentIndex(index)
def removeRowBug(self):
index = self.tableView.currentIndex()
if not index.isValid():
return
row = index.row()
column = index.column()
if (self.model.removeRow(row)):
self.model.datas.pop(row)
if row > self.model.rowCount() - 1:
row -= 1
# QTableView.verticalHeader is not repaint
# this for verify verticalHeader :
print(self.model.rowCount(), self.tableView.verticalHeader().count())
# I test that but nothing is ok :
#self.tableView.resizeRowsToContents()
#self.tableView.verticalHeader().reset()
#self.tableView.verticalHeader().repaint()
QtCore.QCoreApplication.processEvents()
index = self.model.index(row, column)
self.tableView.setFocus()
self.tableView.setCurrentIndex(index)
def removeRow(self):
index = self.tableView.currentIndex()
if not index.isValid():
return
row = index.row()
column = index.column()
if (self.model.removeRow(row)):
self.model.datas.pop(row)
if row > self.model.rowCount() - 1:
row -= 1
# recreate the tableView solve the bug, but is not good :
self.vbox.removeWidget(self.tableView)
self.tableView.setModel(None)
del self.tableView
self.tableView = QtGui.QTableView()
self.tableView.setModel(self.model)
self.vbox.addWidget(self.tableView)
index = self.model.index(row, column)
self.tableView.setFocus()
self.tableView.setCurrentIndex(index)
if __name__ == ‘__main__‘:
app = QtGui.QApplication(sys.argv)
form = MainForm()
form.show()
sys.exit(app.exec_())
