from PyQt4.QtCore import QAbstractListModel, QModelIndex, QVariant, Qt

class DisplayModel(QAbstractListModel):
    def __init__(self, parent=None, list=None, displayAttribute=None, **args):
        QAbstractListModel.__init__(self, parent, *args)
        self.list = list
        self.displayAttribute = displayAttribute

    def rowCount(self, parent=QModelIndex()):
        return len(self.list)

    def data(self, index, role):
        if index.isValid() and role == Qt.DisplayRole:
            row = index.row()
            object = self.list[row]
            displayString = self._stringToDisplay(object)
            return QVariant(displayString)
        else:
            return QVariant()

    def _stringToDisplay(self, object):
        if hasattr(object, self.displayAttribute):
            return getattr(object, self.displayAttribute)
        else:
            return "invalid display attribute: " + self.displayAttribute