File size: 4,691 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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 | """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"])
|