95 lines
3.6 KiB
Python
95 lines
3.6 KiB
Python
from django.db import models
|
|
|
|
|
|
class SubmissionBatch(models.Model):
|
|
"""
|
|
资料包主对象，承接导入、会话绑定和目录汇总结果。
|
|
|
|
Documents 模块负责维护资料包与文件的关系，
|
|
不在模型层耦合 Agent 执行细节。
|
|
"""
|
|
|
|
STATUS_PENDING = "pending"
|
|
STATUS_PROCESSING = "processing"
|
|
STATUS_COMPLETED = "completed"
|
|
STATUS_REVIEW_REQUIRED = "review_required"
|
|
STATUS_FAILED = "failed"
|
|
|
|
batch_id = models.CharField(max_length=64, unique=True, db_index=True)
|
|
product_name = models.CharField(max_length=255, blank=True, db_index=True)
|
|
workflow_type = models.CharField(max_length=64, default="registration")
|
|
conversation_id = models.CharField(max_length=64, blank=True, db_index=True)
|
|
file_count = models.PositiveIntegerField(default=0)
|
|
page_count = models.PositiveIntegerField(default=0)
|
|
chapter_summary = models.JSONField(default=list, blank=True)
|
|
import_status = models.CharField(max_length=32, default=STATUS_PENDING, db_index=True)
|
|
exception_count = models.PositiveIntegerField(default=0)
|
|
created_at = models.DateTimeField(auto_now_add=True, db_index=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
ordering = ["-created_at"]
|
|
|
|
def __str__(self) -> str:
|
|
return self.product_name or self.batch_id
|
|
|
|
def get_import_status_display_text(self) -> str:
|
|
return {
|
|
self.STATUS_PENDING: "待导入",
|
|
self.STATUS_PROCESSING: "处理中",
|
|
self.STATUS_COMPLETED: "已完成",
|
|
self.STATUS_REVIEW_REQUIRED: "待复核",
|
|
self.STATUS_FAILED: "失败",
|
|
}.get(self.import_status, self.import_status)
|
|
|
|
|
|
class UploadedDocument(models.Model):
|
|
"""
|
|
保存用户上传文档的元数据和入库状态。
|
|
|
|
设计上只记录“文件属于哪个场景、当前是否已入库、失败原因是什么”，
|
|
不把 RAG 细节耦合进模型层。
|
|
"""
|
|
# 文档状态用于驱动前端提示和后续可操作项。
|
|
STATUS_UPLOADED = "uploaded"
|
|
STATUS_INDEXED = "indexed"
|
|
STATUS_FAILED = "failed"
|
|
|
|
batch = models.ForeignKey(
|
|
SubmissionBatch,
|
|
related_name="documents",
|
|
null=True,
|
|
blank=True,
|
|
on_delete=models.CASCADE,
|
|
)
|
|
scenario_id = models.CharField(max_length=100, db_index=True)
|
|
original_name = models.CharField(max_length=255)
|
|
file = models.FileField(upload_to="documents/%Y%m%d/")
|
|
file_type = models.CharField(max_length=20)
|
|
size = models.PositiveIntegerField(default=0)
|
|
relative_path = models.CharField(max_length=500, blank=True)
|
|
chapter_code = models.CharField(max_length=32, blank=True)
|
|
document_role = models.CharField(max_length=64, blank=True)
|
|
page_count = models.PositiveIntegerField(default=0)
|
|
page_count_confidence = models.CharField(max_length=32, blank=True)
|
|
chapter_match_status = models.CharField(max_length=32, blank=True)
|
|
needs_manual_review = models.BooleanField(default=False)
|
|
status = models.CharField(max_length=20, default=STATUS_UPLOADED, db_index=True)
|
|
error_message = models.TextField(blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
ordering = ["-created_at"]
|
|
|
|
def __str__(self) -> str:
|
|
return self.original_name
|
|
|
|
def get_status_display_text(self) -> str:
|
|
"""为模板提供更适合演示的中文状态文案。"""
|
|
return {
|
|
self.STATUS_UPLOADED: "已上传，待入库",
|
|
self.STATUS_INDEXED: "已入库，可检索",
|
|
self.STATUS_FAILED: "入库失败",
|
|
}.get(self.status, self.status)
|