Extended EventWrapper class in some events

Author: seshubaws

aws_lambda_powertools/utilities/data_classes/event_bridge_event.py

@@ -1,9 +1,9 @@
 from typing import Any, Dict, List, Optional
 
-from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
+from aws_lambda_powertools.utilities.data_classes.common import DictWrapper, EventWrapper
 
 
-class EventBridgeEvent(DictWrapper):
+class EventBridgeEvent(EventWrapper):
     """Amazon EventBridge Event
 
     Documentation:

aws_lambda_powertools/utilities/data_classes/kinesis_stream_event.py

@@ -6,7 +6,7 @@
 from aws_lambda_powertools.utilities.data_classes.cloud_watch_logs_event import (
     CloudWatchLogsDecodedData,
 )
-from aws_lambda_powertools.utilities.data_classes.common import DictWrapper
+from aws_lambda_powertools.utilities.data_classes.common import DictWrapper, EventWrapper
 
 
 class KinesisStreamRecordPayload(DictWrapper):
@@ -95,7 +95,7 @@ def kinesis(self) -> KinesisStreamRecordPayload:
         return KinesisStreamRecordPayload(self._data)
 
 
-class KinesisStreamEvent(DictWrapper):
+class KinesisStreamEvent(EventWrapper):
     """Kinesis stream event
 
     Documentation: