| """ |
| 文档处理工具API |
| 独立的API模块 |
| """ |
|
|
| from ninja import Router |
|
|
| from apps.automation.schemas import DocumentProcessIn, DocumentProcessOut |
| from apps.core.infrastructure.throttling import rate_limit_from_settings |
|
|
| router = Router(tags=["Document Processor"]) |
|
|
|
|
| from typing import Any |
|
|
|
|
| def _get_document_processor_service() -> Any: |
| from apps.core.dependencies import build_document_processing_service |
|
|
| return build_document_processing_service() |
|
|
|
|
| @router.post("/process", response=DocumentProcessOut) |
| @rate_limit_from_settings("UPLOAD") |
| def process_document(request: Any, payload: DocumentProcessIn) -> DocumentProcessOut: |
| """文档处理API""" |
| |
| service = _get_document_processor_service() |
|
|
| |
| result = service.process_document( |
| file_path=payload.file_path, kind=payload.kind, limit=payload.limit, preview_page=payload.preview_page |
| ) |
|
|
| return DocumentProcessOut(image_url=result.image_url, text_excerpt=result.text_excerpt) |
|
|
|
|
| @router.post("/process-by-path", response=DocumentProcessOut) |
| @rate_limit_from_settings("UPLOAD") |
| def process_document_by_path(request: Any, payload: DocumentProcessIn) -> DocumentProcessOut: |
| """通过路径处理文档""" |
| |
| service = _get_document_processor_service() |
|
|
| |
| result = service.process_document_by_path( |
| file_path=payload.file_path, limit=payload.limit, preview_page=payload.preview_page |
| ) |
|
|
| return DocumentProcessOut(image_url=result.image_url, text_excerpt=result.text_excerpt) |
|
|
