Custom QItemDelegate: mouse hover, margins, colors?

[witsch]

Just what I needed. Although a programmer of 20 years, I struggle very hard with the PyQt stuff. Models are difficult! Anyway many thanks for this wonderful example.

HW

[fede]

You're welcome, I'm glad it's useful. I am myself struggling with the Qt models and I do agree they are difficult.

I also wanted to post an update: a slight change to the code above so that the delegate will work with (key, display text) pairs, where key is a meaningful value stored by the model (in my actual program from which this example is taken it is a column in an sql table which is the foreign key to the table where the display text is stored). In any case, the code above just stores the combobox's index values in the model, whereas the code below is slightly more sophisticated and allows for storing arbitrary values in the model, using the UserRole property of the combobox. Hope it's useful to someone out there!

#! /usr/bin/env python
# -*- coding: utf-8 -*-
import sys
from PyQt4 import QtCore
from PyQt4 import QtGui
 
class MyRelationalDelegate(QtGui.QStyledItemDelegate):
  def __init__(self, parent=None):
    self.choices = {1: 'One', 2: 'Two', 3:'Three', 7: 'Seven'}
    super(MyRelationalDelegate,  self).__init__(parent)
 
  def paint(self,  painter,  option,  index):
    styleOption = QtGui.QStyleOptionViewItemV4(option)
    # define the text to be shown
    text = self.choices[index.data(QtCore.Qt.DisplayRole).toInt()[0]] # value in choices dict corresponding to integer stored in tablewidget model
    styleOption.text = text
    # paint the cell
    self.parent().style().drawControl(QtGui.QStyle.CE_ItemViewItem, styleOption, painter)    
 
  def createEditor(self, parent, option, index):
    combobox = QtGui.QComboBox(parent)
    combobox.setEditable(True)
    for c in sorted(self.choices.items()):      
      combobox.addItem(c[1], c[0]) # Add items to combobox, storing display text in DisplayRole and underlying value in UserRole
    return combobox
 
  def select_line_editor(self):
    self.line_editor.setText("")
 
  def setEditorData(self, editor, index):
    pos = editor.findData(index.data().toInt()[0])
    if pos == -1:  #text not found, set cell value to first item in VALUES
      pos = 0
    editor.setCurrentIndex(pos)
 
  def setModelData(self, editor, model, index):
    model.setData(index, QtCore.QVariant(editor.itemData(editor.currentIndex()))) # editor.currentIndex es la posición actual en el combobox; itemData recupera (por default) la data UserRole almacenada allÃ*, que en este caso es el valor del foreign key
 
 
class myWindow(QtGui.QDialog):
    def __init__(self):
        QtGui.QDialog.__init__(self)
        DATA = [['First row', 1, 1], ['Second Row', 2, 2]]
        self.table = QtGui.QTableWidget(self)
        self.table.setSortingEnabled(False)
        self.table.setRowCount(len(DATA))
        self.table.setColumnCount(len(DATA[0]))
        self.table.setItemDelegateForColumn(1, MyRelationalDelegate(self))
        for row in range(len(DATA)):
            for col in range(len(DATA[row])):
                item = QtGui.QTableWidgetItem(str(DATA[row][col]))
                self.table.setItem(row, col, item)
        layout = QtGui.QVBoxLayout()
        layout.addWidget(self.table)
        self.setLayout(layout)        
 
 
if __name__ == "__main__":
    app = QtGui.QApplication(sys.argv)
    window = myWindow()
    window.show()
    app.exec_()

#! /usr/bin/env python
# -*- coding: utf-8 -*-
import sys
from PyQt4 import QtCore
from PyQt4 import QtGui

class MyRelationalDelegate(QtGui.QStyledItemDelegate):
  def __init__(self, parent=None):
    self.choices = {1: 'One', 2: 'Two', 3:'Three', 7: 'Seven'}
    super(MyRelationalDelegate,  self).__init__(parent)

  def paint(self,  painter,  option,  index):
    styleOption = QtGui.QStyleOptionViewItemV4(option)
    # define the text to be shown
    text = self.choices[index.data(QtCore.Qt.DisplayRole).toInt()[0]] # value in choices dict corresponding to integer stored in tablewidget model
    styleOption.text = text
    # paint the cell
    self.parent().style().drawControl(QtGui.QStyle.CE_ItemViewItem, styleOption, painter)    

  def createEditor(self, parent, option, index):
    combobox = QtGui.QComboBox(parent)
    combobox.setEditable(True)
    for c in sorted(self.choices.items()):      
      combobox.addItem(c[1], c[0]) # Add items to combobox, storing display text in DisplayRole and underlying value in UserRole
    return combobox
  
  def select_line_editor(self):
    self.line_editor.setText("")

  def setEditorData(self, editor, index):
    pos = editor.findData(index.data().toInt()[0])
    if pos == -1:  #text not found, set cell value to first item in VALUES
      pos = 0
    editor.setCurrentIndex(pos)

  def setModelData(self, editor, model, index):
    model.setData(index, QtCore.QVariant(editor.itemData(editor.currentIndex()))) # editor.currentIndex es la posición actual en el combobox; itemData recupera (por default) la data UserRole almacenada allÃ*, que en este caso es el valor del foreign key


class myWindow(QtGui.QDialog):
    def __init__(self):
        QtGui.QDialog.__init__(self)
        DATA = [['First row', 1, 1], ['Second Row', 2, 2]]
        self.table = QtGui.QTableWidget(self)
        self.table.setSortingEnabled(False)
        self.table.setRowCount(len(DATA))
        self.table.setColumnCount(len(DATA[0]))
        self.table.setItemDelegateForColumn(1, MyRelationalDelegate(self))
        for row in range(len(DATA)):
            for col in range(len(DATA[row])):
                item = QtGui.QTableWidgetItem(str(DATA[row][col]))
                self.table.setItem(row, col, item)
        layout = QtGui.QVBoxLayout()
        layout.addWidget(self.table)
        self.setLayout(layout)        

        
if __name__ == "__main__":
    app = QtGui.QApplication(sys.argv)
    window = myWindow()
    window.show()
    app.exec_()

To copy to clipboard, switch view to plain text mode