75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
#!/usr/bin/env python3
|
||
import argparse
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
|
||
def load_template(template_path: Path) -> str:
|
||
return template_path.read_text(encoding="utf-8")
|
||
|
||
|
||
def render_template(template: str, values: dict) -> str:
|
||
rendered = template
|
||
for key, value in values.items():
|
||
rendered = rendered.replace(f"{{{{{key}}}}}", value)
|
||
return rendered
|
||
|
||
|
||
def build_arg_parser() -> argparse.ArgumentParser:
|
||
parser = argparse.ArgumentParser(
|
||
description="生成 SOP 模板（全量/最小版），输出到文件或标准输出"
|
||
)
|
||
parser.add_argument("--title", required=True, help="SOP 标题")
|
||
parser.add_argument("--doc-id", default="SOP-000", help="文档编号")
|
||
parser.add_argument("--version", default="v1.0", help="版本号")
|
||
parser.add_argument("--date", default=None, help="发布日期（默认今天）")
|
||
parser.add_argument("--owner", default="未指定", help="编制人")
|
||
parser.add_argument("--reviewer", default="未指定", help="审核人")
|
||
parser.add_argument("--approver", default="未指定", help="批准人")
|
||
parser.add_argument(
|
||
"--mvp",
|
||
action="store_true",
|
||
help="使用最小可行 SOP 模板（MVP）",
|
||
)
|
||
parser.add_argument(
|
||
"--output",
|
||
default="-",
|
||
help="输出路径，使用 '-' 表示标准输出",
|
||
)
|
||
return parser
|
||
|
||
|
||
def main() -> None:
|
||
args = build_arg_parser().parse_args()
|
||
script_dir = Path(__file__).resolve().parent
|
||
assets_dir = script_dir.parent / "assets"
|
||
template_file = assets_dir / ("sop-mvp.md" if args.mvp else "sop-template.md")
|
||
|
||
if not template_file.exists():
|
||
raise FileNotFoundError(f"找不到模板文件: {template_file}")
|
||
|
||
date_value = args.date or datetime.now().strftime("%Y-%m-%d")
|
||
values = {
|
||
"title": args.title,
|
||
"doc_id": args.doc_id,
|
||
"version": args.version,
|
||
"date": date_value,
|
||
"owner": args.owner,
|
||
"reviewer": args.reviewer,
|
||
"approver": args.approver,
|
||
}
|
||
|
||
template = load_template(template_file)
|
||
rendered = render_template(template, values)
|
||
|
||
if args.output == "-":
|
||
print(rendered)
|
||
return
|
||
|
||
output_path = Path(args.output)
|
||
output_path.write_text(rendered, encoding="utf-8")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|