| """Django admin for document recognition.""" |
|
|
| from __future__ import annotations |
|
|
| from typing import Any, ClassVar |
|
|
| from django.contrib import admin |
| from django.db.models import QuerySet |
| from django.http import HttpRequest, HttpResponse |
| from django.shortcuts import render |
| from django.template.response import TemplateResponse |
| from django.urls import path, reverse |
| from django.utils.html import format_html |
| from django.utils.safestring import SafeString |
| from django.utils.translation import gettext_lazy as _ |
|
|
| from apps.document_recognition.models import DocumentRecognitionStatus, DocumentRecognitionTask, DocumentRecognitionTool |
|
|
|
|
| @admin.register(DocumentRecognitionTool) |
| class DocumentRecognitionToolAdmin(admin.ModelAdmin[DocumentRecognitionTool]): |
| """Admin entry page for the recognition workbench.""" |
|
|
| def changelist_view( |
| self, |
| request: HttpRequest, |
| extra_context: dict[str, Any] | None = None, |
| ) -> TemplateResponse: |
| context = { |
| **self.admin_site.each_context(request), |
| "title": "法院文书智能识别", |
| "opts": self.model._meta, |
| "has_view_permission": self.has_view_permission(request), |
| } |
| return TemplateResponse(request, "admin/document_recognition/recognition.html", context) |
|
|
| def has_add_permission(self, request: HttpRequest) -> bool: |
| return False |
|
|
| def has_change_permission(self, request: HttpRequest, obj: DocumentRecognitionTool | None = None) -> bool: |
| return False |
|
|
| def has_delete_permission(self, request: HttpRequest, obj: DocumentRecognitionTool | None = None) -> bool: |
| return False |
|
|
| def get_model_perms(self, request: HttpRequest) -> dict[str, bool]: |
| return {"view": True} |
|
|
|
|
| @admin.register(DocumentRecognitionTask) |
| class DocumentRecognitionTaskAdmin(admin.ModelAdmin[DocumentRecognitionTask]): |
| """Document recognition task list and detail admin.""" |
|
|
| list_display: ClassVar[list[str]] = [ |
| "id", |
| "status_display", |
| "original_filename", |
| "document_type_display", |
| "case_number", |
| "case_display", |
| "binding_status_display", |
| "notification_status_display", |
| "notification_sent_at", |
| "created_at", |
| ] |
| list_filter: ClassVar[list[str]] = [ |
| "status", |
| "document_type", |
| "binding_success", |
| "notification_sent", |
| "created_at", |
| ] |
| search_fields: ClassVar[list[str]] = ["original_filename", "case_number", "case__name"] |
| ordering: ClassVar[list[str]] = ["-created_at"] |
| list_per_page = 20 |
| readonly_fields: ClassVar[list[str]] = [ |
| "id", |
| "file_path", |
| "original_filename", |
| "status", |
| "document_type", |
| "case_number", |
| "key_time", |
| "confidence", |
| "extraction_method", |
| "raw_text_display", |
| "renamed_file_path", |
| "binding_success", |
| "case", |
| "case_log", |
| "binding_message", |
| "binding_error_code", |
| "error_message", |
| "notification_sent", |
| "notification_sent_at", |
| "notification_error", |
| "notification_file_sent", |
| "created_at", |
| "started_at", |
| "finished_at", |
| ] |
| fieldsets = ( |
| (_("基本信息"), {"fields": ("id", "original_filename", "file_path", "status")}), |
| ( |
| _("识别结果"), |
| { |
| "fields": ( |
| "document_type", |
| "case_number", |
| "key_time", |
| "confidence", |
| "extraction_method", |
| "renamed_file_path", |
| ) |
| }, |
| ), |
| (_("原始文本"), {"fields": ("raw_text_display",), "classes": ("collapse",)}), |
| (_("绑定结果"), {"fields": ("binding_success", "case", "case_log", "binding_message", "binding_error_code")}), |
| ( |
| _("通知状态"), |
| { |
| "fields": ("notification_sent", "notification_sent_at", "notification_file_sent", "notification_error"), |
| "description": "绑定成功后的飞书群通知状态", |
| }, |
| ), |
| (_("错误信息"), {"fields": ("error_message",), "classes": ("collapse",)}), |
| (_("时间戳"), {"fields": ("created_at", "started_at", "finished_at"), "classes": ("collapse",)}), |
| ) |
|
|
| def get_urls(self) -> list[Any]: |
| urls = super().get_urls() |
| custom_urls = [ |
| path( |
| "recognize/", |
| self.admin_site.admin_view(self.recognition_view), |
| name="document_recognition_task_recognize", |
| ), |
| ] |
| return custom_urls + urls |
|
|
| def recognition_view(self, request: HttpRequest) -> HttpResponse: |
| context = { |
| **self.admin_site.each_context(request), |
| "title": "法院文书智能识别", |
| "opts": self.model._meta, |
| "has_view_permission": True, |
| } |
| return render(request, "admin/document_recognition/recognition.html", context) |
|
|
| def status_display(self, obj: DocumentRecognitionTask) -> SafeString: |
| status_colors: dict[str, str] = { |
| DocumentRecognitionStatus.PENDING: "orange", |
| DocumentRecognitionStatus.PROCESSING: "blue", |
| DocumentRecognitionStatus.SUCCESS: "green", |
| DocumentRecognitionStatus.FAILED: "red", |
| } |
| color = status_colors.get(obj.status, "gray") |
| return format_html('<span style="color: {}; font-weight: bold;">{}</span>', color, obj.get_status_display()) |
|
|
| status_display.short_description = _("任务状态") |
| status_display.admin_order_field = "status" |
|
|
| def document_type_display(self, obj: DocumentRecognitionTask) -> str: |
| if not obj.document_type: |
| return "-" |
|
|
| type_icons: dict[str, str] = { |
| "summons": "📋", |
| "execution": "⚖️", |
| "execution_ruling": "⚖️", |
| "other": "📄", |
| } |
| icon = type_icons.get(obj.document_type, "📄") |
| return f"{icon} {obj.document_type}" |
|
|
| document_type_display.short_description = _("文书类型") |
| document_type_display.admin_order_field = "document_type" |
|
|
| def case_display(self, obj: DocumentRecognitionTask) -> SafeString | str: |
| if obj.case: |
| url = reverse("admin:cases_case_change", args=[obj.case.id]) |
| case_name = obj.case.name |
| if len(case_name) > 30: |
| case_name = case_name[:30] + "..." |
| return format_html('<a href="{}" target="_blank">{}</a>', url, case_name) |
| return "-" |
|
|
| case_display.short_description = _("关联案件") |
|
|
| def binding_status_display(self, obj: DocumentRecognitionTask) -> SafeString: |
| if obj.binding_success is None: |
| return format_html('<span style="color: gray;">{}</span>', "- 未绑定") |
| if obj.binding_success: |
| return format_html('<span style="color: green;">{}</span>', "✓ 绑定成功") |
| error_preview = obj.binding_error_code or "未知错误" |
| return format_html( |
| '<span style="color: red;">✗ 绑定失败</span><br><small style="color: #d63384;">{}</small>', |
| error_preview, |
| ) |
|
|
| binding_status_display.short_description = _("绑定状态") |
|
|
| def notification_status_display(self, obj: DocumentRecognitionTask) -> SafeString: |
| if not obj.binding_success: |
| return format_html('<span style="color: gray;">{}</span>', "- 无需通知") |
|
|
| if obj.notification_sent: |
| file_status = "✓ 文件已发送" if obj.notification_file_sent else "✗ 文件未发送" |
| return format_html( |
| '<span style="color: green;">✓ 通知成功</span><br><small style="color: #666;">{}</small>', |
| file_status, |
| ) |
|
|
| if obj.notification_error: |
| error_preview = obj.notification_error[:30] + ("..." if len(obj.notification_error) > 30 else "") |
| return format_html( |
| '<span style="color: red;">✗ 通知失败</span><br><small style="color: #d63384;">{}</small>', |
| error_preview, |
| ) |
|
|
| return format_html('<span style="color: orange;">{}</span>', "⏳ 待发送") |
|
|
| notification_status_display.short_description = _("通知状态") |
|
|
| def raw_text_display(self, obj: DocumentRecognitionTask) -> SafeString | str: |
| if obj.raw_text: |
| return format_html( |
| '<div style="max-height: 300px; overflow-y: auto; ' |
| "white-space: pre-wrap; font-family: monospace; " |
| 'background: #f5f5f5; padding: 10px; border-radius: 4px;">{}</div>', |
| obj.raw_text, |
| ) |
| return "-" |
|
|
| raw_text_display.short_description = _("原始文本") |
|
|
| def get_queryset(self, request: HttpRequest) -> QuerySet[DocumentRecognitionTask]: |
| return super().get_queryset(request).select_related("case", "case_log") |
|
|
| def has_add_permission(self, request: HttpRequest) -> bool: |
| return False |
|
|
| def has_change_permission(self, request: HttpRequest, obj: DocumentRecognitionTask | None = None) -> bool: |
| return False |
|
|
| def has_delete_permission(self, request: HttpRequest, obj: DocumentRecognitionTask | None = None) -> bool: |
| return True |
|
|
