66import griffe
77import griffe2md .rendering
88import mdformat
9- import rendering_override
9+ import _rendering
1010from griffe import GriffeLoader , Object , Parser , Kind
1111from griffe2md import rendering
1212from griffe2md .main import prepare_context , prepare_env
13- from jinja2 import Environment , FileSystemLoader
13+ from jinja2 import FileSystemLoader
1414from extension import StubOverloadExtension
1515
16- sys .path .append (r"D:\Projects\CPP\endstone" )
17-
1816
1917def render_object_docs (obj : Object , config : dict | None = None ) -> str :
2018 env = prepare_env ()
2119 assert isinstance (env .loader , FileSystemLoader )
2220 loader : FileSystemLoader = env .loader
23- loader .searchpath .insert (0 , str (Path (__file__ ).parent / "templates" ))
24- env .filters ["heading" ] = rendering_override .do_heading
25- env .filters ["as_functions_section" ] = rendering_override .do_as_functions_section
21+ loader .searchpath .insert (0 , str (Path (__file__ ).parent / "templates" / "griffe2md" ))
22+ env .filters ["heading" ] = _rendering .do_heading
23+ env .filters ["as_functions_section" ] = _rendering .do_as_functions_section
2624 context = prepare_context (obj , config )
2725 rendered = env .get_template (f"{ obj .kind .value } .md.jinja" ).render (** context )
2826 return mdformat .text (rendered )
2927
3028
3129def render_package_docs (package : str , config : dict | None = None ) -> str :
32- config = config or dict (rendering .default_config )
30+ config = config or dict (_rendering .default_config )
3331 parser = config ["docstring_style" ] and Parser (config ["docstring_style" ])
3432 extensions = griffe .load_extensions (StubOverloadExtension )
3533 loader = GriffeLoader (docstring_parser = parser , extensions = extensions )
@@ -38,7 +36,8 @@ def render_package_docs(package: str, config: dict | None = None) -> str:
3836 return render_object_docs (module , config ) # type: ignore[arg-type]
3937
4038
41- def main ():
39+ def generate (base_dir : Path ):
40+ sys .path .append (str (base_dir .absolute ()))
4241 output_path = (
4342 Path (__file__ ).parent .parent .parent / "content" / "reference" / "python"
4443 )
@@ -81,7 +80,3 @@ def main():
8180 f .write (f'title: "endstone.{ submodule_name } "\n ' ) # Dynamic title
8281 f .write ('---\n \n ' )
8382 f .write (render_package_docs (f"endstone.{ submodule_name } " , config ))
84-
85-
86- if __name__ == "__main__" :
87- main ()
0 commit comments