109 lines
3.7 KiB
Python
109 lines
3.7 KiB
Python
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from dataclasses import dataclass
|
|||
|
|
from datetime import datetime
|
|||
|
|
from pathlib import Path
|
|||
|
|
from uuid import uuid4
|
|||
|
|
|
|||
|
|
from docx import Document
|
|||
|
|
|
|||
|
|
|
|||
|
|
@dataclass(frozen=True)
|
|||
|
|
class AnalysisReport:
|
|||
|
|
source_filename: str
|
|||
|
|
provider_name: str
|
|||
|
|
model_name: str
|
|||
|
|
matched_skills: list[str]
|
|||
|
|
summary: str
|
|||
|
|
findings: list[dict[str, str]]
|
|||
|
|
recommendations: list[str]
|
|||
|
|
raw_model_output: str
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _safe_stem(filename: str) -> str:
|
|||
|
|
stem = Path(filename).stem or "analysis"
|
|||
|
|
safe = "".join(ch if ch.isalnum() or ch in ("-", "_") else "_" for ch in stem)
|
|||
|
|
return safe[:60] or "analysis"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _report_base_path(report: AnalysisReport, output_dir: Path, suffix: str) -> Path:
|
|||
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|||
|
|
return output_dir / f"{_safe_stem(report.source_filename)}_{uuid4().hex[:8]}_analysis.{suffix}"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_markdown_report(report: AnalysisReport, output_dir: Path | str) -> Path:
|
|||
|
|
path = _report_base_path(report, Path(output_dir), "md")
|
|||
|
|
lines = [
|
|||
|
|
"# DOCX 规范分析报告",
|
|||
|
|
"",
|
|||
|
|
"## 基本信息",
|
|||
|
|
"",
|
|||
|
|
f"- 源文件：{report.source_filename}",
|
|||
|
|
f"- 分析时间：{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
|
|||
|
|
f"- 模型供应商：{report.provider_name}",
|
|||
|
|
f"- 模型名称：{report.model_name}",
|
|||
|
|
f"- 命中技能：{', '.join(report.matched_skills) or '无'}",
|
|||
|
|
"",
|
|||
|
|
"## 总体结论",
|
|||
|
|
"",
|
|||
|
|
report.summary,
|
|||
|
|
"",
|
|||
|
|
"## 技能符合性矩阵",
|
|||
|
|
"",
|
|||
|
|
"| 状态 | 检查项 | 证据或说明 |",
|
|||
|
|
"| --- | --- | --- |",
|
|||
|
|
]
|
|||
|
|
for finding in report.findings:
|
|||
|
|
lines.append(
|
|||
|
|
f"| {finding.get('status', '')} | {finding.get('item', '')} | {finding.get('evidence', '').replace('|', '/')} |"
|
|||
|
|
)
|
|||
|
|
lines.extend(["", "## 修改建议", ""])
|
|||
|
|
for item in report.recommendations:
|
|||
|
|
lines.append(f"- {item}")
|
|||
|
|
lines.extend(["", "## 模型分析原文", "", report.raw_model_output])
|
|||
|
|
path.write_text("\n".join(lines), encoding="utf-8")
|
|||
|
|
return path
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_docx_report(report: AnalysisReport, output_dir: Path | str) -> Path:
|
|||
|
|
path = _report_base_path(report, Path(output_dir), "docx")
|
|||
|
|
document = Document()
|
|||
|
|
document.add_heading("DOCX 规范分析报告", level=0)
|
|||
|
|
document.add_heading("基本信息", level=1)
|
|||
|
|
for label, value in [
|
|||
|
|
("源文件", report.source_filename),
|
|||
|
|
("分析时间", datetime.now().strftime("%Y-%m-%d %H:%M:%S")),
|
|||
|
|
("模型供应商", report.provider_name),
|
|||
|
|
("模型名称", report.model_name),
|
|||
|
|
("命中技能", ", ".join(report.matched_skills) or "无"),
|
|||
|
|
]:
|
|||
|
|
document.add_paragraph(f"{label}：{value}")
|
|||
|
|
|
|||
|
|
document.add_heading("总体结论", level=1)
|
|||
|
|
document.add_paragraph(report.summary)
|
|||
|
|
|
|||
|
|
document.add_heading("技能符合性矩阵", level=1)
|
|||
|
|
table = document.add_table(rows=1, cols=3)
|
|||
|
|
table.style = "Table Grid"
|
|||
|
|
headers = table.rows[0].cells
|
|||
|
|
headers[0].text = "状态"
|
|||
|
|
headers[1].text = "检查项"
|
|||
|
|
headers[2].text = "证据或说明"
|
|||
|
|
for finding in report.findings:
|
|||
|
|
row = table.add_row().cells
|
|||
|
|
row[0].text = finding.get("status", "")
|
|||
|
|
row[1].text = finding.get("item", "")
|
|||
|
|
row[2].text = finding.get("evidence", "")
|
|||
|
|
|
|||
|
|
document.add_heading("修改建议", level=1)
|
|||
|
|
for item in report.recommendations:
|
|||
|
|
document.add_paragraph(item, style="List Bullet")
|
|||
|
|
|
|||
|
|
document.add_heading("模型分析原文", level=1)
|
|||
|
|
for line in report.raw_model_output.splitlines() or ["无"]:
|
|||
|
|
document.add_paragraph(line)
|
|||
|
|
|
|||
|
|
document.add_paragraph("说明：模型分析结果需人工复核，不应直接作为正式审查结论。")
|
|||
|
|
document.save(path)
|
|||
|
|
return path
|