first commit

modules/email/draft.py

@@ -0,0 +1,70 @@
+from typing import Optional, Dict, Any
+from datetime import datetime
+import uuid
+
+
+class EmailDraft:
+    """
+    Représente un brouillon d'email généré automatiquement.
+    Stockage fichier: Data/email_drafts/<id>.json
+    """
+    def __init__(
+        self,
+        prospect_id: str,
+        to_email: str,
+        subject: str,
+        content: str,
+        status: str = "draft",  # draft | sent | failed
+        template_id: Optional[str] = None,
+        task_id: Optional[str] = None,
+        id: Optional[str] = None,
+        created_at: Optional[str] = None,
+        sent_at: Optional[str] = None,
+        error_message: Optional[str] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+    ):
+        self.id = id or f"ed_{uuid.uuid4().hex[:10]}"
+        self.prospect_id = prospect_id
+        self.to_email = to_email
+        self.subject = subject
+        self.content = content
+        self.status = status
+        self.template_id = template_id
+        self.task_id = task_id
+        self.created_at = created_at or datetime.utcnow().isoformat()
+        self.sent_at = sent_at
+        self.error_message = error_message
+        self.metadata = metadata or {}
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "id": self.id,
+            "prospect_id": self.prospect_id,
+            "to_email": self.to_email,
+            "subject": self.subject,
+            "content": self.content,
+            "status": self.status,
+            "template_id": self.template_id,
+            "task_id": self.task_id,
+            "created_at": self.created_at,
+            "sent_at": self.sent_at,
+            "error_message": self.error_message,
+            "metadata": self.metadata,
+        }
+
+    @staticmethod
+    def from_dict(data: Dict[str, Any]) -> "EmailDraft":
+        return EmailDraft(
+            id=data.get("id"),
+            prospect_id=data.get("prospect_id", ""),
+            to_email=data.get("to_email", ""),
+            subject=data.get("subject", ""),
+            content=data.get("content", ""),
+            status=data.get("status", "draft"),
+            template_id=data.get("template_id"),
+            task_id=data.get("task_id"),
+            created_at=data.get("created_at"),
+            sent_at=data.get("sent_at"),
+            error_message=data.get("error_message"),
+            metadata=data.get("metadata") or {},
+        )