| """Models for document recognition.""" |
|
|
| from __future__ import annotations |
|
|
| from typing import ClassVar |
|
|
| from django.db import models |
| from django.utils.translation import gettext_lazy as _ |
|
|
|
|
| class DocumentRecognitionStatus(models.TextChoices): |
| """文书识别任务状态。""" |
|
|
| PENDING = "pending", _("待处理") |
| PROCESSING = "processing", _("识别中") |
| SUCCESS = "success", _("成功") |
| FAILED = "failed", _("失败") |
|
|
|
|
| class DocumentRecognitionTool(models.Model): |
| """Admin entry model for document recognition.""" |
|
|
| id: int |
| name = models.CharField(max_length=64, default="Document Recognition") |
|
|
| class Meta: |
| managed = False |
| verbose_name = _("文书智能识别") |
| verbose_name_plural = _("文书智能识别") |
|
|
|
|
| class DocumentRecognitionTask(models.Model): |
| """文书识别任务。""" |
|
|
| id: int |
| file_path = models.CharField(max_length=1024, verbose_name=_("文件路径")) |
| original_filename = models.CharField(max_length=256, verbose_name=_("原始文件名")) |
| status = models.CharField( |
| max_length=32, |
| choices=DocumentRecognitionStatus.choices, |
| default=DocumentRecognitionStatus.PENDING, |
| verbose_name=_("任务状态"), |
| ) |
| document_type = models.CharField(max_length=32, null=True, blank=True, verbose_name=_("文书类型")) |
| case_number = models.CharField(max_length=128, null=True, blank=True, verbose_name=_("案号")) |
| key_time = models.DateTimeField(null=True, blank=True, verbose_name=_("关键时间")) |
| confidence = models.FloatField(null=True, blank=True, verbose_name=_("置信度")) |
| extraction_method = models.CharField(max_length=32, null=True, blank=True, verbose_name=_("提取方式")) |
| raw_text = models.TextField(null=True, blank=True, verbose_name=_("原始文本")) |
| renamed_file_path = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_("重命名后路径")) |
| binding_success = models.BooleanField(null=True, verbose_name=_("绑定成功")) |
| case = models.ForeignKey( |
| "cases.Case", |
| on_delete=models.SET_NULL, |
| null=True, |
| blank=True, |
| related_name="recognition_tasks", |
| verbose_name=_("关联案件"), |
| ) |
| case_log = models.ForeignKey( |
| "cases.CaseLog", |
| on_delete=models.SET_NULL, |
| null=True, |
| blank=True, |
| related_name="recognition_tasks", |
| verbose_name=_("案件日志"), |
| ) |
| binding_message = models.CharField(max_length=512, null=True, blank=True, verbose_name=_("绑定消息")) |
| binding_error_code = models.CharField(max_length=64, null=True, blank=True, verbose_name=_("绑定错误码")) |
| error_message = models.TextField(null=True, blank=True, verbose_name=_("错误信息")) |
| notification_sent = models.BooleanField(default=False, verbose_name=_("通知已发送")) |
| notification_sent_at = models.DateTimeField(null=True, blank=True, verbose_name=_("通知发送时间")) |
| notification_error = models.TextField(null=True, blank=True, verbose_name=_("通知错误信息")) |
| notification_file_sent = models.BooleanField(default=False, verbose_name=_("文件已发送")) |
| created_at = models.DateTimeField(auto_now_add=True, verbose_name=_("创建时间")) |
| started_at = models.DateTimeField(null=True, blank=True, verbose_name=_("开始时间")) |
| finished_at = models.DateTimeField(null=True, blank=True, verbose_name=_("完成时间")) |
|
|
| class Meta: |
| managed = False |
| db_table = "automation_documentrecognitiontask" |
| verbose_name = _("文书识别任务") |
| verbose_name_plural = _("文书识别任务") |
| ordering: ClassVar[list[str]] = ["-created_at"] |
|
|
| def __str__(self) -> str: |
| return f"识别任务 #{self.id} - {self.get_status_display()}" |
|
|
