Register lexers automatically

Author: cdce8p

pep_sphinx_extensions/__init__.py

@@ -14,7 +14,7 @@
     get_from_doctree,
     pep_abstract,
 )
-from pep_sphinx_extensions.lexers import pep_lexers
+from pep_sphinx_extensions.lexers import get_custom_lexers
 from pep_sphinx_extensions.pep_processor.html import (
     pep_html_builder,
     pep_html_translator,
@@ -107,7 +107,7 @@ def setup(app: Sphinx) -> dict[str, bool]:
     app.add_directive("withdrawn", pep_banner_directive.WithdrawnBanner)
 
     # Register custom lexers
-    for lexer in pep_lexers:
+    for lexer in get_custom_lexers():
         app.add_lexer(lexer.name, lexer)
 
     # Register event callbacks

pep_sphinx_extensions/lexers/__init__.py

@@ -3,17 +3,19 @@
 
 from __future__ import annotations
 
+import importlib
+import pkgutil
 from typing import TYPE_CHECKING
 
-from .pep823_lexer import Py823ConsoleLexer, Py823Lexer
-from .pep824_lexer import Py824Lexer
-
 if TYPE_CHECKING:
     from pygments.lexer import Lexer
 
 
-pep_lexers: list[type[Lexer]] = [
-    Py823ConsoleLexer,
-    Py823Lexer,
-    Py824Lexer,
-]
+def get_custom_lexers() -> list[type[Lexer]]:
+    lexers: list[type[Lexer]] = []
+    for module_info in pkgutil.walk_packages(__path__, prefix=f"{__name__}."):
+        module = importlib.import_module(module_info.name)
+        if (register_func := getattr(module, "register", None)) is None:
+            continue
+        lexers.extend(register_func())
+    return lexers

pep_sphinx_extensions/lexers/pep823_lexer.py

@@ -38,3 +38,7 @@ def __init__(self, **options):
                 )
 
         super().__init__(tblexer, _ReplaceInnerCode, Other.Traceback, **options)
+
+
+def register():
+    return [Py823Lexer, Py823ConsoleLexer]

pep_sphinx_extensions/lexers/pep824_lexer.py

@@ -19,3 +19,7 @@ class Py824Lexer(PythonLexer):
             inherit,
         ],
     }
+
+
+def register():
+    return [Py824Lexer]