From e83f0fef0fc7a8b642fdd46c018b93f250bd6538 Mon Sep 17 00:00:00 2001 From: RumovZ Date: Sun, 28 Feb 2021 22:36:31 +0100 Subject: [PATCH] Fix Qt types --- qt/aqt/sidebar.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/qt/aqt/sidebar.py b/qt/aqt/sidebar.py index e19b3595c..d39cb057d 100644 --- a/qt/aqt/sidebar.py +++ b/qt/aqt/sidebar.py @@ -229,7 +229,9 @@ class SidebarModel(QAbstractItemModel): return QVariant(item.tooltip) return QVariant(theme_manager.icon_from_resources(item.icon)) - def setData(self, index: QModelIndex, text: str, _role: int) -> bool: + def setData( + self, index: QModelIndex, text: QVariant, _role: int = Qt.EditRole + ) -> bool: return self.sidebar.rename_node(index.internalPointer(), text) def supportedDropActions(self) -> Qt.DropActions: @@ -390,19 +392,20 @@ class SidebarTreeView(QTreeView): def tool(self, tool: SidebarTool) -> None: self._tool = tool if tool == SidebarTool.SELECT: - # pylint: disable=no-member - selection_mode = QAbstractItemView.ExtendedSelection # type: ignore + selection_mode = QAbstractItemView.ExtendedSelection drag_drop_mode = QAbstractItemView.NoDragDrop edit_triggers = QAbstractItemView.EditKeyPressed elif tool == SidebarTool.SEARCH: - # pylint: disable=no-member - selection_mode = QAbstractItemView.SingleSelection # type: ignore + selection_mode = QAbstractItemView.SingleSelection drag_drop_mode = QAbstractItemView.NoDragDrop edit_triggers = QAbstractItemView.EditKeyPressed elif tool == SidebarTool.EDIT: - selection_mode = QAbstractItemView.SingleSelection # type: ignore + selection_mode = QAbstractItemView.SingleSelection drag_drop_mode = QAbstractItemView.InternalMove - edit_triggers = QAbstractItemView.DoubleClicked | QAbstractItemView.EditKeyPressed + edit_triggers = cast( + QAbstractItemView.EditTriggers, + QAbstractItemView.DoubleClicked | QAbstractItemView.EditKeyPressed, + ) self.setSelectionMode(selection_mode) self.setDragDropMode(drag_drop_mode) self.setEditTriggers(edit_triggers)