| """ |
| 自动命名工具Admin |
| 独立的Admin模块 |
| """ |
|
|
| from __future__ import annotations |
|
|
| from pathlib import Path |
| from typing import Any, ClassVar |
|
|
| from django import forms |
| from django.contrib import admin |
| from django.http import HttpRequest, HttpResponse, HttpResponseRedirect |
| from django.middleware.csrf import get_token |
| from django.urls import path, reverse |
|
|
| from apps.automation.models import NamerTool |
| from apps.automation.services.ai.prompts import DEFAULT_FILENAME_PROMPT |
| from apps.automation.services.document.document_processing import process_uploaded_document |
| from apps.core.interfaces import ServiceLocator |
|
|
|
|
| class AutoNamerToolForm(forms.Form): |
| """自动命名工具表单""" |
|
|
| upload = forms.FileField(required=True, help_text="支持PDF、DOCX和图片文件（JPG、PNG、BMP、TIFF等）") |
| prompt = forms.CharField( |
| required=True, |
| widget=forms.Textarea(attrs={"rows": 8}), |
| initial=DEFAULT_FILENAME_PROMPT, |
| help_text="AI提示词，用于指导模型生成合适的文件名", |
| ) |
| model = forms.CharField(required=True, initial="qwen3:0.6b", help_text="使用的AI模型名称") |
| limit = forms.IntegerField(required=False, help_text="文字提取限制（留空使用默认值1500字）") |
| preview_page = forms.IntegerField(required=False, min_value=1, help_text="PDF预览页码（留空使用默认值第1页）") |
|
|
|
|
| @admin.register(NamerTool) |
| class AutoNamerToolAdmin(admin.ModelAdmin[NamerTool]): |
| """自动命名工具管理类""" |
|
|
| change_list_template = None |
|
|
| def get_urls(self) -> list[Any]: |
| urls = super().get_urls() |
| info = self.model._meta.app_label, self.model._meta.model_name |
| custom: list[Any] = [ |
| path("process/", self.admin_site.admin_view(self.process_view), name="{}_{}_process".format(*info)), |
| path("", self.admin_site.admin_view(self.redirect_to_process)), |
| ] |
| return custom + urls |
|
|
| def redirect_to_process(self, request: HttpRequest) -> HttpResponseRedirect: |
| info = self.model._meta.app_label, self.model._meta.model_name |
| return HttpResponseRedirect(reverse("admin:{}_{}_process".format(*info))) |
|
|
| def _process_view_post(self, request: Any, form: AutoNamerToolForm) -> HttpResponse: |
| """处理 POST 请求""" |
| fp = form.cleaned_data["upload"] |
| prompt = form.cleaned_data["prompt"] |
| model = form.cleaned_data["model"] |
| limit = form.cleaned_data.get("limit") |
| preview_page = form.cleaned_data.get("preview_page") |
|
|
| info = self.model._meta.app_label, self.model._meta.model_name |
| return_url = reverse("admin:{}_{}_process".format(*info)) |
|
|
| try: |
| extraction = process_uploaded_document(fp, limit=limit, preview_page=preview_page) |
| except ValueError as e: |
| return HttpResponse(str(e)) |
|
|
| text = extraction.text or "" |
|
|
| if extraction.kind not in {"pdf", "docx", "image"}: |
| return HttpResponse("不支持的文件类型，支持PDF、DOCX和图片文件（JPG、PNG等）") |
|
|
| if not text.strip(): |
| return self._render_no_text_error(extraction, return_url) |
|
|
| return self._render_ai_result(model, prompt, text, extraction, return_url) |
|
|
| def _render_no_text_error(self, extraction: Any, return_url: str) -> HttpResponse: |
| """渲染无文字内容错误页""" |
| error_msg = "文档中没有提取到文字内容，无法生成命名。" |
| if extraction.kind == "image": |
| error_msg += "<br><em>提示：图片文件可能包含手写文字、复杂排版或图片质量较差，建议尝试更清晰的扫描件。</em>" |
| elif extraction.kind == "pdf": |
| error_msg += "<br><em>提示：PDF可能是扫描版或图片格式，系统已尝试OCR识别但未成功。</em>" |
|
|
| html = f""" |
| <h1>自动命名工具（上传文档 + 提示词 → 模型生成）</h1> |
| <h2>❌ 处理失败</h2> |
| <div style='background:#fff3cd;border:1px solid #ffeaa7; |
| border-radius:5px;padding:15px;margin:15px 0;'> |
| <p style='color:#856404;margin:0;'>{error_msg}</p> |
| </div> |
| <p><strong>文件信息：</strong></p> |
| <ul> |
| <li>文件类型: {extraction.kind.upper()}</li> |
| <li>文件路径: {extraction.file_path}</li> |
| <li>预览图: { |
| '<a href="' + extraction.image_url + '" target="_blank">查看预览图</a>' if extraction.image_url else "无" |
| }</li> |
| </ul> |
| <p><a href='{return_url}'>← 返回重新上传</a></p> |
| """ |
| return HttpResponse(html) |
|
|
| def _render_ai_result(self, model: str, prompt: str, text: str, extraction: Any, return_url: str) -> HttpResponse: |
| """调用 AI 并渲染结果""" |
| try: |
| msg_list = [{"role": "system", "content": prompt}, {"role": "user", "content": text}] |
| llm_service = ServiceLocator.get_llm_service() |
| llm_response = llm_service.chat(messages=msg_list, backend="ollama", model=model, fallback=False) |
| response_text = llm_response.content or "无返回内容" |
|
|
| html = f""" |
| <h1>自动命名工具（上传文档 + 提示词 → 模型生成）</h1> |
| <div style='background:#e7f3ff;border:1px solid #b8daff; |
| border-radius:5px;padding:15px;margin:15px 0;'> |
| <p style='margin:0;'><strong>📄 文件信息：</strong></p> |
| <ul style='margin:5px 0 0 20px;'> |
| <li>文件类型: {extraction.kind.upper()}</li> |
| <li>文本长度: {len(text)} 字符</li> |
| </ul> |
| </div> |
| <h2>📝 提取的文字内容</h2> |
| <div style='background:#f8f9fa;padding:15px;border:1px solid #dee2e6; |
| border-radius:5px;margin:10px 0;'> |
| <pre style='white-space:pre-wrap;max-height:400px;overflow:auto; |
| margin:0;font-family:monospace;'>{text}</pre> |
| </div> |
| <h2>🤖 Ollama 返回结果</h2> |
| <div style='background:#f0f8f0;padding:15px;border:1px solid #c3e6cb; |
| border-radius:5px;margin:10px 0;'> |
| <pre style='white-space:pre-wrap;margin:0;font-family:monospace;'>{response_text}</pre> |
| </div> |
| <div style='margin-top:20px;'> |
| <a href='{return_url}' style='display:inline-block;padding:8px 16px; |
| background:#007bff;color:white;text-decoration:none; |
| border-radius:4px;'>← 返回</a> |
| </div> |
| """ |
| return HttpResponse(html) |
| except Exception as e: |
| import traceback |
|
|
| error_detail = str(e) |
| error_traceback = traceback.format_exc() |
| html = f""" |
| <h1>自动命名工具（上传文档 + 提示词 → 模型生成）</h1> |
| <h2>处理失败</h2> |
| <div style='color: red; margin: 20px 0;'> |
| <h2>错误信息：</h2> |
| <pre style='white-space:pre-wrap;background:#f5f5f5;padding:10px; |
| border:1px solid #ddd;'>{error_detail}</pre> |
| </div> |
| <details style='margin: 20px 0;'> |
| <summary style='cursor: pointer; color: #666;'>查看详细错误堆栈</summary> |
| <pre style='white-space:pre-wrap;background:#f5f5f5;padding:10px; |
| border:1px solid #ddd;font-size:12px;'>{error_traceback}</pre> |
| </details> |
| <p><a href='{return_url}'>返回</a></p> |
| """ |
| return HttpResponse(html) |
|
|
| def process_view(self, request: HttpRequest) -> HttpResponse: |
| """自动命名工具主视图""" |
| if request.method == "POST": |
| form = AutoNamerToolForm(request.POST, request.FILES) |
| if form.is_valid(): |
| return self._process_view_post(request, form) |
| else: |
| form = AutoNamerToolForm() |
|
|
| csrf_token = get_token(request) |
| html = f""" |
| <h1>自动命名工具（上传文档 + 提示词 → 模型生成）</h1> |
| <form method='post' enctype='multipart/form-data'> |
| <input type='hidden' name='csrfmiddlewaretoken' value='{csrf_token}' /> |
| <p> |
| <label>上传文件：</label><br/> |
| <input type='file' name='upload' required/> |
| </p> |
| <p> |
| <label>提示词：</label><br/> |
| <textarea name='prompt' rows='8' required style='width:100%;'>{DEFAULT_FILENAME_PROMPT}</textarea> |
| </p> |
| <p> |
| <label>模型名称：</label><br/> |
| <input type='text' name='model' value='qwen3:0.6b' required style='width:100%;'/> |
| </p> |
| <p> |
| <label>文字提取限制：</label><br/> |
| <input type='number' name='limit' placeholder='留空使用默认值1500字' min='1' style='width:100%;'/> |
| </p> |
| <p> |
| <label>PDF预览页码：</label><br/> |
| <input type='number' name='preview_page' placeholder='留空使用默认值第1页' min='1' style='width:100%;'/> |
| </p> |
| <p><button type='submit' class='default'>提交处理</button></p> |
| </form> |
| """ |
| return HttpResponse(html) |
|
|
| def has_add_permission(self, request: HttpRequest) -> bool: |
| return False |
|
|
| def has_change_permission(self, request: HttpRequest, obj: NamerTool | None = None) -> bool: |
| return True |
|
|
| def has_view_permission(self, request: HttpRequest, obj: NamerTool | None = None) -> bool: |
| return True |
|
|
| def has_delete_permission(self, request: HttpRequest, obj: NamerTool | None = None) -> bool: |
| return False |
|
|
