-
Notifications
You must be signed in to change notification settings - Fork 58
feat: base CloudEvent
class as per v1 specs, including attribute validation
#242
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 17 commits
a2ac762
8db1e29
35dee7d
42b4fe1
f83c363
9d1aa35
aa81ca0
b2b0649
b202325
c5e6df9
6e13f72
e78a70b
443aee9
1d43d68
21493e1
68337f9
d0bba86
599d05c
43f1d0c
7d18098
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
# Copyright 2018-Present The CloudEvents Authors | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); you may | ||
# not use this file except in compliance with the License. You may obtain | ||
# a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | ||
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | ||
# License for the specific language governing permissions and limitations | ||
# under the License. | ||
|
||
""" | ||
CloudEvent implementation for v1.0 | ||
""" |
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,296 @@ | ||
# Copyright 2018-Present The CloudEvents Authors | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); you may | ||
# not use this file except in compliance with the License. You may obtain | ||
# a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | ||
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the | ||
# License for the specific language governing permissions and limitations | ||
# under the License. | ||
|
||
import re | ||
from collections import defaultdict | ||
from datetime import datetime | ||
from typing import Any, Final, Optional | ||
|
||
from cloudevents.core.v1.exceptions import ( | ||
BaseCloudEventException, | ||
CloudEventValidationError, | ||
CustomExtensionAttributeError, | ||
InvalidAttributeTypeError, | ||
InvalidAttributeValueError, | ||
MissingRequiredAttributeError, | ||
) | ||
|
||
REQUIRED_ATTRIBUTES: Final[list[str]] = ["id", "source", "type", "specversion"] | ||
OPTIONAL_ATTRIBUTES: Final[list[str]] = [ | ||
"datacontenttype", | ||
"dataschema", | ||
"subject", | ||
"time", | ||
] | ||
|
||
|
||
class CloudEvent: | ||
PlugaruT marked this conversation as resolved.
Show resolved
Hide resolved
|
||
""" | ||
The CloudEvent Python wrapper contract exposing generically-available | ||
properties and APIs. | ||
|
||
Implementations might handle fields and have other APIs exposed but are | ||
obliged to follow this contract. | ||
""" | ||
|
||
def __init__(self, attributes: dict[str, Any], data: Optional[dict] = None) -> None: | ||
""" | ||
Create a new CloudEvent instance. | ||
|
||
:param attributes: The attributes of the CloudEvent instance. | ||
:param data: The payload of the CloudEvent instance. | ||
|
||
:raises ValueError: If any of the required attributes are missing or have invalid values. | ||
:raises TypeError: If any of the attributes have invalid types. | ||
""" | ||
self._validate_attribute(attributes) | ||
self._attributes: dict[str, Any] = attributes | ||
self._data: Optional[dict] = data | ||
|
||
@staticmethod | ||
def _validate_attribute(attributes: dict[str, Any]) -> None: | ||
""" | ||
Validates the attributes of the CloudEvent as per the CloudEvents specification. | ||
|
||
See https://github.com/cloudevents/spec/blob/main/cloudevents/spec.md#required-attributes | ||
""" | ||
errors: dict[str, list[BaseCloudEventException]] = defaultdict(list) | ||
errors.update(CloudEvent._validate_required_attributes(attributes)) | ||
errors.update(CloudEvent._validate_optional_attributes(attributes)) | ||
errors.update(CloudEvent._validate_extension_attributes(attributes)) | ||
if errors: | ||
raise CloudEventValidationError(dict(errors)) | ||
PlugaruT marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
||
@staticmethod | ||
def _validate_required_attributes( | ||
attributes: dict[str, Any], | ||
) -> dict[str, list[BaseCloudEventException]]: | ||
""" | ||
Validates the types of the required attributes. | ||
|
||
:param attributes: The attributes of the CloudEvent instance. | ||
:return: A dictionary of validation error messages. | ||
""" | ||
errors = defaultdict(list) | ||
|
||
if "id" not in attributes: | ||
errors["id"].append(MissingRequiredAttributeError(attribute_name="id")) | ||
PlugaruT marked this conversation as resolved.
Show resolved
Hide resolved
|
||
if attributes.get("id") is None: | ||
errors["id"].append( | ||
InvalidAttributeValueError("id", "Attribute 'id' must not be None") | ||
) | ||
if not isinstance(attributes.get("id"), str): | ||
errors["id"].append(InvalidAttributeTypeError("id", str)) | ||
|
||
if "source" not in attributes: | ||
errors["source"].append( | ||
MissingRequiredAttributeError(attribute_name="source") | ||
) | ||
if not isinstance(attributes.get("source"), str): | ||
errors["source"].append(InvalidAttributeTypeError("source", str)) | ||
|
||
if "type" not in attributes: | ||
errors["type"].append(MissingRequiredAttributeError(attribute_name="type")) | ||
if not isinstance(attributes.get("type"), str): | ||
errors["type"].append(InvalidAttributeTypeError("type", str)) | ||
|
||
if "specversion" not in attributes: | ||
errors["specversion"].append( | ||
MissingRequiredAttributeError(attribute_name="specversion") | ||
) | ||
if not isinstance(attributes.get("specversion"), str): | ||
errors["specversion"].append(InvalidAttributeTypeError("specversion", str)) | ||
if attributes.get("specversion") != "1.0": | ||
errors["specversion"].append( | ||
InvalidAttributeValueError( | ||
"specversion", "Attribute 'specversion' must be '1.0'" | ||
) | ||
) | ||
return errors | ||
|
||
@staticmethod | ||
def _validate_optional_attributes( | ||
attributes: dict[str, Any], | ||
) -> dict[str, list[BaseCloudEventException]]: | ||
""" | ||
Validates the types and values of the optional attributes. | ||
|
||
:param attributes: The attributes of the CloudEvent instance. | ||
:return: A dictionary of validation error messages. | ||
""" | ||
errors = defaultdict(list) | ||
|
||
if "time" in attributes: | ||
if not isinstance(attributes["time"], datetime): | ||
errors["time"].append(InvalidAttributeTypeError("time", datetime)) | ||
if hasattr(attributes["time"], "tzinfo") and not attributes["time"].tzinfo: | ||
errors["time"].append( | ||
InvalidAttributeValueError( | ||
"time", "Attribute 'time' must be timezone aware" | ||
) | ||
) | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I don't recall this being required. Where did you get this requirement from? There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I think here, we have 2 options, either default to UTC, but that would be a controversial behaviour, as we can't default the timezone of the clients, or, make it an explicit requirement. |
||
if "subject" in attributes: | ||
if not isinstance(attributes["subject"], str): | ||
errors["subject"].append(InvalidAttributeTypeError("subject", str)) | ||
if not attributes["subject"]: | ||
errors["subject"].append( | ||
InvalidAttributeValueError( | ||
"subject", "Attribute 'subject' must not be empty" | ||
) | ||
) | ||
if "datacontenttype" in attributes: | ||
if not isinstance(attributes["datacontenttype"], str): | ||
errors["datacontenttype"].append( | ||
InvalidAttributeTypeError("datacontenttype", str) | ||
) | ||
if not attributes["datacontenttype"]: | ||
errors["datacontenttype"].append( | ||
InvalidAttributeValueError( | ||
"datacontenttype", | ||
"Attribute 'datacontenttype' must not be empty", | ||
) | ||
) | ||
if "dataschema" in attributes: | ||
if not isinstance(attributes["dataschema"], str): | ||
errors["dataschema"].append( | ||
InvalidAttributeTypeError("dataschema", str) | ||
) | ||
if not attributes["dataschema"]: | ||
errors["dataschema"].append( | ||
InvalidAttributeValueError( | ||
"dataschema", "Attribute 'dataschema' must not be empty" | ||
) | ||
) | ||
return errors | ||
|
||
@staticmethod | ||
def _validate_extension_attributes( | ||
attributes: dict[str, Any], | ||
) -> dict[str, list[BaseCloudEventException]]: | ||
""" | ||
Validates the extension attributes. | ||
|
||
:param attributes: The attributes of the CloudEvent instance. | ||
:return: A dictionary of validation error messages. | ||
""" | ||
errors = defaultdict(list) | ||
extension_attributes = [ | ||
key | ||
for key in attributes.keys() | ||
if key not in REQUIRED_ATTRIBUTES and key not in OPTIONAL_ATTRIBUTES | ||
] | ||
for extension_attribute in extension_attributes: | ||
if extension_attribute == "data": | ||
errors[extension_attribute].append( | ||
CustomExtensionAttributeError( | ||
extension_attribute, | ||
"Extension attribute 'data' is reserved and must not be used", | ||
) | ||
) | ||
if not (1 <= len(extension_attribute) <= 20): | ||
errors[extension_attribute].append( | ||
CustomExtensionAttributeError( | ||
extension_attribute, | ||
f"Extension attribute '{extension_attribute}' should be between 1 and 20 characters long", | ||
) | ||
) | ||
if not re.match(r"^[a-z0-9]+$", extension_attribute): | ||
errors[extension_attribute].append( | ||
CustomExtensionAttributeError( | ||
extension_attribute, | ||
f"Extension attribute '{extension_attribute}' should only contain lowercase letters and numbers", | ||
) | ||
) | ||
return errors | ||
|
||
def get_id(self) -> str: | ||
""" | ||
Retrieve the ID of the event. | ||
|
||
:return: The ID of the event. | ||
""" | ||
return self._attributes["id"] # type: ignore | ||
|
||
def get_source(self) -> str: | ||
""" | ||
Retrieve the source of the event. | ||
|
||
:return: The source of the event. | ||
""" | ||
return self._attributes["source"] # type: ignore | ||
|
||
def get_type(self) -> str: | ||
""" | ||
Retrieve the type of the event. | ||
|
||
:return: The type of the event. | ||
""" | ||
return self._attributes["type"] # type: ignore | ||
|
||
def get_specversion(self) -> str: | ||
""" | ||
Retrieve the specversion of the event. | ||
|
||
:return: The specversion of the event. | ||
""" | ||
return self._attributes["specversion"] # type: ignore | ||
|
||
def get_datacontenttype(self) -> Optional[str]: | ||
""" | ||
Retrieve the datacontenttype of the event. | ||
|
||
:return: The datacontenttype of the event. | ||
""" | ||
return self._attributes.get("datacontenttype") | ||
|
||
def get_dataschema(self) -> Optional[str]: | ||
""" | ||
Retrieve the dataschema of the event. | ||
|
||
:return: The dataschema of the event. | ||
""" | ||
return self._attributes.get("dataschema") | ||
|
||
def get_subject(self) -> Optional[str]: | ||
""" | ||
Retrieve the subject of the event. | ||
|
||
:return: The subject of the event. | ||
""" | ||
return self._attributes.get("subject") | ||
|
||
def get_time(self) -> Optional[datetime]: | ||
""" | ||
Retrieve the time of the event. | ||
|
||
:return: The time of the event. | ||
""" | ||
return self._attributes.get("time") | ||
|
||
def get_extension(self, extension_name: str) -> Any: | ||
""" | ||
Retrieve an extension attribute of the event. | ||
|
||
:param extension_name: The name of the extension attribute. | ||
:return: The value of the extension attribute. | ||
""" | ||
return self._attributes.get(extension_name) | ||
|
||
def get_data(self) -> Optional[dict]: | ||
""" | ||
Retrieve data of the event. | ||
|
||
:return: The data of the event. | ||
""" | ||
return self._data |
Uh oh!
There was an error while loading. Please reload this page.