feat(file-summary): 生成汇总报告和导出下载

review_agent/file_summary/services/export_excel.py

@@ -0,0 +1,54 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from openpyxl import Workbook
+
+from review_agent.models import ExportedSummaryFile, FileSummaryBatch
+
+
+def _exports_dir(batch: FileSummaryBatch) -> Path:
+    root = Path(batch.work_dir or Path("media") / "file_summary" / batch.batch_no)
+    export_dir = root / "exports"
+    export_dir.mkdir(parents=True, exist_ok=True)
+    return export_dir
+
+
+def generate_excel_export(batch: FileSummaryBatch) -> ExportedSummaryFile:
+    workbook = Workbook()
+    summary = workbook.active
+    summary.title = "汇总信息"
+    summary.append(["批次号", batch.batch_no])
+    summary.append(["产品名称", batch.product_name or "-"])
+    summary.append(["文件总数", batch.total_files])
+    summary.append(["统计成功", batch.success_files])
+    summary.append(["统计失败", batch.failed_files])
+    summary.append(["不支持", batch.unsupported_files])
+    summary.append(["不确定", batch.uncertain_files])
+    summary.append(["总页数", batch.total_pages])
+
+    detail = workbook.create_sheet("文件明细")
+    detail.append(["序号", "目录层级", "文件名", "类型", "页数", "路径", "状态", "重试次数", "异常说明"])
+    for item in batch.items.order_by("file_index"):
+        detail.append(
+            [
+                item.file_index,
+                item.directory_level,
+                item.file_name,
+                item.file_type,
+                item.page_count,
+                item.relative_path,
+                item.statistics_status,
+                item.retry_count,
+                item.error_message,
+            ]
+        )
+
+    path = _exports_dir(batch) / f"{batch.batch_no}-summary.xlsx"
+    workbook.save(path)
+    return ExportedSummaryFile.objects.create(
+        batch=batch,
+        export_type=ExportedSummaryFile.ExportType.EXCEL,
+        file_name=path.name,
+        storage_path=str(path),
+    )

review_agent/file_summary/services/report.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+from review_agent.models import ExportedSummaryFile, FileSummaryBatch
+
+
+def _exports_dir(batch: FileSummaryBatch) -> Path:
+    root = Path(batch.work_dir or Path("media") / "file_summary" / batch.batch_no)
+    export_dir = root / "exports"
+    export_dir.mkdir(parents=True, exist_ok=True)
+    return export_dir
+
+
+def build_summary_table(batch: FileSummaryBatch) -> str:
+    lines = [
+        "| 序号 | 目录层级 | 文件名 | 类型 | 页数 | 状态 | 异常说明 |",
+        "| --- | --- | --- | --- | --- | --- | --- |",
+    ]
+    for item in batch.items.order_by("file_index"):
+        lines.append(
+            "| {index} | {directory} | {name} | {file_type} | {pages} | {status} | {error} |".format(
+                index=item.file_index,
+                directory=item.directory_level or "-",
+                name=item.file_name,
+                file_type=item.file_type,
+                pages=item.page_count if item.page_count is not None else "-",
+                status=item.statistics_status,
+                error=item.error_message or "-",
+            )
+        )
+    return "\n".join(lines)
+
+
+def build_markdown_report(batch: FileSummaryBatch) -> str:
+    return "\n\n".join(
+        [
+            f"# 文件目录与页数汇总报告\n\n批次号：{batch.batch_no}",
+            (
+                "## 汇总信息\n\n"
+                f"- 产品名称：{batch.product_name or '-'}\n"
+                f"- 文件总数：{batch.total_files}\n"
+                f"- 统计成功：{batch.success_files}\n"
+                f"- 统计失败：{batch.failed_files}\n"
+                f"- 不支持：{batch.unsupported_files}\n"
+                f"- 不确定：{batch.uncertain_files}\n"
+                f"- 总页数：{batch.total_pages}"
+            ),
+            "## 文件明细\n\n" + build_summary_table(batch),
+            "## 处理说明\n\n单文件失败不会阻断批次，失败与不确定文件已在明细中标注。",
+        ]
+    )
+
+
+def generate_markdown_report(batch: FileSummaryBatch) -> tuple[ExportedSummaryFile, str]:
+    content = build_markdown_report(batch)
+    path = _exports_dir(batch) / f"{batch.batch_no}-summary.md"
+    path.write_text(content, encoding="utf-8")
+    exported = ExportedSummaryFile.objects.create(
+        batch=batch,
+        export_type=ExportedSummaryFile.ExportType.MARKDOWN,
+        file_name=path.name,
+        storage_path=str(path),
+    )
+    return exported, build_summary_table(batch)

review_agent/file_summary/skills/summary_report.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+from django.urls import reverse
+
+from review_agent.models import Message
+
+from ..services.export_excel import generate_excel_export
+from ..services.report import generate_markdown_report
+from .base import BaseSkill, SkillResult, WorkflowContext
+
+
+class SummaryReportSkill(BaseSkill):
+    name = "summary_report"
+
+    def run(self, context: WorkflowContext) -> SkillResult:
+        markdown_export, summary_table = generate_markdown_report(context.batch)
+        excel_export = generate_excel_export(context.batch)
+        markdown_url = reverse("file_summary_export_download", args=[markdown_export.pk])
+        excel_url = reverse("file_summary_export_download", args=[excel_export.pk])
+        content = (
+            "文件目录与页数汇总已完成。\n\n"
+            f"{summary_table}\n\n"
+            f"[下载 Markdown 报告]({markdown_url}) | [下载 Excel 明细]({excel_url})"
+        )
+        Message.objects.create(
+            conversation=context.batch.conversation,
+            role=Message.Role.ASSISTANT,
+            content=content,
+        )
+        return SkillResult(
+            success=True,
+            data={"markdown_export_id": markdown_export.pk, "excel_export_id": excel_export.pk},
+        )

review_agent/file_summary/views.py

@@ -1,8 +1,10 @@
 from django.contrib.auth.decorators import login_required
-from django.http import Http404, JsonResponse
+from pathlib import Path
+
+from django.http import FileResponse, Http404, JsonResponse
 from django.views.decorators.http import require_http_methods
 
-from review_agent.models import Conversation, FileAttachment
+from review_agent.models import Conversation, ExportedSummaryFile, FileAttachment
 from review_agent.models import FileSummaryBatch, WorkflowEvent
 from .events import serialize_event
 
@@ -105,3 +107,18 @@ def batch_events(request, batch_id: int):
         after_id = 0
     events = WorkflowEvent.objects.filter(batch=batch, pk__gt=after_id).order_by("id")
     return JsonResponse({"events": [serialize_event(event) for event in events]})
+
+
+@require_http_methods(["GET"])
+@login_required
+def export_download(request, export_id: int):
+    exported = ExportedSummaryFile.objects.filter(
+        pk=export_id,
+        batch__user=request.user,
+    ).first()
+    if not exported:
+        raise Http404("导出文件不存在。")
+    path = Path(exported.storage_path)
+    if not path.exists():
+        return JsonResponse({"error": "文件不存在。"}, status=404)
+    return FileResponse(path.open("rb"), as_attachment=True, filename=exported.file_name)

review_agent/file_summary/workflow.py

@@ -22,6 +22,7 @@ from .skills.document_page_count import DocumentPageCountSkill
 from .skills.file_inventory import FileInventorySkill
 from .skills.product_detect import ProductDetectSkill
 from .skills.registry import SkillRegistry
+from .skills.summary_report import SummaryReportSkill
 
 
 NODE_DEFINITIONS = [
@@ -30,7 +31,7 @@ NODE_DEFINITIONS = [
     ("inventory", "文件扫描", "file_inventory"),
     ("page_count", "页数统计", "document_page_count"),
     ("product_detect", "产品识别", "product_detect"),
-    ("report", "报告输出", ""),
+    ("report", "报告输出", "summary_report"),
     ("complete", "完成", ""),
 ]
 
@@ -41,6 +42,7 @@ def default_skill_registry() -> SkillRegistry:
     registry.register(FileInventorySkill())
     registry.register(DocumentPageCountSkill())
     registry.register(ProductDetectSkill())
+    registry.register(SummaryReportSkill())
     return registry
 
 

review_agent/urls.py

@@ -1,6 +1,6 @@
 from django.urls import path
 
-from .file_summary.views import attachment_detail, attachments, batch_events, batch_status
+from .file_summary.views import attachment_detail, attachments, batch_events, batch_status, export_download
 
 
 urlpatterns = [
@@ -29,4 +29,9 @@ urlpatterns = [
         batch_events,
         name="file_summary_batch_events",
     ),
+    path(
+        "api/review-agent/file-summary/exports/<int:export_id>/download/",
+        export_download,
+        name="file_summary_export_download",
+    ),
 ]