Как мне вернуть свой объект Python из QVariant в PyQt4?

Question

Я создаю подкласс QAbstractItemModel для отображения в QTreeView.

Мой index() и parent() функция создает QModelIndex используя QAbstractItemModel унаследованная функция createIndex и предоставляя ему row, column, и data нужный.Здесь, в целях тестирования, данные представляют собой строку Python.

class TestModel(QAbstractItemModel):
    def __init__(self):
        QAbstractItemModel.__init__(self)

    def index(self, row, column, parent):
        if parent.isValid():
            return self.createIndex(row, column, "bar")
        return self.createIndex(row, column, "foo")

    def parent(self, index):
        if index.isValid():
            if index.data().data() == "bar":                          <--- NEVER TRUE
                return self.createIndex(0, 0, "foo")
        return QModelIndex()

    def rowCount(self, index):
        if index.isValid():
            if index.data().data() == "bar":                          <--- NEVER TRUE
                return 0
        return 1

    def columnCount(self, index):
        return 1

    def data(self, index, role):
        if index.isValid():
            return index.data().data()                                <--- CANNOT DO ANYTHING WITH IT
        return "<None>"

В рамках index(), parent(), и data() функции, которые мне нужны, чтобы вернуть мои данные.Это происходит как QVariant.Как мне вернуть свой объект Python из QVariant?

Answer 1

Главное — использовать internalPointer() прямо на QModelIndex, не занимаясь QVariant совсем.

class TestModel(QAbstractItemModel):
    def __init__(self, plan):
        QAbstractItemModel.__init__(self)

    def index(self, row, column, parent):
        if not parent.isValid():
            return self.createIndex(row, column, "foo")
        return self.createIndex(row, column, "bar")

    def parent(self, index):
         if index.internalPointer() == "bar":
            return self.createIndex(0, 0, "foo")
        return QModelIndex()

    def rowCount(self, index):
        if index.internalPointer() == "bar":
            return 0
        return 1

    def columnCount(self, index):
        return 1

    def data(self, index, role):
        if role == 0:  # Qt.DisplayRole
            return index.internalPointer()
        else:
            return None

Answer 2

Вы пробовали это?

my_python_object = my_qvariant.toPyObject()

http://pyqt.sourceforge.net/Docs/PyQt4/qvariant.html#toPyObject (просто для полноты картины, но смотреть там особо нечего...)