| from __future__ import annotations |
|
|
| from typing import Any |
|
|
| from django.contrib import admin |
| from django.http import HttpRequest |
| from django.template.response import TemplateResponse |
| from django.utils.html import format_html |
| from django.utils.safestring import SafeString |
| from django.utils.translation import gettext_lazy as _ |
|
|
| from apps.pdf_splitting.models import PdfSplitJob, PdfSplittingTool |
|
|
|
|
| @admin.register(PdfSplittingTool) |
| class PdfSplittingToolAdmin(admin.ModelAdmin[PdfSplittingTool]): |
| def changelist_view( |
| self, |
| request: HttpRequest, |
| extra_context: dict[str, Any] | None = None, |
| ) -> TemplateResponse: |
| context = { |
| **self.admin_site.each_context(request), |
| "title": "PDF 拆解", |
| "opts": self.model._meta, |
| "has_view_permission": self.has_view_permission(request), |
| } |
| return TemplateResponse(request, "admin/pdf_splitting/workbench.html", context) |
|
|
| def has_add_permission(self, request: HttpRequest) -> bool: |
| return False |
|
|
| def has_change_permission(self, request: HttpRequest, obj: PdfSplittingTool | None = None) -> bool: |
| return False |
|
|
| def has_delete_permission(self, request: HttpRequest, obj: PdfSplittingTool | None = None) -> bool: |
| return False |
|
|
| def get_model_perms(self, request: HttpRequest) -> dict[str, bool]: |
| return {"view": True} |
|
|
|
|
| @admin.register(PdfSplitJob) |
| class PdfSplitJobAdmin(admin.ModelAdmin[PdfSplitJob]): |
| list_display = [ |
| "id", |
| "status_display", |
| "source_original_name", |
| "source_type", |
| "split_mode", |
| "ocr_profile", |
| "progress", |
| "template_key", |
| "created_by", |
| "created_at", |
| ] |
| list_filter = ["status", "source_type", "split_mode", "ocr_profile", "template_key", "created_at"] |
| search_fields = ["source_original_name", "source_abs_path", "id"] |
| ordering = ["-created_at"] |
| readonly_fields = [ |
| "id", |
| "source_type", |
| "source_abs_path", |
| "source_relpath", |
| "source_original_name", |
| "split_mode", |
| "template_key", |
| "template_version", |
| "ocr_profile", |
| "status", |
| "total_pages", |
| "processed_pages", |
| "progress", |
| "current_page", |
| "task_id", |
| "cancel_requested", |
| "created_by", |
| "summary_payload", |
| "error_message", |
| "export_zip_relpath", |
| "started_at", |
| "finished_at", |
| "created_at", |
| "updated_at", |
| ] |
| change_form_template = "admin/pdf_splitting/pdfsplitjob/change_form.html" |
|
|
| def has_add_permission(self, request: HttpRequest) -> bool: |
| return False |
|
|
| def status_display(self, obj: PdfSplitJob) -> SafeString: |
| color_map = { |
| "pending": "#8d6e63", |
| "processing": "#1565c0", |
| "review_required": "#ef6c00", |
| "exporting": "#5e35b1", |
| "completed": "#2e7d32", |
| "failed": "#c62828", |
| "cancelled": "#546e7a", |
| } |
| color = color_map.get(obj.status, "#546e7a") |
| return format_html('<span style="color:{};font-weight:700;">{}</span>', color, obj.get_status_display()) |
|
|
| status_display.short_description = _("状态") |
|
|
| def change_view( |
| self, |
| request: HttpRequest, |
| object_id: str, |
| form_url: str = "", |
| extra_context: dict[str, Any] | None = None, |
| ) -> TemplateResponse: |
| extra = extra_context or {} |
| extra["pdf_split_job_id"] = object_id |
| extra["pdf_split_status_api"] = f"/api/v1/pdf-splitting/jobs/{object_id}" |
| extra["pdf_split_confirm_api"] = f"/api/v1/pdf-splitting/jobs/{object_id}/confirm" |
| extra["pdf_split_cancel_api"] = f"/api/v1/pdf-splitting/jobs/{object_id}/cancel" |
| extra["pdf_split_download_api"] = f"/api/v1/pdf-splitting/jobs/{object_id}/download" |
| extra["pdf_split_preview_api_base"] = f"/api/v1/pdf-splitting/jobs/{object_id}/pages" |
| extra["pdf_split_pdf_api"] = f"/api/v1/pdf-splitting/jobs/{object_id}/pdf" |
| return super().change_view(request, object_id, form_url, extra_context=extra) |
|
|
