Changeset View
Changeset View
Standalone View
Standalone View
plugins/python/filtermanager/components/filtermanagertreemodel.py
| 1 | ''' | 1 | # This script is licensed CC 0 1.0, so that you can learn from it. | ||
|---|---|---|---|---|---|
| 2 | This script is licensed CC 0 1.0, so that you can learn from it. | | |||
| 3 | 2 | | |||
| 4 | ------ CC 0 1.0 --------------- | 3 | # ------ CC 0 1.0 --------------- | ||
| 5 | 4 | | |||
| 6 | The person who associated a work with this deed has dedicated the work to the public domain by waiving all of his or her rights to the work worldwide under copyright law, including all related and neighboring rights, to the extent allowed by law. | 5 | # The person who associated a work with this deed has dedicated the | ||
| 6 | # work to the public domain by waiving all of his or her rights to the | ||||
| 7 | # work worldwide under copyright law, including all related and | ||||
| 8 | # neighboring rights, to the extent allowed by law. | ||||
| 7 | 9 | | |||
| 8 | You can copy, modify, distribute and perform the work, even for commercial purposes, all without asking permission. | 10 | # You can copy, modify, distribute and perform the work, even for | ||
| 11 | # commercial purposes, all without asking permission. | ||||
| 9 | 12 | | |||
| 10 | https://creativecommons.org/publicdomain/zero/1.0/legalcode | 13 | # https://creativecommons.org/publicdomain/zero/1.0/legalcode | ||
| 11 | ''' | 14 | | ||
| 12 | from PyQt5.QtCore import QAbstractItemModel, QFile, QIODevice, QModelIndex, Qt | 15 | from PyQt5.QtCore import QAbstractItemModel, QModelIndex, Qt | ||
| 13 | from PyQt5.QtWidgets import QApplication, QTreeView | | |||
| 14 | from . import filtermanagertreeitem | | |||
| 15 | from PyQt5.QtGui import QPixmap | 16 | from PyQt5.QtGui import QPixmap | ||
| 17 | from . import filtermanagertreeitem | ||||
| 16 | 18 | | |||
| 17 | 19 | | |||
| 18 | class FilterManagerTreeModel(QAbstractItemModel): | 20 | class FilterManagerTreeModel(QAbstractItemModel): | ||
| 19 | 21 | | |||
| 20 | TYPE_COLUMN = 1 | 22 | TYPE_COLUMN = 1 | ||
| 21 | NODE_COLUMN = 3 | 23 | NODE_COLUMN = 3 | ||
| 22 | DOCUMENT_COLUMN = 4 | 24 | DOCUMENT_COLUMN = 4 | ||
| 23 | 25 | | |||
| 24 | def __init__(self, uiFilterManager, parent=None): | 26 | def __init__(self, uiFilterManager, parent=None): | ||
| 25 | super(FilterManagerTreeModel, self).__init__(parent) | 27 | super(FilterManagerTreeModel, self).__init__(parent) | ||
| 26 | 28 | | |||
| 27 | self.rootItem = filtermanagertreeitem.FilterManagerTreeItem(("Name", "Type", "Thumbnail")) | 29 | self.rootItem = filtermanagertreeitem.FilterManagerTreeItem( | ||
| 30 | ("Name", "Type", "Thumbnail")) | ||||
| 28 | self.uiFilterManager = uiFilterManager | 31 | self.uiFilterManager = uiFilterManager | ||
| 29 | self._loadTreeModel(self.rootItem) | 32 | self._loadTreeModel(self.rootItem) | ||
| 30 | 33 | | |||
| 31 | def index(self, row, column, parent): | 34 | def index(self, row, column, parent): | ||
| 32 | if not self.hasIndex(row, column, parent): | 35 | if not self.hasIndex(row, column, parent): | ||
| 33 | return QModelIndex() | 36 | return QModelIndex() | ||
| 34 | 37 | | |||
| 35 | if parent.isValid(): | 38 | if parent.isValid(): | ||
| ▲ Show 20 Lines • Show All 70 Lines • ▼ Show 20 Line(s) | |||||
| 106 | 109 | | |||
| 107 | def _loadTreeModel(self, parent): | 110 | def _loadTreeModel(self, parent): | ||
| 108 | for index, document in enumerate(self.uiFilterManager.documents): | 111 | for index, document in enumerate(self.uiFilterManager.documents): | ||
| 109 | rootNode = document.rootNode() | 112 | rootNode = document.rootNode() | ||
| 110 | columnData = (document.fileName(), | 113 | columnData = (document.fileName(), | ||
| 111 | "Document", | 114 | "Document", | ||
| 112 | QPixmap.fromImage(document.thumbnail(30, 30)), | 115 | QPixmap.fromImage(document.thumbnail(30, 30)), | ||
| 113 | rootNode, index) | 116 | rootNode, index) | ||
| 114 | item = filtermanagertreeitem.FilterManagerTreeItem(columnData, parent) | 117 | item = filtermanagertreeitem.FilterManagerTreeItem( | ||
| 118 | columnData, parent) | ||||
| 115 | parent.appendChild(item) | 119 | parent.appendChild(item) | ||
| 116 | 120 | | |||
| 117 | childNodes = rootNode.childNodes() | 121 | childNodes = rootNode.childNodes() | ||
| 118 | if len(childNodes): | 122 | if len(childNodes): | ||
| 119 | self._addSubNodes(childNodes[::-1], item, index) | 123 | self._addSubNodes(childNodes[::-1], item, index) | ||
| 120 | 124 | | |||
| 121 | def _addSubNodes(self, nodes, parent, documentIndex): | 125 | def _addSubNodes(self, nodes, parent, documentIndex): | ||
| 122 | for node in nodes: | 126 | for node in nodes: | ||
| 123 | nodeName = node.name() | 127 | nodeName = node.name() | ||
| 124 | nodeType = node.type() | 128 | nodeType = node.type() | ||
| 125 | columnData = ("Unnamed" if nodeName == '' else nodeName, | 129 | columnData = ("Unnamed" if nodeName == '' else nodeName, | ||
| 126 | "Untyped" if nodeType == '' else nodeType, | 130 | "Untyped" if nodeType == '' else nodeType, | ||
| 127 | QPixmap.fromImage(node.thumbnail(30, 30)), | 131 | QPixmap.fromImage(node.thumbnail(30, 30)), | ||
| 128 | node, documentIndex) | 132 | node, documentIndex) | ||
| 129 | item = filtermanagertreeitem.FilterManagerTreeItem(columnData, parent) | 133 | item = filtermanagertreeitem.FilterManagerTreeItem( | ||
| 134 | columnData, parent) | ||||
| 130 | parent.appendChild(item) | 135 | parent.appendChild(item) | ||
| 131 | 136 | | |||
| 132 | childNodes = node.childNodes() | 137 | childNodes = node.childNodes() | ||
| 133 | if len(childNodes): | 138 | if len(childNodes): | ||
| 134 | self._addSubNodes(childNodes[::-1], item, documentIndex) | 139 | self._addSubNodes(childNodes[::-1], item, documentIndex) | ||