Mercurial Repositories > eric / file diff

--- a/eric6/E5Gui/E5TreeWidget.py	Tue Apr 27 17:42:00 2021 +0200
+++ b/eric6/E5Gui/E5TreeWidget.py	Wed Apr 28 19:42:28 2021 +0200
@@ -7,10 +7,20 @@
 Module implementing specialized tree views.
 """
 
+import enum
+
 from PyQt5.QtCore import pyqtSignal, Qt
 from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem, QAbstractItemView
 
 
+class E5TreeWidgetItemsState(enum.Enum):
+    """
+    Class defining the items expansion state.
+    """
+    COLLAPSED = 0
+    EXPANDED = 1
+
+
 class E5TreeWidget(QTreeWidget):
     """
     Class implementing an extended tree widget.
@@ -20,9 +30,6 @@
     @signal itemMiddleButtonClicked(QTreeWidgetItem) emitted after a click
             of the middle button on an item
     """
-    ItemsCollapsed = 0
-    ItemsExpanded = 1
-    
     itemControlClicked = pyqtSignal(QTreeWidgetItem)
     itemMiddleButtonClicked = pyqtSignal(QTreeWidgetItem)
     
@@ -36,7 +43,7 @@
         
         self.__refreshAllItemsNeeded = True
         self.__allTreeItems = []
-        self.__showMode = E5TreeWidget.ItemsCollapsed
+        self.__showMode = E5TreeWidgetItemsState.COLLAPSED
         
         self.setVerticalScrollMode(QAbstractItemView.ScrollMode.ScrollPerPixel)
         
@@ -46,13 +53,9 @@
         """
         Public method to set the default item show mode.
         
-        @param mode default mode (ItemsCollapsed, ItemsExpanded)
-        @exception ValueError raised to indicate a bad parameter value
+        @param mode default mode
+        @type E5TreeWidgetItemsState
         """
-        if mode not in [E5TreeWidget.ItemsCollapsed,
-                        E5TreeWidget.ItemsExpanded]:
-            raise ValueError("Bad value for 'mode' parameter.")
-        
         self.__showMode = mode
     
     def allItems(self):
@@ -229,7 +232,7 @@
                 itm.setHidden(False)
             for index in range(self.topLevelItemCount()):
                 self.topLevelItem(index).setHidden(False)
-            if self.__showMode == E5TreeWidget.ItemsCollapsed:
+            if self.__showMode == E5TreeWidgetItemsState.COLLAPSED:
                 self.collapseAll()
     
     def clear(self):