Radon Code Metrics: comparison RadonMetrics/CyclomaticComplexityDialog.py

-:5f206edea27e
+:62ffe3d426e5
 sys.path.insert(0, os.path.dirname(__file__))
 from PyQt5.QtCore import pyqtSlot, qVersion, Qt, QTimer, QLocale
 from PyQt5.QtWidgets import (
 QDialog, QDialogButtonBox, QAbstractButton, QHeaderView, QTreeWidgetItem,
-QApplication
+QApplication, QMenu
 )
 from .Ui_CyclomaticComplexityDialog import Ui_CyclomaticComplexityDialog
 from E5Gui.E5Application import e5App
 "<tr><td><b>F</b></td><td>Function</td></tr>"
 "<tr><td><b>M</b></td><td>Method</td></tr>"
 "</table>"
 ))
+self.__mappedType = {
+"class": "C",
+"function": "F",
+"method": "M",
+}
+self.__typeColors = {
+"class": Qt.blue,
+"function": Qt.darkCyan,
+"method": Qt.magenta,
+}
 self.__rankColors = {
 "A": Qt.green,
 "B": Qt.green,
 "C": Qt.yellow,
 "D": Qt.yellow,
 "E": Qt.red,
 "F": Qt.red,
 }
+self.__menu = QMenu(self)
+self.__menu.addAction(self.tr("Collapse all"),
+self.__resultCollapse)
+self.__menu.addAction(self.tr("Expand all"), self.__resultExpand)
+self.resultList.setContextMenuPolicy(Qt.CustomContextMenu)
+self.resultList.customContextMenuRequested.connect(
+self.__showContextMenu)
 def __resizeResultColumns(self):
 """
 Private method to resize the list columns.
 """
 "{0:6}".format(values["lineno"]),
 ])
 itm.setTextAlignment(2, Qt.Alignment(Qt.AlignRight))
 itm.setTextAlignment(3, Qt.Alignment(Qt.AlignHCenter))
 itm.setTextAlignment(4, Qt.Alignment(Qt.AlignRight))
-if values["rank"] in ["A", "B", "C", "D", "E", "F"]:
+if values["rank"] in self.__rankColors:
 itm.setBackground(3, self.__rankColors[values["rank"]])
+if values["type"] in self.__typeColors:
+itm.setForeground(0, self.__typeColors[values["type"]])
 if "methods" in values:
 itm.setExpanded(True)
 for method in values["methods"]:
 self.__createResultItem(itm, method)
 "F": 0,
 }
 self.__ccSum = 0
 self.__ccCount = 0
-self.__mappedType = {
-"class": "C",
-"function": "F",
-"method": "M",
-}
 if len(self.files) > 0:
 # disable updates of the list for speed
 self.resultList.setUpdatesEnabled(False)
 self.resultList.setSortingEnabled(False)
 from radon.complexity import cc_rank
 if not self.__finished:
 self.__finished = True
-# reenable updates of the list
+# re-enable updates of the list
 self.resultList.setSortingEnabled(True)
+self.resultList.sortItems(0, Qt.AscendingOrder)
 self.resultList.sortItems(1, Qt.AscendingOrder)
 self.resultList.setUpdatesEnabled(True)
 self.cancelled = True
 self.buttonBox.button(QDialogButtonBox.Close).setEnabled(True)
 for filter in filterList:
 fileList = \
 [f for f in fileList if not fnmatch.fnmatch(f, filter)]
 self.start(fileList)
+def __showContextMenu(self, coord):
+"""
+Private slot to show the context menu of the resultlist.
+@param coord the position of the mouse pointer (QPoint)
+"""
+if self.resultList.topLevelItemCount() > 0:
+self.__menu.popup(self.mapToGlobal(coord))
+def __resultCollapse(self):
+"""
+Private slot to collapse all entries of the resultlist.
+"""
+for index in range(self.resultList.topLevelItemCount()):
+self.resultList.topLevelItem(index).setExpanded(False)
+def __resultExpand(self):
+"""
+Private slot to expand all entries of the resultlist.
+"""
+for index in range(self.resultList.topLevelItemCount()):
+self.resultList.topLevelItem(index).setExpanded(True)

Mercurial Repositories > Radon Code Metrics / file comparison

comparison: RadonMetrics/CyclomaticComplexityDialog.py

RadonMetrics/CyclomaticComplexityDialog.py