File tree 2 files changed +12
-0
lines changed 2 files changed +12
-0
lines changed Original file line number Diff line number Diff line change @@ -215,6 +215,15 @@ def get_time(self) -> Optional[datetime]:
215
215
:return: The time of the event.
216
216
"""
217
217
return self ._attributes .get ("time" )
218
+
219
+ def get_extension (self , extension_name : str ) -> Any :
220
+ """
221
+ Retrieve an extension attribute of the event.
222
+
223
+ :param extension_name: The name of the extension attribute.
224
+ :return: The value of the extension attribute.
225
+ """
226
+ return self ._attributes .get (extension_name )
218
227
219
228
def get_data (self ) -> Optional [dict ]:
220
229
"""
Original file line number Diff line number Diff line change @@ -219,6 +219,7 @@ def test_cloud_event_constructor() -> None:
219
219
subject = "test_subject"
220
220
time = datetime .now (tz = timezone .utc )
221
221
data = {"key" : "value" }
222
+ customextension = "customExtension"
222
223
223
224
event = CloudEvent (
224
225
attributes = {
@@ -230,6 +231,7 @@ def test_cloud_event_constructor() -> None:
230
231
"dataschema" : dataschema ,
231
232
"subject" : subject ,
232
233
"time" : time ,
234
+ "customextension" : customextension ,
233
235
},
234
236
data = data ,
235
237
)
@@ -242,4 +244,5 @@ def test_cloud_event_constructor() -> None:
242
244
assert event .get_dataschema () == dataschema
243
245
assert event .get_subject () == subject
244
246
assert event .get_time () == time
247
+ assert event .get_extension ("customextension" ) == customextension
245
248
assert event .get_data () == data
You can’t perform that action at this time.
0 commit comments