Display variables in tabs

Display every variable context in a separate tab.

Author: robertbasic

pugdebug/gui/variables.py

@@ -10,24 +10,24 @@
 __author__ = "robertbasic"
 
 import base64
-from PyQt5.QtWidgets import (QTreeWidget, QTreeWidgetItem, QDialog,
+
+from PyQt5.QtWidgets import (QTabWidget, QTreeWidget, QTreeWidgetItem, QDialog,
                              QTextEdit, QGridLayout)
 
 
-class PugdebugVariableViewer(QTreeWidget):
+class PugdebugVariableViewer(QTabWidget):
 
-    def __init__(self):
-        super(PugdebugVariableViewer, self).__init__()
+    variable_tables = {}
 
-        self.setColumnCount(3)
-        self.setHeaderLabels(['Name', 'Type', 'Value'])
+    def __init__(self):
+        """Variable viewer
 
-        self.setColumnWidth(0, 250)
-        self.setColumnWidth(1, 150)
+        Every variable context is displayed in a table,
+        contexts are displayed in tabs.
+        """
+        super(PugdebugVariableViewer, self).__init__()
 
-        self.itemDoubleClicked.connect(
-            self.handle_variable_double_clicked
-        )
+        self.setTabsClosable(False)
 
     def handle_variable_double_clicked(self, item):
         """Handle when a variable is double clicked
@@ -40,19 +40,42 @@ def handle_variable_double_clicked(self, item):
             PugdebugVariableDetails(self, item)
 
     def set_variables(self, variables):
-        self.clear()
+        for context in variables:
+            table = self.get_variable_table(context)
+
+            table.clear()
+
+            for var in variables[context]:
+                self.add_variable(table, var)
+
+    def get_variable_table(self, context):
+        context_key = context.replace(' ', '-').lower()
+
+        if context_key in self.variable_tables:
+            table = self.variable_tables[context_key]
+        else:
+            table = QTreeWidget()
+            table.setColumnCount(3)
+            table.setHeaderLabels(['Name', 'Type', 'Value'])
+            table.setColumnWidth(0, 250)
+            table.setColumnWidth(1, 150)
+
+            self.variable_tables[context_key] = table
+
+            if context == 'Locals':
+                self.insertTab(0, table, context)
+            else:
+                self.addTab(table, context)
+
+            table.itemDoubleClicked.connect(
+                self.handle_variable_double_clicked
+            )
 
-        if 'Locals' in variables:
-            for variable in variables['Locals']:
-                self.add_variable(variable)
+            self.setCurrentIndex(0)
 
-        if 'Superglobals' in variables:
-            item = QTreeWidgetItem(['Superglobals', '', ''])
-            self.addTopLevelItem(item)
-            for variable in variables['Superglobals']:
-                self.add_variable(variable, item)
+        return table
 
-    def add_variable(self, variable, parent=None):
+    def add_variable(self, table, variable, parent=None):
         type = variable['type']
         tooltip = None
 
@@ -90,10 +113,10 @@ def add_variable(self, variable, parent=None):
 
         if 'variables' in variable:
             for subvar in variable['variables']:
-                self.add_variable(subvar, item)
+                self.add_variable(table, subvar, item)
 
         if parent is None:
-            self.addTopLevelItem(item)
+            table.addTopLevelItem(item)
         else:
             parent.addChild(item)