77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
import inspect
|
|
from typing import Any
|
|
from typing import Dict
|
|
from typing import Optional
|
|
from typing import Type
|
|
|
|
from tools.docgen.utils import camel_case_to_human
|
|
from tools.docgen.utils import section
|
|
from tools.docgen.utils import static_methods
|
|
from tools.docgen.utils import to_out_dir
|
|
from ytdl_sub.entries.script.custom_functions import CustomFunctions
|
|
from ytdl_sub.script.functions import Functions
|
|
from ytdl_sub.script.utils.type_checking import FunctionSpec
|
|
|
|
|
|
def maybe_get_function_name(function_name: str) -> Optional[str]:
|
|
if function_name in ["register"]:
|
|
return None
|
|
|
|
if function_name.endswith("_"):
|
|
return function_name[:-1]
|
|
return function_name
|
|
|
|
|
|
def function_class_to_name(obj: Type[Any]) -> str:
|
|
assert "Functions" in obj.__name__
|
|
return camel_case_to_human(obj.__name__)
|
|
|
|
|
|
def function_type_hinting(display_function_name: str, function: Any) -> str:
|
|
spec = FunctionSpec.from_callable(function)
|
|
out = "``"
|
|
out += display_function_name
|
|
out += spec.human_readable_input_args()
|
|
out += " -> "
|
|
out += spec.human_readable_output_type()
|
|
out += "``\n\n"
|
|
return out
|
|
|
|
|
|
def get_function_docstring(
|
|
function_name: str, function: Any, level: int, display_function_name: Optional[str] = None
|
|
) -> str:
|
|
display_function_name = display_function_name if display_function_name else function_name
|
|
|
|
docs = section(display_function_name, level=level)
|
|
|
|
docs += function_type_hinting(display_function_name=display_function_name, function=function)
|
|
docs += inspect.cleandoc(function.__doc__)
|
|
docs += "\n"
|
|
return docs
|
|
|
|
|
|
def generate_function_docs() -> str:
|
|
docs = section("Scripting Functions", level=0)
|
|
|
|
parent_objs: Dict[str, Type[Any]] = {
|
|
function_class_to_name(obj): obj for obj in Functions.__bases__
|
|
}
|
|
parent_objs["Ytdl-Sub Functions"] = CustomFunctions
|
|
|
|
for name in sorted(parent_objs.keys()):
|
|
docs += section(name, level=1)
|
|
|
|
for function_name in static_methods(parent_objs[name]):
|
|
if display_function_name := maybe_get_function_name(function_name):
|
|
docs += get_function_docstring(
|
|
function_name=function_name,
|
|
display_function_name=display_function_name,
|
|
function=getattr(parent_objs[name], function_name),
|
|
level=2,
|
|
)
|
|
|
|
return docs
|
|
|
|
|
|
to_out_dir(name="scripting_functions.rst", docs=generate_function_docs())
|