Spaces:

jun2024
/

fc

Paused

fc / apps /documents /admin /external_template_admin.py

jun2024

Upload 2241 files

b4f2238 verified 7 months ago

10.7 kB

"""
外部模板 Admin 配置
Requirements: 5.1, 5.2, 5.3, 5.4, 5.5, 5.6, 5.7, 8.4, 9.5, 9.6, 13.4
"""

from __future__ import annotations

import logging
from typing import Any, ClassVar

from django import forms
from django.contrib import admin
from django.core.validators import FileExtensionValidator
from django.http import HttpRequest, HttpResponse, HttpResponseRedirect
from django.template.response import TemplateResponse
from django.urls import reverse
from django.utils.translation import gettext
from django.utils.translation import gettext_lazy as _

from apps.documents.models import ExternalTemplate, ExternalTemplateFieldMapping

logger: logging.Logger = logging.getLogger(__name__)


_SOURCE_NAME_WIDGET = forms.TextInput(
    attrs={
        "class": "vTextField js-court-autocomplete",
        "placeholder": _("请输入法院或机构名称..."),
        "autocomplete": "off",
    }
)


class ExternalTemplateAddForm(forms.ModelForm[ExternalTemplate]):
    """新增外部模板表单"""

    docx_file = forms.FileField(
        label=_("模板文件 (.docx)"),
        validators=[FileExtensionValidator(["docx"])],
        help_text=_("仅支持 .docx 格式，最大 20MB"),
    )

    class Meta:
        model = ExternalTemplate
        fields = ("name", "source_name")
        widgets: ClassVar[dict[str, Any]] = {
            "source_name": _SOURCE_NAME_WIDGET,
        }


class ExternalTemplateChangeForm(forms.ModelForm[ExternalTemplate]):
    """编辑外部模板表单"""

    class Meta:
        model = ExternalTemplate
        fields = ("name", "source_name", "is_active")
        widgets: ClassVar[dict[str, Any]] = {
            "source_name": _SOURCE_NAME_WIDGET,
        }


def _get_analysis_service() -> Any:
    """工厂函数获取分析服务"""
    from apps.documents.services.infrastructure.wiring import get_analysis_service

    return get_analysis_service()


def _get_filling_service() -> Any:
    """工厂函数获取填充服务"""
    from apps.documents.services.infrastructure.wiring import get_filling_service

    return get_filling_service()


class ExternalTemplateFieldMappingInline(admin.TabularInline):  # type: ignore[type-arg]
    """字段映射 Inline（只读展示，由 LLM 分析自动生成）"""

    model = ExternalTemplateFieldMapping
    extra: int = 0
    fields: tuple[str, ...] = (
        "sort_order",
        "position_description",
        "semantic_label",
        "fill_type",
    )
    readonly_fields: tuple[str, ...] = (
        "position_description",
        "semantic_label",
        "fill_type",
    )


@admin.register(ExternalTemplate)
class ExternalTemplateAdmin(admin.ModelAdmin[ExternalTemplate]):  # type: ignore[type-arg]
    """
    外部模板管理
    列表页展示模板信息, 详情页包含字段映射 Inline 和操作按钮.
    通过工厂函数获取 Service, 不直接实例化.
    """

    change_form_template: str = "admin/documents/external_template/change_form.html"

    list_display: ClassVar[list[str]] = [
        "name",
        "source_name",
        "status",
        "version",
        "is_active",
        "updated_at",
    ]
    list_filter: ClassVar[list[str]] = [
        "status",
        "is_active",
    ]
    search_fields: ClassVar[list[str]] = [
        "name",
        "source_name",
    ]
    inlines: ClassVar[list[type[admin.TabularInline]]] = [  # type: ignore[type-arg]
        ExternalTemplateFieldMappingInline,
    ]

    # 新增时显示的字段（含文件上传）
    add_fields: ClassVar[tuple[str, ...]] = (
        "name",
        "source_name",
        "docx_file",
    )

    # 编辑时显示的字段
    change_fields: ClassVar[tuple[str, ...]] = (
        "name",
        "source_name",
        "status",
        "version",
        "is_active",
    )

    # 编辑时只读字段
    change_readonly_fields: ClassVar[tuple[str, ...]] = (
        "original_filename",
        "file_size_display",
        "uploaded_by_display",
        # "law_firm_display",  # 移除所属律所字段
        "version",
        "status",
        "created_at",
        "updated_at",
    )

    def get_form(
        self,
        request: HttpRequest,
        obj: ExternalTemplate | None = None,
        change: bool = False,
        **kwargs: Any,
    ) -> type[forms.ModelForm[ExternalTemplate]]:
        if obj is None:
            kwargs["form"] = ExternalTemplateAddForm
        else:
            kwargs["form"] = ExternalTemplateChangeForm
        return super().get_form(request, obj, change, **kwargs)  # type: ignore[return-value]

    def get_fields(
        self,
        request: HttpRequest,
        obj: ExternalTemplate | None = None,
    ) -> tuple[str, ...]:
        if obj is None:
            return self.add_fields
        return self.change_fields + self.change_readonly_fields

    def get_readonly_fields(
        self,
        request: HttpRequest,
        obj: ExternalTemplate | None = None,
    ) -> tuple[str, ...]:
        if obj is None:
            return ()
        return self.change_readonly_fields

    @admin.display(description=_("原始文件名"))
    def original_filename(self, obj: ExternalTemplate) -> str:
        return obj.original_filename

    @admin.display(description=_("文件大小"))
    def file_size_display(self, obj: ExternalTemplate) -> str:
        size = obj.file_size
        if size < 1024:
            return f"{size} B"
        if size < 1024 * 1024:
            return f"{size / 1024:.1f} KB"
        return f"{size / 1024 / 1024:.1f} MB"

    @admin.display(description=_("上传者"))
    def uploaded_by_display(self, obj: ExternalTemplate) -> str:
        if obj.uploaded_by:
            return str(obj.uploaded_by.real_name or obj.uploaded_by.username)
        return "-"

    @admin.display(description=_("所属律所"))
    def law_firm_display(self, obj: ExternalTemplate) -> str:
        return str(obj.law_firm.name) if obj.law_firm_id else "-"

    def save_model(
        self,
        request: HttpRequest,
        obj: ExternalTemplate,
        form: Any,
        change: bool,
    ) -> None:
        """新增时通过 AnalysisService.upload_template 处理文件上传"""
        if not change:
            docx_file = form.cleaned_data.get("docx_file")
            if docx_file:
                service = _get_analysis_service()
                template = service.upload_template(
                    file=docx_file,
                    name=form.cleaned_data["name"],
                    source_name=form.cleaned_data.get("source_name", ""),
                    uploaded_by=request.user,
                )
                # 用 service 创建的对象替换 obj，避免重复 save
                obj.pk = template.pk
                obj.__dict__.update(template.__dict__)
                logger.info("模板上传成功: id=%d, name=%s", template.pk, template.name)
                return
        super().save_model(request, obj, form, change)

    def get_urls(self) -> list[Any]:
        """注册自定义 URL: 分析、填充操作页面"""
        from django.urls import path

        urls = super().get_urls()
        custom_urls: list[Any] = [
            path(
                "analyze/<int:template_id>/",
                self.admin_site.admin_view(self.analyze_view),
                name="documents_externaltemplate_analyze",
            ),
            path(
                "fill-action/<int:template_id>/",
                self.admin_site.admin_view(self.fill_action_view),
                name="documents_externaltemplate_fill_action",
            ),
            path(
                "mapping-editor/<int:template_id>/",
                self.admin_site.admin_view(self.mapping_editor_view),
                name="documents_externaltemplate_mapping_editor",
            ),
        ]
        return custom_urls + urls

    def analyze_view(self, request: HttpRequest, template_id: int) -> HttpResponse:
        """触发 LLM 分析并重定向回详情页"""
        from apps.documents.models.external_template import ExternalTemplateFieldMapping

        service = _get_analysis_service()
        try:
            has_mappings = ExternalTemplateFieldMapping.objects.filter(template_id=template_id).exists()
            if has_mappings:
                service.retry_analysis(template_id)
            else:
                service.analyze_template(template_id)
            self.message_user(request, gettext("模板分析已完成"))
        except Exception:
            logger.exception("模板分析失败: template_id=%s", template_id)
            self.message_user(
                request,
                gettext("模板分析失败，请查看日志"),
                level="error",
            )
        change_url = reverse(
            "admin:documents_externaltemplate_change",
            args=[template_id],
        )
        return HttpResponseRedirect(change_url)

    def fill_action_view(self, request: HttpRequest, template_id: int) -> HttpResponse:
        """填充操作页面"""
        template_obj = self.get_object(request, str(template_id))
        if template_obj is None:
            from django.http import Http404

            raise Http404(gettext("模板不存在"))

        service = _get_filling_service()
        custom_fields: list[dict[str, Any]] = service.get_custom_fields(template_id)

        import json as _json

        context: dict[str, Any] = {
            **self.admin_site.each_context(request),
            "opts": self.model._meta,
            "template_obj": template_obj,
            "custom_fields_json": _json.dumps(custom_fields, ensure_ascii=False),
            "title": gettext("填充操作 - %(name)s") % {"name": template_obj.name},
        }
        return TemplateResponse(
            request,
            "admin/documents/external_template/fill_action.html",
            context,
        )

    def mapping_editor_view(self, request: HttpRequest, template_id: int) -> HttpResponse:
        """映射可视化编辑页面"""
        template_obj = self.get_object(request, str(template_id))
        if template_obj is None:
            from django.http import Http404

            raise Http404(gettext("模板不存在"))

        context: dict[str, Any] = {
            **self.admin_site.each_context(request),
            "opts": self.model._meta,
            "template_obj": template_obj,
            "title": gettext("映射编辑 - %(name)s") % {"name": template_obj.name},
        }
        return TemplateResponse(
            request,
            "admin/documents/external_template/mapping_editor.html",
            context,
        )

	"""
	外部模板 Admin 配置

	Requirements: 5.1, 5.2, 5.3, 5.4, 5.5, 5.6, 5.7, 8.4, 9.5, 9.6, 13.4
	"""

	from __future__ import annotations

	import logging
	from typing import Any, ClassVar

	from django import forms
	from django.contrib import admin
	from django.core.validators import FileExtensionValidator
	from django.http import HttpRequest, HttpResponse, HttpResponseRedirect
	from django.template.response import TemplateResponse
	from django.urls import reverse
	from django.utils.translation import gettext
	from django.utils.translation import gettext_lazy as _

	from apps.documents.models import ExternalTemplate, ExternalTemplateFieldMapping

	logger: logging.Logger = logging.getLogger(__name__)


	_SOURCE_NAME_WIDGET = forms.TextInput(
	attrs={
	"class": "vTextField js-court-autocomplete",
	"placeholder": _("请输入法院或机构名称..."),
	"autocomplete": "off",
	}
	)


	class ExternalTemplateAddForm(forms.ModelForm[ExternalTemplate]):
	"""新增外部模板表单"""

	docx_file = forms.FileField(
	label=_("模板文件 (.docx)"),
	validators=[FileExtensionValidator(["docx"])],
	help_text=_("仅支持 .docx 格式，最大 20MB"),
	)

	class Meta:
	model = ExternalTemplate
	fields = ("name", "source_name")
	widgets: ClassVar[dict[str, Any]] = {
	"source_name": _SOURCE_NAME_WIDGET,
	}


	class ExternalTemplateChangeForm(forms.ModelForm[ExternalTemplate]):
	"""编辑外部模板表单"""

	class Meta:
	model = ExternalTemplate
	fields = ("name", "source_name", "is_active")
	widgets: ClassVar[dict[str, Any]] = {
	"source_name": _SOURCE_NAME_WIDGET,
	}


	def _get_analysis_service() -> Any:
	"""工厂函数获取分析服务"""
	from apps.documents.services.infrastructure.wiring import get_analysis_service

	return get_analysis_service()


	def _get_filling_service() -> Any:
	"""工厂函数获取填充服务"""
	from apps.documents.services.infrastructure.wiring import get_filling_service

	return get_filling_service()


	class ExternalTemplateFieldMappingInline(admin.TabularInline): # type: ignore[type-arg]
	"""字段映射 Inline（只读展示，由 LLM 分析自动生成）"""

	model = ExternalTemplateFieldMapping
	extra: int = 0
	fields: tuple[str, ...] = (
	"sort_order",
	"position_description",
	"semantic_label",
	"fill_type",
	)
	readonly_fields: tuple[str, ...] = (
	"position_description",
	"semantic_label",
	"fill_type",
	)


	@admin.register(ExternalTemplate)
	class ExternalTemplateAdmin(admin.ModelAdmin[ExternalTemplate]): # type: ignore[type-arg]
	"""
	外部模板管理

	列表页展示模板信息, 详情页包含字段映射 Inline 和操作按钮.
	通过工厂函数获取 Service, 不直接实例化.
	"""

	change_form_template: str = "admin/documents/external_template/change_form.html"

	list_display: ClassVar[list[str]] = [
	"name",
	"source_name",
	"status",
	"version",
	"is_active",
	"updated_at",
	]
	list_filter: ClassVar[list[str]] = [
	"status",
	"is_active",
	]
	search_fields: ClassVar[list[str]] = [
	"name",
	"source_name",
	]
	inlines: ClassVar[list[type[admin.TabularInline]]] = [ # type: ignore[type-arg]
	ExternalTemplateFieldMappingInline,
	]

	# 新增时显示的字段（含文件上传）
	add_fields: ClassVar[tuple[str, ...]] = (
	"name",
	"source_name",
	"docx_file",
	)

	# 编辑时显示的字段
	change_fields: ClassVar[tuple[str, ...]] = (
	"name",
	"source_name",
	"status",
	"version",
	"is_active",
	)

	# 编辑时只读字段
	change_readonly_fields: ClassVar[tuple[str, ...]] = (
	"original_filename",
	"file_size_display",
	"uploaded_by_display",
	# "law_firm_display", # 移除所属律所字段
	"version",
	"status",
	"created_at",
	"updated_at",
	)

	def get_form(
	self,
	request: HttpRequest,
	obj: ExternalTemplate \| None = None,
	change: bool = False,
	**kwargs: Any,
	) -> type[forms.ModelForm[ExternalTemplate]]:
	if obj is None:
	kwargs["form"] = ExternalTemplateAddForm
	else:
	kwargs["form"] = ExternalTemplateChangeForm
	return super().get_form(request, obj, change, **kwargs) # type: ignore[return-value]

	def get_fields(
	self,
	request: HttpRequest,
	obj: ExternalTemplate \| None = None,
	) -> tuple[str, ...]:
	if obj is None:
	return self.add_fields
	return self.change_fields + self.change_readonly_fields

	def get_readonly_fields(
	self,
	request: HttpRequest,
	obj: ExternalTemplate \| None = None,
	) -> tuple[str, ...]:
	if obj is None:
	return ()
	return self.change_readonly_fields

	@admin.display(description=_("原始文件名"))
	def original_filename(self, obj: ExternalTemplate) -> str:
	return obj.original_filename

	@admin.display(description=_("文件大小"))
	def file_size_display(self, obj: ExternalTemplate) -> str:
	size = obj.file_size
	if size < 1024:
	return f"{size} B"
	if size < 1024 * 1024:
	return f"{size / 1024:.1f} KB"
	return f"{size / 1024 / 1024:.1f} MB"

	@admin.display(description=_("上传者"))
	def uploaded_by_display(self, obj: ExternalTemplate) -> str:
	if obj.uploaded_by:
	return str(obj.uploaded_by.real_name or obj.uploaded_by.username)
	return "-"

	@admin.display(description=_("所属律所"))
	def law_firm_display(self, obj: ExternalTemplate) -> str:
	return str(obj.law_firm.name) if obj.law_firm_id else "-"

	def save_model(
	self,
	request: HttpRequest,
	obj: ExternalTemplate,
	form: Any,
	change: bool,
	) -> None:
	"""新增时通过 AnalysisService.upload_template 处理文件上传"""
	if not change:
	docx_file = form.cleaned_data.get("docx_file")
	if docx_file:
	service = _get_analysis_service()
	template = service.upload_template(
	file=docx_file,
	name=form.cleaned_data["name"],
	source_name=form.cleaned_data.get("source_name", ""),
	uploaded_by=request.user,
	)
	# 用 service 创建的对象替换 obj，避免重复 save
	obj.pk = template.pk
	obj.__dict__.update(template.__dict__)
	logger.info("模板上传成功: id=%d, name=%s", template.pk, template.name)
	return
	super().save_model(request, obj, form, change)

	def get_urls(self) -> list[Any]:
	"""注册自定义 URL: 分析、填充操作页面"""
	from django.urls import path

	urls = super().get_urls()
	custom_urls: list[Any] = [
	path(
	"analyze/<int:template_id>/",
	self.admin_site.admin_view(self.analyze_view),
	name="documents_externaltemplate_analyze",
	),
	path(
	"fill-action/<int:template_id>/",
	self.admin_site.admin_view(self.fill_action_view),
	name="documents_externaltemplate_fill_action",
	),
	path(
	"mapping-editor/<int:template_id>/",
	self.admin_site.admin_view(self.mapping_editor_view),
	name="documents_externaltemplate_mapping_editor",
	),
	]
	return custom_urls + urls

	def analyze_view(self, request: HttpRequest, template_id: int) -> HttpResponse:
	"""触发 LLM 分析并重定向回详情页"""
	from apps.documents.models.external_template import ExternalTemplateFieldMapping

	service = _get_analysis_service()
	try:
	has_mappings = ExternalTemplateFieldMapping.objects.filter(template_id=template_id).exists()
	if has_mappings:
	service.retry_analysis(template_id)
	else:
	service.analyze_template(template_id)
	self.message_user(request, gettext("模板分析已完成"))
	except Exception:
	logger.exception("模板分析失败: template_id=%s", template_id)
	self.message_user(
	request,
	gettext("模板分析失败，请查看日志"),
	level="error",
	)
	change_url = reverse(
	"admin:documents_externaltemplate_change",
	args=[template_id],
	)
	return HttpResponseRedirect(change_url)

	def fill_action_view(self, request: HttpRequest, template_id: int) -> HttpResponse:
	"""填充操作页面"""
	template_obj = self.get_object(request, str(template_id))
	if template_obj is None:
	from django.http import Http404

	raise Http404(gettext("模板不存在"))

	service = _get_filling_service()
	custom_fields: list[dict[str, Any]] = service.get_custom_fields(template_id)

	import json as _json

	context: dict[str, Any] = {
	**self.admin_site.each_context(request),
	"opts": self.model._meta,
	"template_obj": template_obj,
	"custom_fields_json": _json.dumps(custom_fields, ensure_ascii=False),
	"title": gettext("填充操作 - %(name)s") % {"name": template_obj.name},
	}
	return TemplateResponse(
	request,
	"admin/documents/external_template/fill_action.html",
	context,
	)

	def mapping_editor_view(self, request: HttpRequest, template_id: int) -> HttpResponse:
	"""映射可视化编辑页面"""
	template_obj = self.get_object(request, str(template_id))
	if template_obj is None:
	from django.http import Http404

	raise Http404(gettext("模板不存在"))

	context: dict[str, Any] = {
	**self.admin_site.each_context(request),
	"opts": self.model._meta,
	"template_obj": template_obj,
	"title": gettext("映射编辑 - %(name)s") % {"name": template_obj.name},
	}
	return TemplateResponse(
	request,
	"admin/documents/external_template/mapping_editor.html",
	context,
	)