diff options
Diffstat (limited to 'scripts/build-declarations.py')
| -rw-r--r-- | scripts/build-declarations.py | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/scripts/build-declarations.py b/scripts/build-declarations.py index 17a27b6..2a28e43 100644 --- a/scripts/build-declarations.py +++ b/scripts/build-declarations.py @@ -7,13 +7,14 @@ import re import sys from collections import OrderedDict +TOOL_ENTRY_FUNC = "run" -def main(): +def main(is_tool = True): scriptfile = sys.argv[1] with open(scriptfile, "r", encoding="utf-8") as f: contents = f.read() - functions = extract_functions(contents) + functions = extract_functions(contents, is_tool) declarations = [] for function in functions: func_name, docstring, func_args = function @@ -22,15 +23,16 @@ def main(): build_declaration(func_name, description, params, func_args) ) - name = os.path.splitext(os.path.basename(scriptfile))[0] - if declarations: - declarations = declarations[0:1] - declarations[0]["name"] = name + if is_tool: + name = os.path.splitext(os.path.basename(scriptfile))[0] + if declarations: + declarations = declarations[0:1] + declarations[0]["name"] = name print(json.dumps(declarations, indent=2)) -def extract_functions(contents: str): +def extract_functions(contents: str, is_tool: bool): tree = ast.parse(contents) output = [] for node in ast.walk(tree): @@ -39,6 +41,8 @@ def extract_functions(contents: str): func_name = node.name if func_name.startswith("_"): continue + if is_tool and func_name != TOOL_ENTRY_FUNC: + continue docstring = ast.get_docstring(node) or "" func_args = OrderedDict() for arg in node.args.args: |