feat: Add Ses lambda event support to Parser utility #213

Author: Ran Isenberg

Diff for: aws_lambda_powertools/utilities/parser/models/__init__.py

@@ -1,5 +1,6 @@
 from .dynamodb import DynamoDBStreamChangedRecordModel, DynamoDBStreamModel, DynamoDBStreamRecordModel
 from .event_bridge import EventBridgeModel
+from .ses import SesModel, SesRecordModel
 from .sns import SnsModel, SnsNotificationModel, SnsRecordModel
 from .sqs import SqsModel, SqsRecordModel
 
@@ -8,6 +9,8 @@
     "EventBridgeModel",
     "DynamoDBStreamChangedRecordModel",
     "DynamoDBStreamRecordModel",
+    "SesModel",
+    "SesRecordModel",
     "SnsModel",
     "SnsNotificationModel",
     "SnsRecordModel",

Diff for: aws_lambda_powertools/utilities/parser/models/ses.py

@@ -0,0 +1,71 @@
+from datetime import datetime
+from typing import List, Optional
+
+from pydantic import BaseModel, Field
+from pydantic.networks import EmailStr
+from pydantic.types import PositiveInt
+from typing_extensions import Literal
+
+
+class SesReceiptVerdict(BaseModel):
+    status: Literal["PASS", "FAIL", "GRAY", "PROCESSING_FAILED"]
+
+
+class SesReceiptAction(BaseModel):
+    type: Literal["Lambda"]  # noqa A003,VNE003
+    invocationType: Literal["Event"]
+    functionArn: str
+
+
+class SesReceipt(BaseModel):
+    timestamp: datetime
+    processingTimeMillis: PositiveInt
+    recipients: List[EmailStr]
+    spamVerdict: SesReceiptVerdict
+    virusVerdict: SesReceiptVerdict
+    spfVerdict: SesReceiptVerdict
+    dmarcVerdict: SesReceiptVerdict
+    action: SesReceiptAction
+
+
+class SesMailHeaders(BaseModel):
+    name: str
+    value: str
+
+
+class SesMailCommonHeaders(BaseModel):
+    header_from: List[str] = Field(None, alias="from")
+    to: List[str]
+    cc: Optional[List[str]]
+    bcc: Optional[List[str]]
+    sender: Optional[List[str]]
+    reply_to: Optional[List[str]] = Field(None, alias="reply-to")
+    returnPath: EmailStr
+    messageId: str
+    date: str
+    subject: str
+
+
+class SesMail(BaseModel):
+    timestamp: datetime
+    source: EmailStr
+    messageId: str
+    destination: List[EmailStr]
+    headersTruncated: bool
+    headers: List[SesMailHeaders]
+    commonHeaders: SesMailCommonHeaders
+
+
+class SesMessage(BaseModel):
+    mail: SesMail
+    receipt: SesReceipt
+
+
+class SesRecordModel(BaseModel):
+    eventSource: Literal["aws:ses"]
+    eventVersion: str
+    ses: SesMessage
+
+
+class SesModel(BaseModel):
+    Records: List[SesRecordModel]