refactor: Support Griffe 0.26

pawamoy · pawamoy · commit 075735ce8d86 · 2023-04-03T16:18:05.000+02:00
diff --git a/src/mkdocstrings_handlers/python/handler.py b/src/mkdocstrings_handlers/python/handler.py
@@ -12,7 +12,6 @@
 from contextlib import suppress
 from typing import TYPE_CHECKING, Any, BinaryIO, Iterator, Mapping
 
-from griffe.agents.extensions import load_extensions
 from griffe.collections import LinesCollection, ModulesCollection
 from griffe.docstrings.parsers import Parser
 from griffe.exceptions import AliasResolutionError
@@ -25,6 +24,12 @@
 
 from mkdocstrings_handlers.python import rendering
 
+try:
+    from griffe.extensions import load_extensions
+except ImportError:
+    # TODO: remove once support for Griffe 0.25 is dropped
+    from griffe.agents.extensions import load_extensions  # type: ignore[no-redef]
+
 if TYPE_CHECKING:
     from markdown import Markdown
 
