support darglint

orsinium-labs · May 23, 2022 · fc94435 · fc94435
1 parent f3cf4da
commit fc94435
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 2 deletions.
diff --git a/flake8_codes/__init__.py b/flake8_codes/__init__.py
@@ -6,5 +6,5 @@
 from ._plugins import Plugin, get_installed
 
 
-__version__ = '0.2.1'
+__version__ = '0.2.2'
 __all__ = ['extract', 'Plugin', 'get_installed', '__version__']
diff --git a/flake8_codes/_codes/_adhoc.py b/flake8_codes/_codes/_adhoc.py
@@ -264,6 +264,24 @@ def extract_flake8_pie() -> Dict[str, str]:
     return codes
 
 
+@registry.add
+def extract_darglint() -> Dict[str, str]:
+    # external
+    import darglint.errors
+
+    codes = dict()
+    for obj_name in dir(darglint.errors):
+        cls = getattr(darglint.errors, obj_name)
+        if not isinstance(cls, type):
+            continue
+        if not issubclass(cls, darglint.errors.DarglintError):
+            continue
+        if cls.error_code is None:
+            continue
+        codes[cls.error_code] = cls.description
+    return codes
+
+
 @registry.add
 def extract_flake8_docstrings() -> Dict[str, str]:
     # external

diff --git a/flake8_codes/_plugins/_plugin.py b/flake8_codes/_plugins/_plugin.py
@@ -7,6 +7,7 @@
 ALIASES = {
     'aaa': 'flake8-aaa',
     'flake-mutable': 'flake8-mutable',
+    'flake8-darglint': 'darglint',
     'flake8-pandas-vet': 'pandas-vet',
     'import-order': 'flake8-import-order',
     'logging-format': 'flake8-logging-format',

diff --git a/tests/test_codes.py b/tests/test_codes.py
@@ -2,14 +2,19 @@
 import pytest
 
 # project
-from flake8_codes import extract
+from flake8_codes import extract, get_installed
 from flake8_codes._codes._default import extract_default
 from flake8_codes._codes._registry import registry
 
 # app
 from ._constants import KNOWN_PLUGINS
 
 
+def test_can_extract_all_installed():
+    for plugin in get_installed():
+        assert extract(plugin.name)
+
+
 @pytest.mark.parametrize('plugin_name', KNOWN_PLUGINS)
 def test_smoke_extract(plugin_name):
     codes = extract(plugin_name)