Added error registry.

Author: Nurdok

pep257.py

@@ -442,50 +442,83 @@ def __lt__(self, other):
         return (self.filename, self.line) < (other.filename, other.line)
 
 
-errors = []
-
-
-def register_error(error_cls):
-    errors.append(error_cls)
-
-
-def error(code, message):
-    _code = code
-
-    class _Error(Error):
-        code = _code
-        _message = message
-
-    register_error(_Error)
-    return _Error
-
-
-D100 = error('D100', 'Missing docstring in public module')
-D101 = error('D101', 'Missing docstring in public class')
-D102 = error('D102', 'Missing docstring in public method')
-D103 = error('D103', 'Missing docstring in public function')
-D200 = error('D200', 'One-line docstring should fit on one line with quotes, '
-                     'found %s')
-D201 = error('D201', 'No blank lines allowed before function docstring, '
-                     'found %s')
-D202 = error('D202', 'No blank lines allowed after function docstring, '
-                     'found %s')
-D203 = error('D203', '1 blank line required before class docstring, found %s')
-D204 = error('D204', '1 blank line required after class docstring, found %s')
-D205 = error('D205', '1 blank line required between summary line and '
-                     'description, found %s')
-D206 = error('D206', 'Docstring should be indented with spaces, not tabs')
-D207 = error('D207', 'Docstring is under-indented')
-D208 = error('D208', 'Docstring is over-indented')
-D209 = error('D209', 'Multi-line docstring closing quotes should be on a '
-                     'separate line')
-D210 = error('D210', 'No whitespaces allowed surrounding docstring text')
-D300 = error('D300', 'Use """triple double quotes""", found %s-quotes')
-D301 = error('D301', 'Use r""" if any backslashes in a docstring')
-D302 = error('D302', 'Use u""" for Unicode docstrings')
-D400 = error('D400', 'First line should end with a period, not %r')
-D401 = error('D401', 'First line should be in imperative mood: %r, not %r')
-D402 = error('D402', 'First line should not be the function\'s "signature"')
+class ErrorRegistry(object):
+    groups = []
+
+    class ErrorGroup(object):
+        errors = []
+
+        def __init__(self, prefix, name):
+            self.prefix = prefix
+            self.name = name
+
+        def create_error(self, error_code, error_message):
+            # TODO: check prefix
+
+            class _Error(Error):
+                code = error_code
+                _message = error_message
+
+            self.errors.append(_Error)
+            return _Error
+
+    @classmethod
+    def create_group(cls, prefix, name):
+        group = cls.ErrorGroup(prefix, name)
+        cls.groups.append(group)
+        return group
+
+    @classmethod
+    def get_error_codes(cls):
+        # TODO: implement
+        pass
+
+    @classmethod
+    def to_rst(cls):
+        # TODO: implement
+        pass
+
+
+D1xx = ErrorRegistry.create_group('D1', 'Missing Docstrings')
+D100 = D1xx.create_error('D100', 'Missing docstring in public module')
+D101 = D1xx.create_error('D101', 'Missing docstring in public class')
+D102 = D1xx.create_error('D102', 'Missing docstring in public method')
+D103 = D1xx.create_error('D103', 'Missing docstring in public function')
+
+D2xx = ErrorRegistry.create_group('D2', 'Whitespace Issues')
+D200 = D2xx.create_error('D200', 'One-line docstring should fit on one line '
+                                 'with quotes, found %s')
+D201 = D2xx.create_error('D201', 'No blank lines allowed before function '
+                                 'docstring, found %s')
+D202 = D2xx.create_error('D202', 'No blank lines allowed after function '
+                                 'docstring, found %s')
+D203 = D2xx.create_error('D203', '1 blank line required before class '
+                                 'docstring, found %s')
+D204 = D2xx.create_error('D204', '1 blank line required after class '
+                                 'docstring, found %s')
+D205 = D2xx.create_error('D205', '1 blank line required between summary line '
+                                 'and description, found %s')
+D206 = D2xx.create_error('D206', 'Docstring should be indented with spaces, '
+                                 'not tabs')
+D207 = D2xx.create_error('D207', 'Docstring is under-indented')
+D208 = D2xx.create_error('D208', 'Docstring is over-indented')
+D209 = D2xx.create_error('D209', 'Multi-line docstring closing quotes should '
+                                 'be on a separate line')
+D210 = D2xx.create_error('D210', 'No whitespaces allowed surrounding '
+                                 'docstring text')
+
+D3xx = ErrorRegistry.create_group('D3', 'Quotes Issues')
+D300 = D3xx.create_error('D300', 'Use """triple double quotes""", found '
+                                 '%s-quotes')
+D301 = D3xx.create_error('D301', 'Use r""" if any backslashes in a docstring')
+D302 = D3xx.create_error('D302', 'Use u""" for Unicode docstrings')
+
+D4xx = ErrorRegistry.create_group('D4', 'Docstring Content Issues')
+D400 = D4xx.create_error('D400', 'First line should end with a period, not %r')
+D401 = D4xx.create_error('D401', 'First line should be in imperative mood: '
+                                 '%r, not %r')
+D402 = D4xx.create_error('D402', 'First line should not be the function\'s '
+                                 '"signature"')
 
 
 def get_option_parser():