您最多选择25个主题
主题必须以中文或者字母或数字开头，可以包含连字符 (-)，并且长度不得超过35个字符
75 行
2.4 KiB
75 行
2.4 KiB
#!/usr/bin/env python3
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import yaml
|
|
import argparse
|
|
import hashlib
|
|
|
|
|
|
# pydoc-markdown -I . -m module_name --render_toc > doc.md
|
|
|
|
|
|
def hash_file(filename):
|
|
if os.path.exists(filename):
|
|
hasher = hashlib.md5()
|
|
with open(filename, "rb") as file_to_hash:
|
|
buffer = file_to_hash.read()
|
|
hasher.update(buffer)
|
|
return hasher.hexdigest()
|
|
else:
|
|
return 0
|
|
|
|
|
|
def remove_trailing_whitespace(filename):
|
|
nchanged = 0
|
|
with open(filename, "rb") as f:
|
|
code1 = f.read().decode()
|
|
lines = [line.rstrip() for line in code1.splitlines()]
|
|
while lines and not lines[-1]:
|
|
lines.pop(-1)
|
|
lines.append("") # always end with a newline
|
|
code2 = "\n".join(lines)
|
|
if code1 != code2:
|
|
nchanged += 1
|
|
with open(filename, "wb") as f:
|
|
f.write(code2.encode())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--package_dirs", nargs="+")
|
|
args = parser.parse_args()
|
|
|
|
ok = False
|
|
return_code = 0
|
|
for package_dir in args.package_dirs:
|
|
config_path = os.path.join(os.getcwd(), package_dir, "pydoc-config.yaml")
|
|
print(config_path)
|
|
with open(config_path) as config_file:
|
|
config = yaml.full_load(config_file)
|
|
for module in config["modules"]:
|
|
module_name = module["name"]
|
|
submodules = module["submodules"]
|
|
output_file_name = f"./{config['folder']}/{module['file_name']}"
|
|
old_hash = hash_file(output_file_name)
|
|
module_args = []
|
|
for submodule in submodules:
|
|
module_args.append("-m")
|
|
module_args.append(f"{module_name}.{submodule}")
|
|
with open(output_file_name, "w") as output_file:
|
|
subprocess_args = [
|
|
"pydoc-markdown",
|
|
"-I",
|
|
f"./{package_dir}",
|
|
*module_args,
|
|
"--render-toc",
|
|
]
|
|
subprocess.check_call(subprocess_args, stdout=output_file)
|
|
remove_trailing_whitespace(output_file_name)
|
|
new_hash = hash_file(output_file_name)
|
|
ok = old_hash == new_hash
|
|
|
|
return_code = 0 if ok else 1
|
|
|
|
sys.exit(return_code)
|