Fixed triple nested to not use iterator

Author: seshubaws

aws_lambda_powertools/utilities/data_classes/poc.py

@@ -1,12 +1,11 @@
 from aws_lambda_powertools.utilities.data_classes import S3Event, SQSEvent, SNSEvent, EventBridgeEvent, KinesisFirehoseEvent
 from aws_lambda_powertools.utilities.data_classes.sns_event import SNSMessage
-from aws_lambda_powertools.utilities.data_classes.ses_event import SESMessage
+from aws_lambda_powertools.utilities.data_classes.ses_event import SESEventRecord, SESMailCommonHeaders
 from aws_lambda_powertools.utilities.data_classes.cloud_watch_logs_event import CloudWatchLogsLogEvent
 from aws_lambda_powertools.utilities.data_classes.s3_event import S3EventBridgeNotificationDetail
 import aws_lambda_powertools.utilities.data_classes.nested_test_events as nested_test_events
 
 
-
 def lambda_handler_sqs_s3(event: SQSEvent = nested_test_events.sqs_s3_event): # sqs(s3)
     sqs_event = SQSEvent(event)
     s3_event = sqs_event.decode_nested_events(S3Event)
@@ -41,21 +40,28 @@ def lambda_handler_sqs_s3_multi(event: SQSEvent = nested_test_events.sqs_s3_mult
         print('sqs_s3_multi_event bucket:', rec.bucket_name)
 
 
-def lambda_handler_sqs_sns_s3(event: SQSEvent = nested_test_events.sqs_sns_s3_event): # sqs(sns(s3))
+def lambda_handler_sqs_sns_s3_direct(event: SQSEvent = nested_test_events.sqs_sns_s3_event): # sqs(sns(s3))
     sqs_event = SQSEvent(event)
-    sns_event = sqs_event.decode_nested_events(SNSMessage)
-    for rec in sns_event:
-        # print('sns rec:', type(rec), rec)
-        # print('sns message:', rec.message)
+    sns_event = sqs_event.decode_nested_event(SNSMessage)
+    s3_event = sns_event.decode_nested_event(S3Event)
+    print('sqs_sns_s3_direct bucket:', s3_event.bucket_name)
 
-        s3_event = rec.decode_nested_events(S3Event)
-        # print('s3 rec:', type(s3_event), s3_event)
-        print('sqs_sns_s3_event bucket:', next(s3_event).bucket_name)
+
+def lambda_handler_sqs_sns_s3(event: SQSEvent = nested_test_events.sqs_sns_s3_event): # sqs(sns(s3))
+    sqs_event = SQSEvent(event)
+    for rec in sqs_event:
+        sns_event = sqs_event.decode_nested_events(SNSMessage)
+        for rec in sns_event:
+            # print('rec:', type(rec), rec)
+            s3_event = rec.decode_nested_events(S3Event)
+            for r in s3_event:
+                # print(type(r))
+                print('sqs_sns_s3 bucket:', r.bucket_name)
 
 
 def lambda_handler_sns_ses(event: SNSEvent = nested_test_events.sns_ses_event): # sns(ses)
     sns_event = SNSEvent(event)
-    ses_event = sns_event.decode_nested_events(SESMessage)
+    ses_event = sns_event.decode_nested_events(SESEventRecord)
     for rec in ses_event:
         # print('rec:', type(rec), rec)
         print('sns_ses_event email:', rec.get("mail").get('source'))
@@ -68,6 +74,7 @@ def lambda_handler_eb_s3(event: EventBridgeEvent = nested_test_events.eb_s3_even
         # print('type:', type(rec))
         print('eb_s3_event bucket:', rec.bucket.name)
 
+
 def lambda_handler_sqs_eb_s3(event: nested_test_events.sqs_eb_s3_event): # sqs(eventbridge(s3))
     sqs_event = SQSEvent(event)
     for rec in sqs_event:
@@ -79,10 +86,10 @@ def lambda_handler_sqs_eb_s3(event: nested_test_events.sqs_eb_s3_event): # sqs(e
                 # print(type(r))
                 print('sqs_eb_s3_event bucket:', r.bucket.name)
 
+
 def lambda_handler_firehose_sns_event(event = nested_test_events.firehose_sns_event): # firehose(sns)
     firehose_event = KinesisFirehoseEvent(event)
     sns_event = firehose_event.decode_nested_events(SNSMessage)
-    #just gives back the encrypted data, so can't cast it into SNSEvent, have to decrypt it and then cast it to get the SNSEvent
     for rec in sns_event:
         print('firehose_sns_event message:', rec.message)
 
@@ -94,7 +101,7 @@ def lambda_handler_firehose_sns_event(event = nested_test_events.firehose_sns_ev
 #         print('type:', type(rec), rec)
 
 
-
+lambda_handler_sqs_sns_s3_direct(nested_test_events.sqs_sns_s3_event)
 lambda_handler_sqs_s3(nested_test_events.sqs_s3_event)
 lambda_handler_sqs_s3_single(nested_test_events.sqs_s3_event)
 lambda_handler_sqs_sns(nested_test_events.sqs_sns_event)

aws_lambda_powertools/utilities/data_classes/ses_event.py

@@ -3,7 +3,7 @@
 from aws_lambda_powertools.utilities.data_classes.common import DictWrapper, EventWrapper
 
 
-class SESMailHeader(DictWrapper):
+class SESMailHeader(EventWrapper):
     @property
     def name(self) -> str:
         return self["name"]
@@ -13,7 +13,7 @@ def value(self) -> str:
         return self["value"]
 
 
-class SESMailCommonHeaders(DictWrapper):
+class SESMailCommonHeaders(EventWrapper):
     @property
     def return_path(self) -> str:
         """The values in the Return-Path header of the email."""
@@ -111,7 +111,7 @@ def common_headers(self) -> SESMailCommonHeaders:
         return SESMailCommonHeaders(self["commonHeaders"])
 
 
-class SESReceiptStatus(DictWrapper):
+class SESReceiptStatus(EventWrapper):
     @property
     def status(self) -> str:
         """Receipt status
@@ -120,7 +120,7 @@ def status(self) -> str:
         return str(self["status"])
 
 
-class SESReceiptAction(DictWrapper):
+class SESReceiptAction(EventWrapper):
     @property
     def get_type(self) -> str:
         """String that indicates the type of action that was executed.
@@ -149,7 +149,7 @@ def invocation_type(self) -> str:
         return self["invocationType"]
 
 
-class SESReceipt(DictWrapper):
+class SESReceipt(EventWrapper):
     @property
     def timestamp(self) -> str:
         """String that specifies the date and time at which the action was triggered, in ISO 8601 format."""