© EVILEG 2015-2018
Рекомендует хостинг
TIMEWEB
18 сентября 2018 г. 13:38

Прокси-модель, содержащая на 1 столбец больше, чем модель-источник.

qt5, model, proxy, QIdentityProxyModel, column, PyQt5

Здравствуйте!

Пытаюсь переопределить QIdentityProxyModel так, чтобы она имела на 1 столбец больше, чем ее модель-источник, но никак не могу добиться успешного результата.

Вот заготовка, очищенная от мусора:


#!/usr/bin/python3
# -*- coding: utf-8 -*-

"""Попытка реализовать прокси-модель, содержащую на 1 столбец больше, чем модель-источник."""
import sys
import typing

from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *


app = None


class MyProxyModel(QIdentityProxyModel):
    """Прокси-модель, содержащая на 1 столбец больше, чем модель-источник."""

    VALUE = "==="

    def columnCount(self, parent: QModelIndex = None) -> int:
        """Переопределяет соответствующий родительский метод.
        columnCount(self, parent: QModelIndex = QModelIndex()) -> int
        """
        if not self.sourceModel():
            return 0
        return super().columnCount() + 1

    def data(self, proxyIndex: QModelIndex, role: int = None) -> typing.Any:
        """Переопределяет соответствующий родительский метод.
        data(self, index: QModelIndex, role: int = Qt.DisplayRole) -> Any
        """
        assert self.checkIndex(proxyIndex)
        if role is None:
            role = Qt.DisplayRole

        if not self.sourceModel():
            return None

        if role == Qt.DisplayRole:
            if proxyIndex.column() == self.columnCount() - 1:
                # text = ""
                # for column in range(self.columnCount() - 1):
                #     # index = self.index(proxyIndex.row(), column, proxyIndex.parent())
                #     index = self.sibling(proxyIndex.row(), column, proxyIndex)
                #     text += "->" + str(super().data(index))
                # return text
                return self.VALUE

        return super().data(proxyIndex, role)

    def headerData(self, section: int, orientation: int, role: int = None) -> typing.Any:
        """Переопределяет соответствующий родительский метод.
        headerData(self, section: int, orientation: Qt.Orientation, role: int = Qt.DisplayRole) -> Any
        """
        if role is None:
            role = Qt.DisplayRole

        if orientation == Qt.Horizontal and role == Qt.DisplayRole and section == self.columnCount() - 1:
            return "New column"

        return super().headerData(section, orientation, role)

    # def mapFromSource(self, sourceIndex: QModelIndex) -> QModelIndex:
    #     # return super().mapFromSource(sourceIndex)
    #     if not self.sourceModel() or not sourceIndex.isValid():
    #         return QModelIndex()
    #
    #     assert sourceIndex.model() is self.sourceModel()
    #     return self.createIndex(sourceIndex.row(), sourceIndex.column(), sourceIndex.internalPointer())
    #
    # def mapToSource(self, proxyIndex: QModelIndex) -> QModelIndex:
    #     # return super().mapToSource(proxyIndex)
    #     if not self.sourceModel() or not proxyIndex.isValid():
    #         return QModelIndex()
    #
    #     assert proxyIndex.model() is self
    #     return self.sourceModel().createIndex(proxyIndex.row(), proxyIndex.column(), proxyIndex.internalPointer())


def main():
    global app
    app = QApplication(sys.argv)

    sourceModel = QFileSystemModel(app)
    sourceModel.setRootPath(".")

    proxyModel = MyProxyModel(app)
    proxyModel.setSourceModel(sourceModel)

    treeView = QTreeView()
    treeView.setMinimumWidth(800)
    treeView.setModel(proxyModel)
    treeView.setColumnWidth(0, 400)
    # TODO: По прежнему выделяет только ячейку. Надо переопределить MyProxyModel.mapSelection<From/To>Source()!
    treeView.setSelectionBehavior(treeView.SelectRows)
    # treeView.setSelectionMode(treeView.SingleSelection)
    treeView.show()

    def checkProxyModel():
        print("BEGIN checkProxyModel()")

        assert proxyModel.columnCount() == sourceModel.columnCount() + 1

        proxyIndexWithZeroColumn = proxyModel.index(0, 0)
        sourceIndexWithZeroColumn = sourceModel.index(0, 0)
        assert proxyModel.data(proxyIndexWithZeroColumn) == sourceModel.data(sourceIndexWithZeroColumn)

        proxyIndexWithPenultimateColumn = proxyModel.index(0, proxyModel.columnCount() - 2)
        sourceIndexWithLastColumn = sourceModel.index(0, sourceModel.columnCount() - 1)
        assert proxyModel.data(proxyIndexWithPenultimateColumn) == sourceModel.data(sourceIndexWithLastColumn)

        proxyIndexWithLastColumn = proxyModel.index(0, proxyModel.columnCount() - 1)
        assert proxyModel.data(proxyIndexWithLastColumn, Qt.DisplayRole) == proxyModel.VALUE

        print("END checkProxyModel()")

    QTimer.singleShot(0, checkProxyModel)

    sys.exit(app.exec())


if __name__ == '__main__':
    main()

В отображении появляется новый столбец с нужным заголовком, но без данных.

Я не понимаю, каким образом мне нужно переопределить методы mapFromSource/mapToSource, чтоб все заработало...  Или же вообще нужно переопределять index(), parent() и прочие?..

Добрый день.

Вот рабочий вариант

#!/usr/bin/python3
# -*- coding: utf-8 -*-

"""Попытка реализовать прокси-модель, содержащую на 1 столбец больше, чем модель-источник."""
import sys
import typing


from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *


app = None


class MyProxyModel(QIdentityProxyModel):
    """Прокси-модель, содержащая на 1 столбец больше, чем модель-источник."""

    VALUE = "==="

    def columnCount(self, parent: QModelIndex = None) -> int:
        """Переопределяет соответствующий родительский метод.
        columnCount(self, parent: QModelIndex = QModelIndex()) -> int
        """
        if not self.sourceModel():
            return 0
        return super().columnCount() + 1

    def data(self, proxyIndex: QModelIndex, role: int = None) -> typing.Any:
        """Переопределяет соответствующий родительский метод.
        data(self, index: QModelIndex, role: int = Qt.DisplayRole) -> Any
        """

        print(proxyIndex.column())
        if role is None:
            role = Qt.DisplayRole

        if not self.sourceModel():
            return None

        if role == Qt.DisplayRole:
            if proxyIndex.column() == self.columnCount() - 1:
                return self.VALUE

        return super().data(proxyIndex, role)

    def index(self, row: int, column: int, parent: QModelIndex = QModelIndex()) -> typing.Any:
        if column == self.columnCount(parent) - 1:
            return self.createIndex(row, column)

        return super().index(row, column, parent)

    def headerData(self, section: int, orientation: int, role: int = None) -> typing.Any:
        """Переопределяет соответствующий родительский метод.
        headerData(self, section: int, orientation: Qt.Orientation, role: int = Qt.DisplayRole) -> Any
        """
        if role is None:
            role = Qt.DisplayRole

        if orientation == Qt.Horizontal and role == Qt.DisplayRole and section == self.columnCount() - 1:
            return "New column"

        return super().headerData(section, orientation, role)


def main():
    global app
    app = QApplication(sys.argv)

    sourceModel = QFileSystemModel(app)
    sourceModel.setRootPath(".")

    proxyModel = MyProxyModel(app)
    proxyModel.setSourceModel(sourceModel)

    treeView = QTreeView()
    treeView.setMinimumWidth(800)
    treeView.setModel(proxyModel)
    treeView.setColumnWidth(0, 400)
    # TODO: По прежнему выделяет только ячейку. Надо переопределить MyProxyModel.mapSelection<From/To>Source()!
    treeView.setSelectionBehavior(treeView.SelectRows)
    treeView.show()

    def checkProxyModel():
        print("BEGIN checkProxyModel()")

        assert proxyModel.columnCount() == sourceModel.columnCount() + 1

        proxyIndexWithZeroColumn = proxyModel.index(0, 0)
        sourceIndexWithZeroColumn = sourceModel.index(0, 0)
        assert proxyModel.data(proxyIndexWithZeroColumn) == sourceModel.data(sourceIndexWithZeroColumn)

        proxyIndexWithPenultimateColumn = proxyModel.index(0, proxyModel.columnCount() - 2)
        sourceIndexWithLastColumn = sourceModel.index(0, sourceModel.columnCount() - 1)
        assert proxyModel.data(proxyIndexWithPenultimateColumn) == sourceModel.data(sourceIndexWithLastColumn)

        proxyIndexWithLastColumn = proxyModel.index(0, proxyModel.columnCount() - 1)
        assert proxyModel.data(proxyIndexWithLastColumn, Qt.DisplayRole) == proxyModel.VALUE

        print("END checkProxyModel()")

    QTimer.singleShot(0, checkProxyModel)

    sys.exit(app.exec())


if __name__ == '__main__':
    main()

Вам надо было метод index переопределить, чтобы создавался QModelIndex для вашей спец-колонки



Для Django рекомендую VDS-хостинг TIMEWEB

Огромное спасибо за ответ!:)

Но вариант все же не является рабочим.

Почти всегда дерево отображается корректно и даже раскрываются новые уровни, если нажать точно на треугольник слева. Но если сменить выделенный элемент, например, щелкнуть мышкой на какую-нибудь ячейку, то приложение падает, и единственным выводом в консоль является:

Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)

qVersion() == 5.11.1, OS == Kubuntu 18.04

Также приложение падает, если вернуть в метод data() строчку

assert self.checkIndex(proxyIndex)

Изучение исходников метода index() в QFileSystemModel и QStandardItemModel ничем пока не помогло. :(

Падение с assert - это уже питоновские заморочки, дайте туда более корректную проверку, поскольку проверка на None просто не работает адекватно в том случае.

Что касается установки информации, то возможно, что ещё потребуется перепределить метод setData. Переопределите его и посмотрите через print, что туда прилетает в индексе и роли.

print(proxyIndex.row(), proxyIndex.column(), role)

Для Django рекомендую VDS-хостинг TIMEWEB

До setData() я еще не добрался, так как пока что даже reed only модель не работает.

В конкретном данном случае в

assert self.checkIndex(proxyIndex)

всегда передается экземпляр QModelIndex. Поэтому это не является проблемой обработки None.
Да и на падения при смене текущей ячейки это не влияет.

Я Вам очень благодарен за то, что все равно пытаетесь помочь!

Если определить метод index() таким костыльным и затратным образом:

def index(self, row: int, column: int, parent: QModelIndex = QModelIndex()) -> QModelIndex:
    assert self.checkIndex(parent)
    if column == self.columnCount() - 1:
        return self.createIndex(row, column, QPersistentModelIndex(parent))

    return super().index(row, column, parent)

то приложение уже перестает падать, но строку вьюха выделить все еще не может, так как для модельного индекса из моего столбца internalPointer() хранит данные, отличающиеся от данных для других столбцов.

Также в этом случае в консоль выводится соответствующее сообщение:

Can't select indexes from different model or with different parents

В итоге, эмулировать существование дополнительных столбцов через QIdentityModel на PyQt5(5.11.1) получается только для плоских (табличных, списочных) моделей.

Для древовидных моделей при создании модельного индекса в internalPointer необходимо хранить некие данные о родителе. Но если установить в него данные можно, то при считывании PyQt не может определить их тип из-за того, что в С++ они объявлены как void*, если считывать через internalPointer(), и qintptr, если считывать через internalId().

В следствие чего, программа крашится.

Попробуйте ещё PySide 2 - это официально поддерживаемый пакет привязок Python к Qt, возможно, что там не будет таких проблем.

Для Django рекомендую VDS-хостинг TIMEWEB

Вот работающий вариант переопределения метода index():

def index(self, row: int, column: int, parent: QModelIndex = QModelIndex()) -> QModelIndex:
    """Переопределяет соответствующий родительский метод.
    index(self, row: int, column: int, parent: QModelIndex = QModelIndex()) -> QModelIndex
    """
    if column == self.columnCount() - 1:
        assert self.checkIndex(parent)
        return self.createIndex(row, column, super().index(row, 0, parent).internalId())

    return super().index(row, column, parent)

Теперь корректно устанавливаются дынные об элементе-предке и программа больше не крашится.

Но остается вопрос, как заставить вьюху менять выделение/текущий индекс при щелчке мышкой по ячейкам из нового столбца?..

Изучение исходников QTreeView и QAbstractItemView показало, что для осуществления обработки нажатий по элементам нового столбца необходимо переопределить метод sibling():

def sibling(self, row: int, column: int, index: QModelIndex) -> QModelIndex:
    """Переопределяет соответствующий родительский метод.
    index(self, row: int, column: int, parent: QModelIndex = QModelIndex()) -> QModelIndex
    """
    if column == self.columnCount() - 1:
        assert self.checkIndex(index)
        assert index.isValid()  # Это можно будет проверить в строке выше, когда в PyQt5 реализуют соотв. параметры.
        return self.createIndex(row, column, super().index(row, 0, index.parent()).internalId())
    return super().sibling(row, column, index)

Полагаю, что вам придётся везде создавать соответствующий валидный индекс для дополнительного столбца, если что-то не работает

Для Django рекомендую VDS-хостинг TIMEWEB

Ответы

Только авторизованные пользователи могут отвечать на форуме.
Пожалуйста, Авторизуйтесь или Зарегистрируйтесь
15 октября 2018 г. 21:36
Allyans .

C++ - Тест 001. Первая программа и типы данных

  • Результат 60баллов,
  • Очки рейтинга-1
15 октября 2018 г. 11:25
Екатерина Самойлова

C++ - Тест 002. Константы

  • Результат 33баллов,
  • Очки рейтинга-10
15 октября 2018 г. 11:17
Екатерина Самойлова

C++ - Тест 006. Перечисления

  • Результат 80баллов,
  • Очки рейтинга4
Последние комментарии
16 октября 2018 г. 16:14
pasagir

Qt/C++ - Урок 006. QSqlQueryModel - Таблицы в Qt с помощью SQL-запросов

В Qt 5.11. при попытке вставить в БД запись выдает ошибку QSqlQuery::prepare: database not openQSqlDatabasePrivate::database: requested database does not belong to the calling thread. ...
10 октября 2018 г. 9:50
Евгений Легоцкой

Qt/C++ - Урок 083. Создание динамической библиотеки и подключение её в другой проект

Если и начинать писать о плагинах, то нужно тогда с Qt Creator начинать, там наверняка будет одинаковый принцип, но по Qt Creator хотя бы информация есть.
10 октября 2018 г. 9:48
ost.vld

Qt/C++ - Урок 083. Создание динамической библиотеки и подключение её в другой проект

О плагинах к QtCreator в целом, тоже интересно.
10 октября 2018 г. 9:46
ost.vld

Qt/C++ - Урок 083. Создание динамической библиотеки и подключение её в другой проект

ну типа того, создание программы, функционал которой можно расширять плагинами, и, в перспективе, создание API.
Сейчас обсуждают на форуме
15 октября 2018 г. 12:45
Allyans .

QGraphicsItem change color

Хорошо)
11 октября 2018 г. 10:13
Arrow

Работа с WebView в QML

Нашел в чем проблема. Пишу на случай если кому-то попадется такое же счастье с WebView как и мне. Проблема как оказалась с Debug версией, так как в Release и Profile все работает (...
10 октября 2018 г. 12:49
Виталий Антипов

Кто что делает на Qt?

Работаем по локальной сети. Файл базы, схемы и фото лежат на сервере. Чтобы не было проблем при одновременной работе с одним файлом, все запросы обернул в транзакции, как указано в документаци...
10 октября 2018 г. 11:21
Arrow

Редактирование записи на форуме

Добрый день! К сожалению встретил небольшой баг на форуме. При создании записи на форуме и вставке кода, через соответствующую кнопку номера строк проставляются верно, но каждый...
10 октября 2018 г. 9:46
Arrow

Настройка Qt Creator для Android

Я Genymotion  ставил с VBox в комплекте для личного использования, после добавил из списка телефон с нужным API. Запустил его и при компиляции Qt Creator сам нашел его и все запустилось...
Присоединяйтесь к нам в социальных сетях