46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
from __future__ import annotations
|
||
|
||
from django.utils import timezone
|
||
|
||
from review_agent.models import (
|
||
ApplicationFormFillBatch,
|
||
ApplicationFormFillNotificationRecord,
|
||
ExportedSummaryFile,
|
||
)
|
||
|
||
|
||
def notify_completion(
|
||
batch: ApplicationFormFillBatch,
|
||
exports: list[ExportedSummaryFile],
|
||
*,
|
||
fail: bool = False,
|
||
) -> ApplicationFormFillNotificationRecord:
|
||
export_ids = [export.pk for export in exports]
|
||
message_summary = (
|
||
f"自动填表批次 {batch.batch_no} 已完成，"
|
||
f"模板 {', '.join(batch.selected_templates or []) or '未识别'}，"
|
||
f"冲突字段 {len(batch.conflict_summary or [])} 个。"
|
||
)
|
||
if fail:
|
||
return ApplicationFormFillNotificationRecord.objects.create(
|
||
batch=batch,
|
||
recipient=batch.user,
|
||
channel=ApplicationFormFillNotificationRecord.Channel.MOCK,
|
||
template_codes=batch.selected_templates,
|
||
export_ids=export_ids,
|
||
message_summary=message_summary,
|
||
send_status=ApplicationFormFillNotificationRecord.SendStatus.FAILED,
|
||
retry_count=1,
|
||
error_message="mock notification failed",
|
||
)
|
||
return ApplicationFormFillNotificationRecord.objects.create(
|
||
batch=batch,
|
||
recipient=batch.user,
|
||
channel=ApplicationFormFillNotificationRecord.Channel.MOCK,
|
||
template_codes=batch.selected_templates,
|
||
export_ids=export_ids,
|
||
message_summary=message_summary,
|
||
send_status=ApplicationFormFillNotificationRecord.SendStatus.SUCCESS,
|
||
sent_at=timezone.now(),
|
||
)
|