| """Async tasks for document recognition.""" |
|
|
| from __future__ import annotations |
|
|
| import logging |
| from typing import Any |
|
|
| logger = logging.getLogger("apps.document_recognition") |
|
|
|
|
| def execute_document_recognition_task(task_id: int) -> dict[str, Any] | None: |
| """Execute one document recognition task in Django-Q worker.""" |
| from django.utils import timezone |
|
|
| from apps.core.interfaces import ServiceLocator |
| from apps.document_recognition.models import DocumentRecognitionStatus, DocumentRecognitionTask |
|
|
| logger.info("开始执行文书识别任务 #%s", task_id) |
|
|
| try: |
| task = DocumentRecognitionTask.objects.get(id=task_id) |
| except DocumentRecognitionTask.DoesNotExist: |
| logger.error("识别任务不存在: %s", task_id) |
| return None |
|
|
| task.status = DocumentRecognitionStatus.PROCESSING |
| task.started_at = timezone.now() |
| task.save(update_fields=["status", "started_at"]) |
|
|
| try: |
| service = ServiceLocator.get_court_document_recognition_service() |
| result = service.recognize_document(task.file_path, user=None) |
|
|
| recognition = result.recognition |
| task.document_type = recognition.document_type.value |
| task.case_number = recognition.case_number |
| task.key_time = recognition.key_time |
| task.confidence = recognition.confidence |
| task.extraction_method = recognition.extraction_method |
| task.raw_text = recognition.raw_text[:10000] if recognition.raw_text else None |
| task.renamed_file_path = result.file_path |
|
|
| if result.binding: |
| task.binding_success = result.binding.success |
| task.binding_message = result.binding.message |
| task.binding_error_code = result.binding.error_code |
| if result.binding.case_id: |
| task.case_id = result.binding.case_id |
| if result.binding.case_log_id: |
| task.case_log_id = result.binding.case_log_id |
|
|
| task.status = DocumentRecognitionStatus.SUCCESS |
| task.finished_at = timezone.now() |
| task.save() |
|
|
| if result.binding and result.binding.success: |
| _send_recognition_notification(task, result) |
|
|
| logger.info("文书识别任务 #%s 完成: %s", task_id, task.document_type) |
| return {"task_id": task_id, "status": "success", "document_type": task.document_type} |
|
|
| except Exception as exc: |
| logger.error("文书识别任务 #%s 失败: %s", task_id, exc, exc_info=True) |
|
|
| task.status = DocumentRecognitionStatus.FAILED |
| task.error_message = str(exc) |
| task.finished_at = timezone.now() |
| task.save(update_fields=["status", "error_message", "finished_at"]) |
|
|
| return {"task_id": task_id, "status": "failed", "error": str(exc)} |
|
|
|
|
| def _send_recognition_notification(task: Any, result: Any) -> None: |
| """Send recognition notification after successful binding.""" |
| try: |
| from apps.document_recognition.services.notification_service import DocumentRecognitionNotificationService |
|
|
| notification_service = DocumentRecognitionNotificationService() |
| file_path = task.renamed_file_path or task.file_path |
|
|
| notification_result = notification_service.send_notification( |
| case_id=result.binding.case_id, |
| document_type=task.document_type, |
| case_number=task.case_number, |
| key_time=task.key_time, |
| file_path=file_path, |
| case_name=result.binding.case_name, |
| ) |
|
|
| task.notification_sent = notification_result.success |
| task.notification_sent_at = notification_result.sent_at |
| task.notification_file_sent = notification_result.file_sent |
|
|
| if not notification_result.success: |
| task.notification_error = notification_result.message |
| logger.warning( |
| "文书识别通知发送失败: task_id=%s, error=%s", |
| task.id, |
| notification_result.message, |
| ) |
| else: |
| logger.info( |
| "文书识别通知发送成功: task_id=%s, file_sent=%s", |
| task.id, |
| notification_result.file_sent, |
| ) |
|
|
| task.save( |
| update_fields=[ |
| "notification_sent", |
| "notification_sent_at", |
| "notification_file_sent", |
| "notification_error", |
| ] |
| ) |
|
|
| except Exception as exc: |
| logger.error("发送文书识别通知异常: task_id=%s, error=%s", task.id, exc, exc_info=True) |
| task.notification_sent = False |
| task.notification_error = str(exc) |
| task.save(update_fields=["notification_sent", "notification_error"]) |
|
|
