Differential D6385 Diff 15861 plugins/extensions/pykrita/plugin/plugins/filtermanager/components/filtermanagertreemodel.py
Changeset View
Changeset View
Standalone View
Standalone View
plugins/extensions/pykrita/plugin/plugins/filtermanager/components/filtermanagertreemodel.py
| 1 | from PyQt5.QtCore import QAbstractItemModel, QFile, QIODevice, QModelIndex, Qt | 1 | from PyQt5.QtCore import QAbstractItemModel, QFile, QIODevice, QModelIndex, Qt | ||
|---|---|---|---|---|---|
| 2 | from PyQt5.QtWidgets import QApplication, QTreeView | 2 | from PyQt5.QtWidgets import QApplication, QTreeView | ||
| 3 | from filtermanager.components import filtermanagertreeitem | 3 | from filtermanager.components import filtermanagertreeitem | ||
| 4 | from PyQt5.QtGui import QPixmap | 4 | from PyQt5.QtGui import QPixmap | ||
| 5 | 5 | | |||
| 6 | 6 | | |||
| 7 | class FilterManagerTreeModel(QAbstractItemModel): | 7 | class FilterManagerTreeModel(QAbstractItemModel): | ||
| 8 | 8 | | |||
| 9 | TYPE_COLUMN = 1 | ||||
| 9 | NODE_COLUMN = 3 | 10 | NODE_COLUMN = 3 | ||
| 10 | DOCUMENT_COLUMN = 4 | 11 | DOCUMENT_COLUMN = 4 | ||
| 11 | 12 | | |||
| 12 | def __init__(self, uiFilterManager, parent=None): | 13 | def __init__(self, uiFilterManager, parent=None): | ||
| 13 | super(FilterManagerTreeModel, self).__init__(parent) | 14 | super(FilterManagerTreeModel, self).__init__(parent) | ||
| 14 | 15 | | |||
| 15 | self.rootItem = filtermanagertreeitem.FilterManagerTreeItem(("Name", "Type", "Thumbnail")) | 16 | self.rootItem = filtermanagertreeitem.FilterManagerTreeItem(("Name", "Type", "Thumbnail")) | ||
| 16 | self.uiFilterManager = uiFilterManager | 17 | self.uiFilterManager = uiFilterManager | ||
| ▲ Show 20 Lines • Show All 50 Lines • ▼ Show 20 Line(s) | 64 | def data(self, index, role): | |||
| 67 | item = index.internalPointer() | 68 | item = index.internalPointer() | ||
| 68 | 69 | | |||
| 69 | if role == Qt.UserRole + 1: | 70 | if role == Qt.UserRole + 1: | ||
| 70 | return item.data(self.NODE_COLUMN) | 71 | return item.data(self.NODE_COLUMN) | ||
| 71 | 72 | | |||
| 72 | if role == Qt.UserRole + 2: | 73 | if role == Qt.UserRole + 2: | ||
| 73 | return item.data(self.DOCUMENT_COLUMN) | 74 | return item.data(self.DOCUMENT_COLUMN) | ||
| 74 | 75 | | |||
| 76 | if role == Qt.UserRole + 3: | ||||
| 77 | return item.data(self.TYPE_COLUMN) | ||||
| 78 | | ||||
| 75 | if role != Qt.DisplayRole and role != Qt.DecorationRole: | 79 | if role != Qt.DisplayRole and role != Qt.DecorationRole: | ||
| 76 | return None | 80 | return None | ||
| 77 | 81 | | |||
| 78 | return item.data(index.column()) | 82 | return item.data(index.column()) | ||
| 79 | 83 | | |||
| 80 | def flags(self, index): | 84 | def flags(self, index): | ||
| 81 | if not index.isValid(): | 85 | if not index.isValid(): | ||
| 82 | return Qt.NoItemFlags | 86 | return Qt.NoItemFlags | ||
| Show All 37 Lines | |||||