first version able to send

server/app/mailer/persistence/campaigns.py

@@ -14,6 +14,7 @@ from app.db.models import (
     CampaignJob,
     CampaignStatus,
     CampaignVersion,
+    CampaignVersionWorkflowState,
     JobBuildStatus,
     JobImapStatus,
     JobQueueStatus,
@@ -154,7 +155,15 @@ def load_version_config(session: Session, version_id: str):
     return version, path, load_campaign_config(path)
 
 
-def validate_campaign_version(session: Session, *, tenant_id: str, version_id: str, check_files: bool = False) -> dict[str, Any]:
+def validate_campaign_version(
+    session: Session,
+    *,
+    tenant_id: str,
+    version_id: str,
+    check_files: bool = False,
+    user_id: str | None = None,
+    lock_on_success: bool = True,
+) -> dict[str, Any]:
     version, snapshot_path, config = load_version_config(session, version_id)
     campaign = session.get(Campaign, version.campaign_id)
     if not campaign or campaign.tenant_id != tenant_id:
@@ -186,8 +195,15 @@ def validate_campaign_version(session: Session, *, tenant_id: str, version_id: s
 
     campaign.status = CampaignStatus.VALIDATED.value if report.ok else CampaignStatus.NEEDS_REVIEW.value
     if report.ok:
-        version.workflow_state = "under_review"
+        version.workflow_state = CampaignVersionWorkflowState.APPROVED.value
         version.is_complete = True
+        if lock_on_success and version.locked_at is None:
+            from datetime import UTC, datetime
+
+            version.locked_at = datetime.now(UTC)
+            version.locked_by_user_id = user_id
+    else:
+        version.workflow_state = CampaignVersionWorkflowState.EDITING.value
     session.add(version)
     session.add(campaign)
     session.commit()
@@ -248,6 +264,15 @@ def build_campaign_version(
     campaign = session.get(Campaign, version.campaign_id)
     if not campaign or campaign.tenant_id != tenant_id:
         raise CampaignPersistenceError("Campaign version is not accessible for this tenant")
+    if version.workflow_state == CampaignVersionWorkflowState.COMPLETED.value:
+        raise CampaignPersistenceError("Sent campaign versions cannot be rebuilt")
+    validation_summary = version.validation_summary if isinstance(version.validation_summary, dict) else {}
+    if not validation_summary.get("ok"):
+        raise CampaignPersistenceError("Campaign version must be successfully validated before messages are built")
+    if version.locked_at is None:
+        from datetime import UTC, datetime
+
+        version.locked_at = datetime.now(UTC)
 
     output_dir = BUILD_OUTPUT_DIR / campaign.id / version.id
     result = build_campaign_messages(config, campaign_file=snapshot_path, output_dir=output_dir, write_eml=write_eml)
@@ -296,10 +321,10 @@ def build_campaign_version(
 
     if result.report.needs_review_count or result.report.blocked_count:
         campaign.status = CampaignStatus.NEEDS_REVIEW.value
-        version.workflow_state = "under_review"
+        version.workflow_state = CampaignVersionWorkflowState.APPROVED.value
     elif result.report.queueable_count > 0:
         campaign.status = CampaignStatus.READY_TO_QUEUE.value
-        version.workflow_state = "built"
+        version.workflow_state = CampaignVersionWorkflowState.BUILT.value
     else:
         campaign.status = CampaignStatus.VALIDATED.value