eric6/E5Gui/E5TreeView.py

changeset 6942
2602857055c5
parent 6645
ad476851d7e0
child 7229
53054eb5b15a
diff -r f99d60d6b59b -r 2602857055c5 eric6/E5Gui/E5TreeView.py
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/eric6/E5Gui/E5TreeView.py	Sun Apr 14 15:09:21 2019 +0200
@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2009 - 2019 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing specialized tree views.
+"""
+
+from __future__ import unicode_literals
+
+from PyQt5.QtCore import Qt
+from PyQt5.QtWidgets import QTreeView
+
+
+class E5TreeView(QTreeView):
+    """
+    Class implementing a tree view supporting removal of entries.
+    """
+    def keyPressEvent(self, evt):
+        """
+        Protected method implementing special key handling.
+        
+        @param evt reference to the event (QKeyEvent)
+        """
+        if evt.key() in [Qt.Key_Delete, Qt.Key_Backspace] and \
+           self.model() is not None:
+            self.removeSelected()
+            evt.setAccepted(True)
+        else:
+            super(E5TreeView, self).keyPressEvent(evt)
+    
+    def removeSelected(self):
+        """
+        Public method to remove the selected entries.
+        """
+        if self.model() is None or \
+           self.selectionModel() is None or \
+           not self.selectionModel().hasSelection():
+            # no models available or nothing selected
+            return
+        
+        selectedRows = self.selectionModel().selectedRows()
+        for idx in reversed(sorted(selectedRows)):
+            self.model().removeRow(idx.row(), idx.parent())
+    
+    def removeAll(self):
+        """
+        Public method to clear the view.
+        """
+        if self.model() is not None:
+            self.model().removeRows(0, self.model().rowCount(self.rootIndex()),
+                                    self.rootIndex())

eric ide

mercurial