| """Preservation date extraction API.""" |
|
|
| import logging |
| from datetime import datetime |
| from typing import Any |
|
|
| from django.http import HttpRequest |
| from ninja import File, Router |
| from ninja.files import UploadedFile |
|
|
| from apps.core.infrastructure.throttling import rate_limit_from_settings |
| from apps.preservation_date.services import PreservationDateExtractionService |
| from apps.preservation_date.services.models import PreservationExtractionResult |
|
|
| logger = logging.getLogger("apps.preservation_date") |
|
|
| router = Router(tags=["财产保全日期识别"]) |
|
|
|
|
| def _format_date(value: datetime | None) -> str | None: |
| if value is None: |
| return None |
| return value.strftime("%Y-%m-%d") |
|
|
|
|
| def _serialize_result(result: PreservationExtractionResult) -> dict[str, Any]: |
| return { |
| "success": result.success, |
| "measures": [ |
| { |
| "measure_type": item.measure_type, |
| "property_description": item.property_description, |
| "duration": item.duration, |
| "start_date": _format_date(item.start_date), |
| "end_date": _format_date(item.end_date), |
| "is_pending": item.is_pending, |
| "pending_note": item.pending_note, |
| "raw_text": item.raw_text, |
| } |
| for item in result.measures |
| ], |
| "reminders": [ |
| { |
| "reminder_type": item.reminder_type, |
| "content": item.content, |
| "due_at": _format_date(item.due_at), |
| "metadata": item.metadata, |
| } |
| for item in result.reminders |
| ], |
| "model_used": result.model_used, |
| "extraction_method": result.extraction_method, |
| "error": result.error, |
| "raw_response": result.raw_response, |
| } |
|
|
|
|
| @router.post("/extract") |
| @rate_limit_from_settings("UPLOAD", by_user=True) |
| def extract_preservation_date( |
| request: HttpRequest, |
| file: UploadedFile = File(...), |
| ) -> dict[str, Any]: |
| """Extract asset preservation measures and expiration dates from a PDF.""" |
| service = PreservationDateExtractionService() |
| result = service.extract_from_uploaded_file(file.chunks(), file.name) |
| return _serialize_result(result) |
|
|
