Mercurial Repositories > eric / file diff

--- a/Plugins/CheckerPlugins/Pep8/Pep8CodeSelectionDialog.py	Sat Sep 14 18:09:53 2013 +0200
+++ b/Plugins/CheckerPlugins/Pep8/Pep8CodeSelectionDialog.py	Sun Sep 15 19:55:39 2013 +0200
@@ -12,9 +12,12 @@
 
 from . import pep8
 from .Pep8NamingChecker import Pep8NamingChecker
+from .Pep257Checker import Pep257Checker
 
 from .Ui_Pep8CodeSelectionDialog import Ui_Pep8CodeSelectionDialog
 
+import UI.PixmapCache
+
 
 class Pep8CodeSelectionDialog(QDialog, Ui_Pep8CodeSelectionDialog):
     """
@@ -40,6 +43,7 @@
         else:
             selectableCodes = list(pep8.pep8_messages.keys())
             selectableCodes.extend(Pep8NamingChecker.Messages.keys())
+            selectableCodes.extend(Pep257Checker.Messages.keys())
         for code in sorted(selectableCodes):
             if code in pep8.pep8_messages_sample_args:
                 message = QCoreApplication.translate(
@@ -52,9 +56,20 @@
                 message = QCoreApplication.translate(
                     "Pep8NamingChecker",
                     Pep8NamingChecker.Messages[code])
+            elif code in Pep257Checker.Messages:
+                message = QCoreApplication.translate(
+                    "Pep257Checker", Pep257Checker.Messages[code])
             else:
                 continue
             itm = QTreeWidgetItem(self.codeTable, [code, message])
+            if code.startswith("W"):
+                itm.setIcon(0, UI.PixmapCache.getIcon("warning.png"))
+            elif code.startswith("E"):
+                itm.setIcon(0, UI.PixmapCache.getIcon("syntaxError.png"))
+            elif code.startswith("N"):
+                itm.setIcon(0, UI.PixmapCache.getIcon("namingError.png"))
+            elif code.startswith("D"):
+                itm.setIcon(0, UI.PixmapCache.getIcon("docstringError.png"))
             if code in codeList:
                 itm.setSelected(True)
                 codeList.remove(code)