Das deutsche Python-Forum

Code: Alles auswählen

from PyQt4 import QtGui, QtCore, QtSql
from PyQt4.QtCore import Qt

class MyListModel(QtCore.QAbstractListModel):
    def __init__(self, masterModel, indexes, parent=None):
        QtCore.QAbstractListModel.__init__(self, parent)
        self._masterModel = masterModel
        self._masterModel.dataChanged.connect(self.triggerUpdate)
        self._model = indexes

    # update associated views when master model has new data
    def triggerUpdate(self, topLeft, bottomRight):
        self.dataChanged.emit(self.index(0, topLeft.column()),
                                self.index(0, bottomRight.column()))

    def rowCount(self, index):
        return len(self._model)

    # implement display and edit role for now
    def data(self, index, role):
        if role == Qt.DisplayRole or role == Qt.EditRole:
            return self._masterModel.data(self._model[index.row()])

    # set appropriate flags of the items
    def flags(self, index):
        return Qt.ItemIsSelectable | Qt.ItemIsEditable | Qt.ItemIsEnabled

    # set data of the master model directly
    def setData(self, index, value, role):
        return self._masterModel.setData(self._model[index.row()], value, role)

class MyWidget(QtGui.QWidget):
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)
        self.query = QtSql.QSqlQuery('select * from person')
        self.model = QtSql.QSqlTableModel()
        self.model.setQuery(self.query)
        self.table = QtGui.QTableView(self)
        self.table.setSelectionBehavior(1)
        self.table.setModel(self.model)
        self.selectionModel = QtGui.QItemSelectionModel(self.model)
        self.table.setSelectionModel(self.selectionModel)
        self.gridLayout = QtGui.QGridLayout(self)
        self.gridLayout.addWidget(self.table, 0,0)
        self.listView = QtGui.QListView(self)
        self.gridLayout.addWidget(self.listView, 0,1)

        # change listview model on selection change
        self.selectionModel.selectionChanged.connect(
            lambda index : self.listView.setModel(
                                MyListModel(self.model, index.indexes())))

def prepareDB():
    db = QtSql.QSqlDatabase.addDatabase("QSQLITE")
    db.setDatabaseName(":memory:")
    db.open()
    query = QtSql.QSqlQuery('CREATE TABLE person (forename char(50),surname char(50))')
    query.prepare("INSERT INTO person (forename, surname) "
                  "VALUES (:forename, :surname)")
    query.bindValue(":forename", "Bart")
    query.bindValue(":surname", "Simpson")
    query.exec_()
    query.bindValue(":forename", "Hans")
    query.bindValue(":surname", "Mustermann")
    query.exec_()
    return db

if __name__ == '__main__':
    app= QtGui.QApplication([])
    db = prepareDB()
    win = MyWidget()
    win.show()
    app.exec_()
    db.close()
    del win, db