messages: add "check_messages" decorator to store messages

refactor PyLintASTWalker.add_checker to check the stored method's messages;
in PyLintASTWalker.add_checker, we only disable methods
that has been decorated and have no activated message.

1 files changed, 9 insertions, 0 deletions

diff --git a/checkers/utils.py b/checkers/utils.py
index 94ac8b0..1cbd006 100644
--- a/checkers/utils.py
+++ b/checkers/utils.py
@@ -202,3 +202,12 @@ PYMETHODS = set(('__new__', '__init__', '__del__', '__hash__',
                  '__xor__', '__ixor__', '__rxor__',
                  # XXX To be continued
                  ))
+
+def check_messages(*messages):
+    """decorator to store messages that are handled by a checker method"""
+
+    def store_messages(func):
+        func.checks_msgs = messages
+        return func
+    return store_messages
+