acceldata-sdk-python 26.7.0__py3-none-any.whl

acceldata/models/api/pipeline/span_event.py

@@ -0,0 +1,114 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from acceldata.models.api.pipeline.event_alert_object import EventAlertObject
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class SpanEvent(BaseModel):
+    """
+    SpanEvent
+    """ # noqa: E501
+    alert: Optional[EventAlertObject] = None
+    context_data: Optional[Dict[str, Any]] = Field(default=None, alias="contextData")
+    created_at: Optional[datetime] = Field(default=None, alias="createdAt")
+    event_uid: StrictStr = Field(alias="eventUid")
+    id: Optional[StrictInt] = None
+    meta: Optional[Dict[str, Any]] = None
+    span_id: StrictInt = Field(alias="spanId")
+    span_uid: Optional[StrictStr] = Field(default=None, alias="spanUid")
+    type: StrictStr = Field(description="Allowed values at generation time: SUCCESS, WARNING, ERROR, SKIPPED.")
+    __properties: ClassVar[List[str]] = ["alert", "contextData", "createdAt", "eventUid", "id", "meta", "spanId", "spanUid", "type"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of SpanEvent from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of alert
+        if self.alert:
+            _dict['alert'] = self.alert.to_dict()
+        # set to None if span_uid (nullable) is None
+        # and model_fields_set contains the field
+        if self.span_uid is None and "span_uid" in self.model_fields_set:
+            _dict['spanUid'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SpanEvent from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "alert": EventAlertObject.from_dict(obj["alert"]) if obj.get("alert") is not None else None,
+            "contextData": obj.get("contextData"),
+            "createdAt": obj.get("createdAt"),
+            "eventUid": obj.get("eventUid"),
+            "id": obj.get("id"),
+            "meta": obj.get("meta"),
+            "spanId": obj.get("spanId"),
+            "spanUid": obj.get("spanUid"),
+            "type": obj.get("type")
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/span_event_request.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict
+from typing import Any, ClassVar, Dict, List
+from acceldata.models.api.pipeline.create_span_event import CreateSpanEvent
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class SpanEventRequest(BaseModel):
+    """
+    SpanEventRequest
+    """ # noqa: E501
+    event: CreateSpanEvent
+    __properties: ClassVar[List[str]] = ["event"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of SpanEventRequest from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of event
+        if self.event:
+            _dict['event'] = self.event.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SpanEventRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "event": CreateSpanEvent.from_dict(obj["event"]) if obj.get("event") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/span_event_response.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict
+from typing import Any, ClassVar, Dict, List
+from acceldata.models.api.pipeline.span_event import SpanEvent
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class SpanEventResponse(BaseModel):
+    """
+    SpanEventResponse
+    """ # noqa: E501
+    event: SpanEvent
+    __properties: ClassVar[List[str]] = ["event"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of SpanEventResponse from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of event
+        if self.event:
+            _dict['event'] = self.event.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SpanEventResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "event": SpanEvent.from_dict(obj["event"]) if obj.get("event") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/span_request.py

@@ -0,0 +1,94 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List
+from acceldata.models.api.pipeline.span import Span
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class SpanRequest(BaseModel):
+    """
+    SpanRequest
+    """ # noqa: E501
+    associated_job_uids: List[StrictStr] = Field(alias="associatedJobUids")
+    span: Span
+    __properties: ClassVar[List[str]] = ["associatedJobUids", "span"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of SpanRequest from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of span
+        if self.span:
+            _dict['span'] = self.span.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SpanRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "associatedJobUids": obj.get("associatedJobUids"),
+            "span": Span.from_dict(obj["span"]) if obj.get("span") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/spans_response.py

@@ -0,0 +1,96 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict
+from typing import Any, ClassVar, Dict, List
+from acceldata.models.api.pipeline.span import Span
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class SpansResponse(BaseModel):
+    """
+    SpansResponse
+    """ # noqa: E501
+    spans: List[Span]
+    __properties: ClassVar[List[str]] = ["spans"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of SpansResponse from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in spans (list)
+        _items = []
+        if self.spans:
+            for _item_spans in self.spans:
+                if _item_spans:
+                    _items.append(_item_spans.to_dict())
+            _dict['spans'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SpansResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "spans": [Span.from_dict(_item) for _item in obj["spans"]] if obj.get("spans") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/tag.py

@@ -0,0 +1,97 @@
+# coding: utf-8
+
+"""
+    Pipeline Control Plane API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 1.0
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class Tag(BaseModel):
+    """
+    Tag
+    """ # noqa: E501
+    created_at: Optional[datetime] = Field(default=None, alias="createdAt")
+    display_name: StrictStr = Field(alias="displayName")
+    id: Optional[StrictInt] = None
+    name: StrictStr
+    updated_at: Optional[datetime] = Field(default=None, alias="updatedAt")
+    __properties: ClassVar[List[str]] = ["createdAt", "displayName", "id", "name", "updatedAt"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of Tag from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of Tag from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "createdAt": obj.get("createdAt"),
+            "displayName": obj.get("displayName"),
+            "id": obj.get("id"),
+            "name": obj.get("name"),
+            "updatedAt": obj.get("updatedAt")
+        })
+        return _obj
+
+

acceldata/models/sdk/catalog/__init__.py

@@ -0,0 +1,105 @@
+from acceldata.models.sdk.catalog.asset_source_type import AssetSourceType
+from acceldata.models.sdk.catalog.policy_execution_type import PolicyExecutionType
+__all__ = [
+    "AssetSourceType",
+    "PolicyExecutionRequest",
+    "PolicyExecutionRequestInput",
+    "PolicyExecutionType",
+    "AssetResource",
+    "ProfileRequestResource",
+    "CustomAssetMetadata",
+    "ExecutionType",
+    "ProfilingType",
+    "AssetProfilingParams",
+    "PolicyFilter",
+    "PolicyType",
+    "RuleType",
+    "ExecutionPeriod",
+    "RuleResultStatus",
+    "RuleExecutionStatus",
+    "DataQualityRuleResource",
+    "ReconciliationRuleResource",
+    "DataCadenceRuleResource",
+    "DQPolicyResource",
+    "ReconPolicyResource",
+    "CadencePolicyResource",
+    "ReconRuleResource",
+]
+
+
+def __getattr__(name: str):
+    if name == "PolicyExecutionRequest":
+        from acceldata.models.sdk.catalog.policy_execution_request import (
+            PolicyExecutionInput,
+        )
+
+        return PolicyExecutionInput
+    if name == "PolicyExecutionRequestInput":
+        from acceldata.models.sdk.catalog.policy_execution_request import (
+            PolicyExecutionRequestInput,
+        )
+
+        return PolicyExecutionRequestInput
+    if name == "AssetResource":
+        from acceldata.models.sdk.catalog.asset_resource import AssetResource
+
+        return AssetResource
+    if name == "ProfileRequestResource":
+        from acceldata.models.sdk.catalog.profile_request_resource import (
+            ProfileRequestResource,
+        )
+
+        return ProfileRequestResource
+    if name == "CustomAssetMetadata":
+        from acceldata.models.sdk.catalog.custom_asset_metadata import CustomAssetMetadata
+
+        return CustomAssetMetadata
+    if name == "ExecutionType":
+        from acceldata.models.sdk.catalog.profiling import ExecutionType
+
+        return ExecutionType
+    if name == "ProfilingType":
+        from acceldata.models.sdk.catalog.profiling import ProfilingType
+
+        return ProfilingType
+    if name == "AssetProfilingParams":
+        from acceldata.models.sdk.catalog.profiling import AssetProfilingParams
+
+        return AssetProfilingParams
+    if name == "PolicyFilter":
+        from acceldata.models.sdk.catalog.policy_filter import PolicyFilter
+
+        return PolicyFilter
+    if name == "PolicyType":
+        from acceldata.models.sdk.catalog.policy_filter import PolicyType
+
+        return PolicyType
+    if name == "RuleType":
+        from acceldata.models.sdk.catalog.policy_filter import RuleType
+
+        return RuleType
+    if name == "ExecutionPeriod":
+        from acceldata.models.sdk.catalog.policy_filter import ExecutionPeriod
+
+        return ExecutionPeriod
+    if name == "RuleResultStatus":
+        from acceldata.models.sdk.catalog.policy_filter import RuleResultStatus
+
+        return RuleResultStatus
+    if name == "RuleExecutionStatus":
+        from acceldata.models.sdk.catalog.policy_filter import RuleExecutionStatus
+
+        return RuleExecutionStatus
+    if name in (
+        "DataQualityRuleResource",
+        "ReconciliationRuleResource",
+        "DataCadenceRuleResource",
+        "DQPolicyResource",
+        "ReconPolicyResource",
+        "CadencePolicyResource",
+        "ReconRuleResource",
+    ):
+        from acceldata.models.sdk.catalog import policy_rule_resources as _pr
+
+        return getattr(_pr, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")