Updates for latest griffe

docs/scripts/_dynamic_doc.py

@@ -1,33 +1,53 @@
 from __future__ import annotations
 
-from pathlib import Path
 import warnings
-from griffe.dataclasses import Docstring
-from griffe.docstrings import numpy
+from pathlib import Path
 
-from cylindra import _shared_doc, start, instance
+from griffe import Docstring, Parser, parsers
+
+from cylindra import _shared_doc, instance, start
 
 warnings.simplefilter("ignore", DeprecationWarning)
 
-parse_orig = numpy._read_parameters
+parse_orig = parsers[Parser.numpy]
 
 DOCS = Path(__file__).parent.parent
 PATH_13_3 = DOCS.parent / "tests" / "13pf_MT.tif"
 
 
-def on_startup(**kwargs):
-    def _read_parameters(docstring: Docstring, **kwargs):
-        if docstring.value:
-            docstring.value = _shared_doc.update_doc(docstring.value, indent=0)
-        return parse_orig(docstring, **kwargs)
+def _dynamic_parse(docstring: Docstring, **kwargs):
+    if docstring.value:
+        docstring.value = _shared_doc.update_doc(docstring.value, indent=0)
+        if ">>> " in docstring.value:
+            docstring.value = _replace_codes(docstring.value)
+    return parse_orig(docstring, **kwargs)
 
-    numpy._read_parameters = _read_parameters
+
+def _replace_codes(docstring: str):
+    lines = docstring.split("\n")
+    is_code = False
+    for i in range(len(lines)):
+        if is_code:
+            if lines[i].startswith((">>> ", "... ")):
+                lines[i] = lines[i][4:]
+            else:
+                is_code = False
+                lines[i] = f"```\n{lines[i]}"
+        else:
+            if lines[i].startswith(">>> "):
+                lines[i] = f"``` python\n{lines[i][4:]}"
+                is_code = True
+    return "\n".join(lines)
+
+
+def on_startup(**kwargs):
+    parsers[Parser.numpy] = _dynamic_parse
     ui = start()
     ui.open_image(PATH_13_3)
 
 
 def on_shutdown(**kwargs):
-    numpy._read_parameters = parse_orig
+    parsers[Parser.numpy] = parse_orig
     ui = instance()
     try:
         ui.parent_viewer.close()

docs/scripts/_screenshots.py

@@ -6,7 +6,7 @@
 from magicgui import magicgui
 from qtpy import QtWidgets as QtW
 
-from cylindra import instance
+from cylindra import instance, widgets
 
 DOCS = Path(__file__).parent.parent
 PATH_13_3 = DOCS.parent / "tests" / "13pf_MT.tif"
@@ -22,9 +22,11 @@ def _imsave(widget: QtW.QWidget, name: str):
         widget.grab().save(f.name, "png")
 
 
-def _viewer_screenshot(ui, name: str, canvas_only: bool = True):
+def _viewer_screenshot(
+    ui: widgets.CylindraMainWidget, name: str, canvas_only: bool = True
+):
     with mkdocs_gen_files.FilesEditor.current().open(f"images/{name}.png", "wb") as f:
-        ui.parent_viewer.screenshot(f.name, canvas_only=canvas_only)
+        ui.parent_viewer.screenshot(f.name, canvas_only=canvas_only, flash=False)
 
 
 def main():