File size: 1,617 Bytes
b4f2238 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | """
文档处理工具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)
|