INNER CODE UNIT · Python

main

CybercentreCanada/assemblyline · gen_docs.py:16

def main(root_dir, site_prefix=None):
    BASE_PATH = f"{root_dir}/assemblyline-base/assemblyline/"
    DOCS_BASE_PATH = f"{root_dir}/assemblyline4_docs/"
    DOCS_DOCS_PATH = os.path.join(DOCS_BASE_PATH, "docs/")

    def render_markdown(file) -> None:
        if not file.endswith(".py"):
            return
        lib = SourceFileLoader(file, file).load_module()
        markdown_list = [(c.markdown(), c.__name__) for c in lib.__dict__.values() if isclass(
            c) and issubclass(c, Model) and not (c.__module__.startswith("assemblyline.odm.models"))]

        if markdown_list:
            # Bring last item to the front
            name = markdown_list[-1][1]
            compiled_md = "\n\n".join(m[0] for m in markdown_list)
            if len(markdown_list) > 1:
                compiled_md = markdown_list.pop(-1)[0]

View source record →