Change color of a row of a QSqlQueryModel (QTableView)?

[panoss]

I 'm trying to change the color of rows in a QTableView which has a QSqlQueryModel as it's model.

Here is a compilable code sample in python:

import sys
from PyQt4 import QtGui, QtCore, QtSql
 
def main():
   app = QtGui.QApplication(sys.argv)
   w = MyWindow()
   w.show()
   sys.exit(app.exec_())
 
class MyWindow(QtGui.QTableView):
   def __init__(self, *args):
       QtGui.QTableView.__init__(self, *args)
 
       # connect to db (if doesn't exist, it's auto-created)
       self.db = QtSql.QSqlDatabase.addDatabase('QSQLITE')
       self.db.setDatabaseName('test.db')
       self.db.open()
 
       #create a table in db and add some data
       query = QtSql.QSqlQuery()
       query.exec_("DROP TABLE IF EXISTS games")
       query.exec_("CREATE TABLE games(id INTEGER PRIMARY KEY, hometeam TEXT, visitorteam TEXT) ")
       query.exec("INSERT INTO games (hometeam, visitorteam) VALUES ('Star', 'Eagles')")
       query.exec("INSERT INTO games (hometeam, visitorteam) VALUES ('Best team', 'Reds');")
 
       # set the model
       model = QtSql.QSqlQueryModel(self)#QtGui.QStandardItemModel(0, 2)
       self.setModel(model)
       model.setQuery("SELECT * FROM games")
 
       # paint first two rows
       for i in range(0, 2):
           model.setData(model.index(i, 0), QtGui.QBrush(QtCore.Qt.red), QtCore.Qt.BackgroundRole)
           model.setData(model.index(i, 1), QtGui.QBrush(QtCore.Qt.red), QtCore.Qt.BackgroundRole)
 
 
if __name__ == "__main__":
   main()

import sys
from PyQt4 import QtGui, QtCore, QtSql
 
def main():
   app = QtGui.QApplication(sys.argv)
   w = MyWindow()
   w.show()
   sys.exit(app.exec_())
 
class MyWindow(QtGui.QTableView):
   def __init__(self, *args):
       QtGui.QTableView.__init__(self, *args)
 
       # connect to db (if doesn't exist, it's auto-created)
       self.db = QtSql.QSqlDatabase.addDatabase('QSQLITE')
       self.db.setDatabaseName('test.db')
       self.db.open()
 
       #create a table in db and add some data
       query = QtSql.QSqlQuery()
       query.exec_("DROP TABLE IF EXISTS games")
       query.exec_("CREATE TABLE games(id INTEGER PRIMARY KEY, hometeam TEXT, visitorteam TEXT) ")
       query.exec("INSERT INTO games (hometeam, visitorteam) VALUES ('Star', 'Eagles')")
       query.exec("INSERT INTO games (hometeam, visitorteam) VALUES ('Best team', 'Reds');")
 
       # set the model
       model = QtSql.QSqlQueryModel(self)#QtGui.QStandardItemModel(0, 2)
       self.setModel(model)
       model.setQuery("SELECT * FROM games")
 
       # paint first two rows
       for i in range(0, 2):
           model.setData(model.index(i, 0), QtGui.QBrush(QtCore.Qt.red), QtCore.Qt.BackgroundRole)
           model.setData(model.index(i, 1), QtGui.QBrush(QtCore.Qt.red), QtCore.Qt.BackgroundRole)
 
 
if __name__ == "__main__":
   main()

To copy to clipboard, switch view to plain text mode 

What am I doing wrong?

[panoss]

I subclassed it:

class ColorfullSqlQueryModel(QtSql.QSqlQueryModel):
    def __init__(self, dbcursor=None):
        super(ColorfullSqlQueryModel, self).__init__()
 
    def data(self, QModelIndex, role=None):
        v = QtSql.QSqlQueryModel.data(self, QModelIndex, role);
        if role == QtCore.Qt.BackgroundRole:
            return QtGui.QColor(QtCore.Qt.yellow)
        return (v);

class ColorfullSqlQueryModel(QtSql.QSqlQueryModel):
    def __init__(self, dbcursor=None):
        super(ColorfullSqlQueryModel, self).__init__()

    def data(self, QModelIndex, role=None):
        v = QtSql.QSqlQueryModel.data(self, QModelIndex, role);
        if role == QtCore.Qt.BackgroundRole:
            return QtGui.QColor(QtCore.Qt.yellow)
        return (v);

To copy to clipboard, switch view to plain text mode 

It works!! Turns all rows to yellow!
But how can I make it paint only the rows I want?
How can I pass the color to function data?

[d_stranz]

But how can I make it paint only the rows I want?
How can I pass the color to function data?

You can't pass any additional parameters to the data() method. If you need to set the color on an item-by-item basis, then you need to look at the row and column of the QModelIndex argument. Unless you can decide based on the row and column alone, generally you will need to implement an extra data structure in your custom model where you can look up the color for the index.