Refactor plugin to group patches and augmentations

Author: reverbc

CHANGELOG.md

@@ -4,6 +4,9 @@
 ### Added
 - Suppressing FP `no-member` from [using workaround of accessing cls in setup fixture](https://github.com/pytest-dev/pytest/issues/3778#issuecomment-411899446)
 
+### Changed
+- Refactor plugin to group patches and augmentations
+
 ## [0.1.2] - 2020-05-22
 ### Fixed
 - Fix fixtures defined with `@pytest.yield_fixture` decorator still showing FP

pylint_pytest.py

@@ -0,0 +1,30 @@
+import os
+import sys
+import inspect
+import importlib
+import glob
+
+from .checkers import BasePytestChecker
+
+
+# pylint: disable=protected-access
+def register(linter):
+    '''auto discover pylint checker classes'''
+    dirname = os.path.dirname(__file__)
+    for module in glob.glob(os.path.join(dirname, 'checkers', '*.py')):
+        # trim file extension
+        module = os.path.splitext(module)[0]
+
+        # use relative path only
+        module = module.replace(dirname, '', 1)
+
+        # translate file path into module import path
+        module = module.replace('/', '.')
+
+        checker = importlib.import_module(module, package=os.path.basename(dirname))
+        for attr_name in dir(checker):
+            attr_val = getattr(checker, attr_name)
+            if attr_val != BasePytestChecker and \
+                    inspect.isclass(attr_val) and \
+                    issubclass(attr_val, BasePytestChecker):
+                linter.register_checker(attr_val(linter))

pylint_pytest/__init__.py

@@ -0,0 +1,5 @@
+from pylint.checkers import BaseChecker
+
+
+class BasePytestChecker(BaseChecker):
+    name = 'pylint-pytest'