| """ |
| 文档生成 API |
| |
| 提供合同文件夹和补充协议的下载接口. |
| """ |
|
|
| from __future__ import annotations |
|
|
| import logging |
| from typing import Any |
|
|
| from django.utils.translation import gettext_lazy as _ |
| from ninja import Router |
|
|
| from apps.core.exceptions import ValidationException |
| from apps.core.infrastructure.throttling import rate_limit_from_settings |
| from apps.core.security.auth import JWTOrSessionAuth |
|
|
| from .download_response_factory import build_download_response |
|
|
| logger = logging.getLogger("apps.documents.api") |
| router = Router(auth=JWTOrSessionAuth()) |
|
|
|
|
| def _require_contract_access(request: Any, contract_id: int) -> None: |
| from apps.core.security.access_context import get_request_access_context |
| from apps.documents.services.infrastructure.wiring import get_contract_service |
|
|
| ctx = get_request_access_context(request) |
| get_contract_service().ensure_contract_access_ctx(contract_id=contract_id, ctx=ctx) |
|
|
|
|
| def _get_folder_generation_service() -> Any: |
| """工厂函数:获取 FolderGenerationService 实例""" |
| from apps.documents.services.generation.folder_generation_service import FolderGenerationService |
| from apps.documents.services.infrastructure.wiring import get_contract_folder_binding_service, get_contract_service |
|
|
| return FolderGenerationService( |
| contract_service=get_contract_service(), |
| folder_binding_service=get_contract_folder_binding_service(), |
| ) |
|
|
|
|
| def _get_supplementary_agreement_service() -> Any: |
| """工厂函数:获取 SupplementaryAgreementGenerationService 实例""" |
| from apps.documents.services.infrastructure.wiring import get_supplementary_agreement_generation_service |
|
|
| return get_supplementary_agreement_generation_service() |
|
|
|
|
| def _get_contract_generation_service() -> Any: |
| """工厂函数:获取 ContractGenerationService 实例""" |
| from apps.documents.services.infrastructure.wiring import get_contract_generation_service |
|
|
| return get_contract_generation_service() |
|
|
|
|
| @router.get("/contracts/{contract_id}/preview") |
| def preview_contract_context(request: Any, contract_id: int) -> Any: |
| """合同占位符预览""" |
| _require_contract_access(request, contract_id) |
| service = _get_contract_generation_service() |
| rows = service.get_preview_context(contract_id) |
| return {"success": True, "data": rows} |
|
|
|
|
| @router.get("/contracts/{contract_id}/supplementary-agreements/{agreement_id}/preview") |
| def preview_supplementary_agreement_context(request: Any, contract_id: int, agreement_id: int) -> Any: |
| """补充协议占位符预览""" |
| _require_contract_access(request, contract_id) |
| service = _get_supplementary_agreement_service() |
| rows = service.get_preview_context(contract_id, agreement_id) |
| return {"success": True, "data": rows} |
|
|
|
|
| @router.get("/contracts/{contract_id}/download") |
| @rate_limit_from_settings("EXPORT", by_user=True) |
| def download_contract_document(request: Any, contract_id: int, split_fee: bool = True) -> Any: |
| """ |
| 下载合同文档(DOCX 格式) |
| |
| 生成单个合同 Word 文档. |
| 如果合同有文件夹绑定,文件会保存到绑定文件夹,返回 JSON 响应. |
| 如果没有绑定,返回文件下载响应. |
| |
| Args: |
| contract_id: 合同 ID |
| |
| Returns: |
| DOCX 文件下载响应或 JSON 响应 |
| """ |
| _require_contract_access(request, contract_id) |
| service = _get_contract_generation_service() |
|
|
| |
| content, filename, saved_path, error = service.generate_contract_document_result(contract_id, split_fee=split_fee) |
|
|
| if error: |
| logger.warning("生成合同文档失败: %s", error, extra={"contract_id": contract_id, "error": error}) |
| raise ValidationException( |
| message=_("生成合同文档失败"), code="CONTRACT_GENERATION_FAILED", errors={"detail": error} |
| ) |
|
|
| if saved_path: |
| logger.info( |
| "合同文档已保存到绑定文件夹", |
| extra={ |
| "contract_id": contract_id, |
| "doc_filename": filename, |
| "saved_path": saved_path, |
| }, |
| ) |
| return { |
| "success": True, |
| "message": f"合同文档已生成并保存到: {saved_path}", |
| "filename": filename, |
| "folder_path": saved_path, |
| } |
|
|
| response = build_download_response( |
| content=content, |
| filename=filename, |
| content_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document", |
| ) |
|
|
| logger.info("合同文档下载成功", extra={"contract_id": contract_id, "doc_filename": filename}) |
|
|
| return response |
|
|
|
|
| @router.get("/contracts/{contract_id}/folder/download") |
| @rate_limit_from_settings("EXPORT", by_user=True) |
| def download_contract_folder(request: Any, contract_id: int) -> Any: |
| """ |
| 下载合同文件夹(ZIP 格式) |
| |
| 生成包含完整文件夹结构和合同文书的 ZIP 压缩包. |
| |
| Args: |
| contract_id: 合同 ID |
| |
| Returns: |
| ZIP 文件下载响应 |
| """ |
| _require_contract_access(request, contract_id) |
| service = _get_folder_generation_service() |
|
|
| |
| zip_content, zip_filename, extract_path, error = service.generate_folder_with_documents_result(contract_id) |
|
|
| if error: |
| logger.warning("生成合同文件夹失败: %s", error, extra={"contract_id": contract_id, "error": error}) |
| raise ValidationException( |
| message=_("生成合同文件夹失败"), code="FOLDER_GENERATION_FAILED", errors={"detail": error} |
| ) |
|
|
| if extract_path: |
| logger.info( |
| "合同文件夹已解压到绑定文件夹", |
| extra={"contract_id": contract_id, "zip_filename": zip_filename, "extract_path": extract_path}, |
| ) |
| return { |
| "success": True, |
| "message": f"合同文件夹已生成并解压到: {extract_path}", |
| "filename": zip_filename, |
| "folder_path": extract_path, |
| } |
|
|
| response = build_download_response(content=zip_content, filename=zip_filename, content_type="application/zip") |
|
|
| logger.info("合同文件夹下载成功", extra={"contract_id": contract_id, "zip_filename": zip_filename}) |
|
|
| return response |
|
|
|
|
| @router.get("/contracts/{contract_id}/supplementary-agreements/{agreement_id}/download") |
| @rate_limit_from_settings("EXPORT", by_user=True) |
| def download_supplementary_agreement(request: Any, contract_id: int, agreement_id: int) -> Any: |
| """ |
| 下载补充协议文档(DOCX 格式) |
| |
| 生成补充协议 Word 文档. |
| 如果合同有文件夹绑定,文件会保存到绑定文件夹,返回 JSON 响应. |
| 如果没有绑定,返回文件下载响应. |
| |
| Args: |
| contract_id: 合同 ID |
| agreement_id: 补充协议 ID |
| |
| Returns: |
| DOCX 文件下载响应或 JSON 响应 |
| """ |
| _require_contract_access(request, contract_id) |
| service = _get_supplementary_agreement_service() |
|
|
| |
| content, filename, saved_path, error = service.generate_supplementary_agreement_result(contract_id, agreement_id) |
|
|
| if error: |
| logger.warning( |
| "生成补充协议失败: %s", |
| error, |
| extra={"contract_id": contract_id, "agreement_id": agreement_id, "error": error}, |
| ) |
| raise ValidationException( |
| message=_("生成补充协议失败"), code="AGREEMENT_GENERATION_FAILED", errors={"detail": error} |
| ) |
|
|
| if saved_path: |
| logger.info( |
| "补充协议已保存到绑定文件夹", |
| extra={ |
| "contract_id": contract_id, |
| "agreement_id": agreement_id, |
| "doc_filename": filename, |
| "saved_path": saved_path, |
| }, |
| ) |
| return { |
| "success": True, |
| "message": f"补充协议已生成并保存到: {saved_path}", |
| "filename": filename, |
| "folder_path": saved_path, |
| } |
|
|
| response = build_download_response( |
| content=content, |
| filename=filename, |
| content_type="application/vnd.openxmlformats-officedocument.wordprocessingml.document", |
| ) |
|
|
| logger.info( |
| "补充协议下载成功", extra={"contract_id": contract_id, "agreement_id": agreement_id, "doc_filename": filename} |
| ) |
|
|
| return response |
|
|
