File tree 2 files changed +6
-1
lines changed
aws_lambda_powertools/utilities/data_classes
2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -31,9 +31,13 @@ def sequence_number(self) -> str:
31
31
"""The unique identifier of the record within its shard"""
32
32
return self ["kinesis" ]["sequenceNumber" ]
33
33
34
+ def data_as_bytes (self ) -> bytes :
35
+ """Decode binary encoded data as bytes"""
36
+ return base64 .b64decode (self .data )
37
+
34
38
def data_as_text (self ) -> str :
35
39
"""Decode binary encoded data as text"""
36
- return base64 . b64decode ( self .data ).decode ("utf-8" )
40
+ return self .data_as_bytes ( ).decode ("utf-8" )
37
41
38
42
def data_as_json (self ) -> dict :
39
43
"""Decode binary encoded data as json"""
Original file line number Diff line number Diff line change @@ -1077,6 +1077,7 @@ def test_kinesis_stream_event():
1077
1077
assert kinesis .partition_key == "1"
1078
1078
assert kinesis .sequence_number == "49590338271490256608559692538361571095921575989136588898"
1079
1079
1080
+ assert kinesis .data_as_bytes () == b"Hello, this is a test."
1080
1081
assert kinesis .data_as_text () == "Hello, this is a test."
1081
1082
1082
1083
You can’t perform that action at this time.
0 commit comments