| """ |
| 文书生成任务和配置模型 |
| |
| 本模块定义文书生成相关的数据模型. |
| """ |
|
|
| from __future__ import annotations |
|
|
| import logging |
| from typing import Any, ClassVar |
|
|
| from django.db import models |
| from django.utils.translation import gettext_lazy as _ |
|
|
| logger = logging.getLogger("apps.documents") |
|
|
|
|
| class GenerationMethod(models.TextChoices): |
| """生成方式""" |
|
|
| TEMPLATE = "template", _("模板生成") |
| AI = "ai", _("AI生成") |
|
|
|
|
| class GenerationStatus(models.TextChoices): |
| """生成状态""" |
|
|
| PENDING = "pending", _("等待中") |
| PROCESSING = "processing", _("生成中") |
| COMPLETED = "completed", _("已完成") |
| FAILED = "failed", _("失败") |
|
|
|
|
| class GenerationTask(models.Model): |
| """ |
| 文书生成任务 |
| |
| 记录每次文书生成的任务信息,包括生成参数、结果等. |
| 支持模板生成和 AI 生成两种方式. |
| |
| Requirements: 6.1.3 |
| """ |
|
|
| id: int |
| case_id: int |
| contract_id: int |
| litigation_session_id: int |
| created_by_id: int |
| case = models.ForeignKey( |
| "cases.Case", |
| on_delete=models.CASCADE, |
| null=True, |
| blank=True, |
| related_name="generation_tasks", |
| verbose_name=_("关联案件"), |
| ) |
| contract = models.ForeignKey( |
| "contracts.Contract", |
| on_delete=models.CASCADE, |
| null=True, |
| blank=True, |
| related_name="generation_tasks", |
| verbose_name=_("关联合同"), |
| ) |
| litigation_session = models.ForeignKey( |
| "litigation_ai.LitigationSession", |
| on_delete=models.SET_NULL, |
| null=True, |
| blank=True, |
| related_name="generation_tasks", |
| verbose_name=_("AI生成会话"), |
| help_text=_("如果是 AI 生成,关联到对应的会话"), |
| ) |
| generation_method = models.CharField( |
| max_length=20, choices=GenerationMethod.choices, default=GenerationMethod.TEMPLATE, verbose_name=_("生成方式") |
| ) |
| document_type = models.CharField( |
| max_length=100, verbose_name=_("文书类型"), help_text=_("如:起诉状、答辩状、合同等") |
| ) |
| template_id = models.IntegerField( |
| null=True, blank=True, verbose_name=_("模板ID"), help_text=_("使用的模板ID(模板生成时)") |
| ) |
| status = models.CharField( |
| max_length=20, choices=GenerationStatus.choices, default=GenerationStatus.PENDING, verbose_name=_("生成状态") |
| ) |
| result_file = models.FileField( |
| upload_to="generated_documents/%Y/%m/", null=True, blank=True, verbose_name=_("生成文件") |
| ) |
| error_message = models.TextField(blank=True, verbose_name=_("错误信息")) |
| metadata: Any = models.JSONField( |
| default=dict, verbose_name=_("任务元数据"), help_text=_("存储生成参数、token消耗等信息") |
| ) |
| created_by = models.ForeignKey( |
| "organization.Lawyer", |
| on_delete=models.SET_NULL, |
| null=True, |
| blank=True, |
| related_name="created_generation_tasks", |
| verbose_name=_("创建人"), |
| ) |
| created_at = models.DateTimeField(auto_now_add=True, verbose_name=_("创建时间")) |
| completed_at = models.DateTimeField(null=True, blank=True, verbose_name=_("完成时间")) |
|
|
| class Meta: |
| app_label = "documents" |
| verbose_name = _("文书生成任务") |
| verbose_name_plural = _("文书生成任务") |
| ordering: ClassVar = ["-created_at"] |
| indexes: ClassVar = [ |
| models.Index(fields=["case", "-created_at"]), |
| models.Index(fields=["contract", "-created_at"]), |
| models.Index(fields=["litigation_session"]), |
| models.Index(fields=["status"]), |
| models.Index(fields=["generation_method"]), |
| models.Index(fields=["created_by", "-created_at"]), |
| ] |
|
|
| def __str__(self) -> str: |
| resource = self.case or self.contract |
| resource_name = resource.name if resource else "未关联" |
| return f"{resource_name} - {self.document_type} ({self.get_status_display()})" |
|
|
| @property |
| def is_ai_generated(self) -> bool: |
| """是否为 AI 生成""" |
| return self.generation_method == GenerationMethod.AI |
|
|
| @property |
| def duration_seconds(self) -> int: |
| """生成耗时(秒)""" |
| if self.completed_at and self.created_at: |
| return int((self.completed_at - self.created_at).total_seconds()) |
| return 0 |
|
|
| @property |
| def folder_template_id(self) -> int | None: |
| return (self.metadata or {}).get("folder_template_id") |
|
|
| @folder_template_id.setter |
| def folder_template_id(self, value: int | None) -> None: |
| self.metadata = self.metadata or {} |
| self.metadata["folder_template_id"] = value |
|
|
| @property |
| def output_path(self) -> str | None: |
| return (self.metadata or {}).get("output_path") |
|
|
| @output_path.setter |
| def output_path(self, value: str | None) -> None: |
| self.metadata = self.metadata or {} |
| self.metadata["output_path"] = value |
|
|
| @property |
| def generated_files(self) -> list[Any]: |
| return list((self.metadata or {}).get("generated_files", [])) |
|
|
| @generated_files.setter |
| def generated_files(self, value: list[Any] | None) -> None: |
| self.metadata = self.metadata or {} |
| self.metadata["generated_files"] = list(value or []) |
|
|
| @property |
| def error_logs(self) -> list[Any]: |
| return list((self.metadata or {}).get("error_logs", [])) |
|
|
| @error_logs.setter |
| def error_logs(self, value: list[Any] | None) -> None: |
| self.metadata = self.metadata or {} |
| self.metadata["error_logs"] = list(value or []) |
|
|
|
|
| class GenerationConfig(models.Model): |
| """ |
| 文书生成配置 |
| |
| 存储文书生成的全局配置,如默认模板、生成参数等. |
| |
| Requirements: 6.1.3 |
| """ |
|
|
| id: int |
| name = models.CharField(max_length=100, unique=True, verbose_name=_("配置名称")) |
| config_type = models.CharField( |
| max_length=50, verbose_name=_("配置类型"), help_text=_("如:default_template、ai_model、generation_params") |
| ) |
| value: Any = models.JSONField(verbose_name=_("配置值")) |
| description = models.TextField(blank=True, verbose_name=_("配置说明")) |
| is_active = models.BooleanField(default=True, verbose_name=_("是否启用")) |
| created_at = models.DateTimeField(auto_now_add=True, verbose_name=_("创建时间")) |
| updated_at = models.DateTimeField(auto_now=True, verbose_name=_("更新时间")) |
|
|
| class Meta: |
| app_label = "documents" |
| verbose_name = _("文书生成配置") |
| verbose_name_plural = _("文书生成配置") |
| ordering: ClassVar = ["config_type", "name"] |
| indexes: ClassVar = [ |
| models.Index(fields=["config_type"]), |
| models.Index(fields=["is_active"]), |
| ] |
|
|
| def __str__(self) -> str: |
| return f"{self.config_type} - {self.name}" |
|
|
| @property |
| def case_type(self) -> str | None: |
| return (self.value or {}).get("case_type") |
|
|
| @property |
| def case_stage(self) -> str | None: |
| return (self.value or {}).get("case_stage") |
|
|
| @property |
| def document_template_id(self) -> int | None: |
| return (self.value or {}).get("document_template_id") |
|
|
| @property |
| def folder_path(self) -> str | None: |
| return (self.value or {}).get("folder_path") |
|
|
| @property |
| def priority(self) -> int: |
| return int((self.value or {}).get("priority", 0)) |
|
|
| @property |
| def condition(self) -> dict[str, Any]: |
| return (self.value or {}).get("condition") or {} |
|
|
