code cleanup

Author: robertbasic

pugdebug/gui/documents.py

@@ -9,9 +9,9 @@
 
 __author__ = "robertbasic"
 
+from PyQt5.QtCore import Qt, pyqtSignal
+from PyQt5.QtWidgets import QTabWidget, QTabBar
 
-from PyQt5.QtWidgets import QTabWidget
-from pugdebug.gui.widgets.tabbar import PugdebugTabBar;
 
 class PugdebugDocumentViewer(QTabWidget):
 
@@ -75,3 +75,40 @@ def get_document_by_path(self, path):
     def remove_line_highlights(self):
         for index, path in self.tabs.items():
             self.widget(index).remove_line_highlights()
+
+
+class PugdebugTabBar(QTabBar):
+    """Adds a signal when the middle button is clicked to the QTabBar widget.
+
+    A click is defined as a press event followed by a release event over the
+    same tab.
+
+    Stolen from: http://stackoverflow.com/a/9445581/84245
+    """
+    middle_clicked_signal = pyqtSignal(int)
+
+    def __init__(self):
+        super(QTabBar, self).__init__()
+        self.previousMiddleIndex = -1
+
+    def mousePressEvent(self, mouseEvent):
+        """Triggered when a mouse button is pressed.
+
+        If it's the middle button, remember which over which tab it
+        was pressed.
+        """
+        if mouseEvent.button() == Qt.MidButton:
+            self.previousIndex = self.tabAt(mouseEvent.pos())
+        QTabBar.mousePressEvent(self, mouseEvent)
+
+    def mouseReleaseEvent(self, mouseEvent):
+        """Triggered when a mouse button is released.
+
+        If a middle button was pressed previously, and if it is released over
+        the same tab, emit the signal.
+        """
+        if (mouseEvent.button() == Qt.MidButton and
+                self.previousIndex == self.tabAt(mouseEvent.pos())):
+            self.middle_clicked_signal.emit(self.previousIndex)
+        self.previousIndex = -1
+        QTabBar.mouseReleaseEvent(self, mouseEvent)

pugdebug/gui/expressions.py

@@ -157,7 +157,7 @@ def set_variable(self, parent, index, variable):
         value = self.decode_value(variable)
 
         item = parent.child(index)
-        if item == None:
+        if item is None:
             # Item does not exist, create it
             item = QTreeWidgetItem([name, type, value])
             parent.insertChild(index, item)

pugdebug/gui/stacktraces.py

@@ -9,7 +9,7 @@
 
 __author__ = "robertbasic"
 
-from PyQt5.QtCore import Qt, pyqtSignal
+from PyQt5.QtCore import pyqtSignal
 from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem