7741475ae0
* Fix wrong hook being torn down * Fix item models not being destroyed * Add missing gc for FilteredDeckConfigDialog * Add missing type annotation * Pass calling widget as parent to QTimer Implicitly passing `self.mw` as the parent means that the QTimer won't get destroyed before quitting the app, which also thwarts garbage collection of any data captured by a passed closure. * Make `Editor._links` an instance variable Browser is inserting a closure into this dict capturing itself. As a class variable, it won't get destroyed, so neither will the browser. * Make `Editor._links` funcs take instance again * Deprecate calling progress.timer() without parent * show caller location when printing deprecation warning (dae)
125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
# Copyright: Ankitects Pty Ltd and contributors
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
from __future__ import annotations
|
|
|
|
from typing import cast
|
|
|
|
import aqt
|
|
import aqt.browser
|
|
from aqt.browser.sidebar.item import SidebarItem
|
|
from aqt.qt import *
|
|
from aqt.theme import theme_manager
|
|
|
|
|
|
class SidebarModel(QAbstractItemModel):
|
|
def __init__(
|
|
self, sidebar: aqt.browser.sidebar.SidebarTreeView, root: SidebarItem
|
|
) -> None:
|
|
super().__init__(sidebar)
|
|
self.sidebar = sidebar
|
|
self.root = root
|
|
self._cache_rows(root)
|
|
|
|
def _cache_rows(self, node: SidebarItem) -> None:
|
|
"Cache index of children in parent."
|
|
for row, item in enumerate(node.children):
|
|
item._row_in_parent = row
|
|
self._cache_rows(item)
|
|
|
|
def item_for_index(self, idx: QModelIndex) -> SidebarItem:
|
|
return idx.internalPointer()
|
|
|
|
def index_for_item(self, item: SidebarItem) -> QModelIndex:
|
|
return self.createIndex(item._row_in_parent, 0, item)
|
|
|
|
def search(self, text: str) -> bool:
|
|
return self.root.search(text.lower())
|
|
|
|
# Qt API
|
|
######################################################################
|
|
|
|
def rowCount(self, parent: QModelIndex = QModelIndex()) -> int:
|
|
if not parent.isValid():
|
|
return len(self.root.children)
|
|
else:
|
|
item: SidebarItem = parent.internalPointer()
|
|
return len(item.children)
|
|
|
|
def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
|
|
return 1
|
|
|
|
def index(
|
|
self, row: int, column: int, parent: QModelIndex = QModelIndex()
|
|
) -> QModelIndex:
|
|
if not self.hasIndex(row, column, parent):
|
|
return QModelIndex()
|
|
|
|
parentItem: SidebarItem
|
|
if not parent.isValid():
|
|
parentItem = self.root
|
|
else:
|
|
parentItem = parent.internalPointer()
|
|
|
|
item = parentItem.children[row]
|
|
return self.createIndex(row, column, item)
|
|
|
|
def parent(self, child: QModelIndex) -> QModelIndex: # type: ignore
|
|
if not child.isValid():
|
|
return QModelIndex()
|
|
|
|
childItem: SidebarItem = child.internalPointer()
|
|
parentItem = childItem._parent_item
|
|
|
|
if parentItem is None or parentItem == self.root:
|
|
return QModelIndex()
|
|
|
|
row = parentItem._row_in_parent
|
|
|
|
return self.createIndex(row, 0, parentItem)
|
|
|
|
def data(
|
|
self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole
|
|
) -> QVariant:
|
|
if not index.isValid():
|
|
return QVariant()
|
|
|
|
if role not in (
|
|
Qt.ItemDataRole.DisplayRole,
|
|
Qt.ItemDataRole.DecorationRole,
|
|
Qt.ItemDataRole.ToolTipRole,
|
|
Qt.ItemDataRole.EditRole,
|
|
):
|
|
return QVariant()
|
|
|
|
item: SidebarItem = index.internalPointer()
|
|
|
|
if role in (Qt.ItemDataRole.DisplayRole, Qt.ItemDataRole.EditRole):
|
|
return QVariant(item.name)
|
|
if role == Qt.ItemDataRole.ToolTipRole:
|
|
return QVariant(item.tooltip)
|
|
return QVariant(theme_manager.icon_from_resources(item.icon))
|
|
|
|
def setData(
|
|
self, index: QModelIndex, text: str, _role: int = Qt.ItemDataRole.EditRole
|
|
) -> bool:
|
|
return self.sidebar._on_rename(index.internalPointer(), text)
|
|
|
|
def supportedDropActions(self) -> Qt.DropAction:
|
|
return cast(Qt.DropAction, Qt.DropAction.MoveAction)
|
|
|
|
def flags(self, index: QModelIndex) -> Qt.ItemFlag:
|
|
if not index.isValid():
|
|
return cast(Qt.ItemFlag, Qt.ItemFlag.ItemIsEnabled)
|
|
flags = (
|
|
Qt.ItemFlag.ItemIsEnabled
|
|
| Qt.ItemFlag.ItemIsSelectable
|
|
| Qt.ItemFlag.ItemIsDragEnabled
|
|
)
|
|
item: SidebarItem = index.internalPointer()
|
|
if item.item_type in self.sidebar.valid_drop_types:
|
|
flags |= Qt.ItemFlag.ItemIsDropEnabled
|
|
if item.item_type.is_editable():
|
|
flags |= Qt.ItemFlag.ItemIsEditable
|
|
|
|
return flags
|