feat(regulatory): 增加整改复核闭环
This commit is contained in:
@@ -0,0 +1,77 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from django.utils import timezone
|
||||
|
||||
from review_agent.models import FileSummaryBatch, RegulatoryIssue, RegulatoryReviewBatch
|
||||
from review_agent.regulatory_review.services.rule_loader import load_rule_file
|
||||
from review_agent.regulatory_review.storage import save_artifact
|
||||
|
||||
|
||||
def review_missing_issues(
|
||||
*,
|
||||
batch: RegulatoryReviewBatch,
|
||||
issue_ids: list[int],
|
||||
file_summary_batch: FileSummaryBatch,
|
||||
) -> dict[str, object]:
|
||||
rule_set = load_rule_file()
|
||||
rules_by_code = {rule["code"]: rule for rule in rule_set.get("requirements", [])}
|
||||
items = list(file_summary_batch.items.order_by("file_index"))
|
||||
record = {
|
||||
"type": "review_record",
|
||||
"reviewed_at": timezone.localtime().isoformat(),
|
||||
"source_review_batch_id": batch.pk,
|
||||
"source_review_batch_no": batch.batch_no,
|
||||
"file_summary_batch_id": file_summary_batch.pk,
|
||||
"file_summary_batch_no": file_summary_batch.batch_no,
|
||||
"items": [],
|
||||
}
|
||||
issues = RegulatoryIssue.objects.filter(batch=batch, pk__in=issue_ids).order_by("id")
|
||||
for issue in issues:
|
||||
rule = rules_by_code.get(issue.rule_code, {})
|
||||
matched_files = _match_items(items, [*rule.get("file_keywords", []), issue.title])
|
||||
passed = bool(matched_files)
|
||||
issue.status = RegulatoryIssue.Status.REVIEW_PASSED if passed else RegulatoryIssue.Status.REVIEW_FAILED
|
||||
issue.evidence = {
|
||||
**(issue.evidence or {}),
|
||||
"latest_review": {
|
||||
"file_summary_batch_id": file_summary_batch.pk,
|
||||
"file_summary_batch_no": file_summary_batch.batch_no,
|
||||
"matched_files": matched_files,
|
||||
},
|
||||
}
|
||||
issue.save(update_fields=["status", "evidence", "updated_at"])
|
||||
record["items"].append(
|
||||
{
|
||||
"issue_id": issue.pk,
|
||||
"rule_code": issue.rule_code,
|
||||
"title": issue.title,
|
||||
"status": issue.status,
|
||||
"matched_files": matched_files,
|
||||
}
|
||||
)
|
||||
artifact = save_artifact(
|
||||
batch,
|
||||
name=f"review_record_{timezone.now().strftime('%Y%m%d%H%M%S')}.json",
|
||||
artifact_type="json",
|
||||
content=json.dumps(record, ensure_ascii=False, indent=2),
|
||||
metadata={"artifact": "review_record", "file_summary_batch_id": file_summary_batch.pk},
|
||||
)
|
||||
record["artifact_id"] = artifact.pk
|
||||
return record
|
||||
|
||||
|
||||
def _match_items(items, keywords: list[str]) -> list[dict[str, str]]:
|
||||
normalized_keywords = [str(keyword).lower() for keyword in keywords if keyword]
|
||||
matched = []
|
||||
for item in items:
|
||||
haystack = f"{item.file_name} {item.relative_path} {item.directory_level}".lower()
|
||||
if any(keyword in haystack for keyword in normalized_keywords):
|
||||
matched.append(
|
||||
{
|
||||
"file_name": item.file_name,
|
||||
"relative_path": item.relative_path,
|
||||
"directory_level": item.directory_level,
|
||||
}
|
||||
)
|
||||
return matched
|
||||
Reference in New Issue
Block a user