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

acceldata/models/api/catalog/absolute_configs.py

@@ -0,0 +1,197 @@
+# coding: utf-8
+
+"""
+    Catalog 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 json
+import pprint
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
+from typing import Any, List, Optional
+from acceldata.models.api.catalog.above_configs import AboveConfigs
+from acceldata.models.api.catalog.absolute_configs_fallback import AbsoluteConfigsFallback
+from acceldata.models.api.catalog.below_configs import BelowConfigs
+from acceldata.models.api.catalog.range_configs import RangeConfigs
+from pydantic import StrictStr, Field
+from typing import Union, List, Set, Optional, Dict
+from typing_extensions import Literal, Self
+
+ABSOLUTECONFIGS_ONE_OF_SCHEMAS = ["AboveConfigs", "AbsoluteConfigsFallback", "BelowConfigs", "RangeConfigs"]
+
+class AbsoluteConfigs(BaseModel):
+    """
+    AbsoluteConfigs
+    """
+    # data type: AboveConfigs
+    oneof_schema_1_validator: Optional[AboveConfigs] = None
+    # data type: BelowConfigs
+    oneof_schema_2_validator: Optional[BelowConfigs] = None
+    # data type: RangeConfigs
+    oneof_schema_3_validator: Optional[RangeConfigs] = None
+    # data type: AbsoluteConfigsFallback
+    oneof_schema_4_validator: Optional[AbsoluteConfigsFallback] = None
+    actual_instance: Optional[Union[AboveConfigs, AbsoluteConfigsFallback, BelowConfigs, RangeConfigs]] = None
+    one_of_schemas: Set[str] = { "AboveConfigs", "AbsoluteConfigsFallback", "BelowConfigs", "RangeConfigs" }
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    discriminator_value_class_map: Dict[str, str] = {
+    }
+
+    def __init__(self, *args, **kwargs) -> None:
+        if args:
+            if len(args) > 1:
+                raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
+            if kwargs:
+                raise ValueError("If a position argument is used, keyword arguments cannot be used.")
+            super().__init__(actual_instance=args[0])
+        else:
+            super().__init__(**kwargs)
+
+    @field_validator('actual_instance')
+    def actual_instance_must_validate_oneof(cls, v):
+        instance = AbsoluteConfigs.model_construct()
+        error_messages = []
+        match = 0
+        # validate data type: AboveConfigs
+        if not isinstance(v, AboveConfigs):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AboveConfigs`")
+        else:
+            match += 1
+        # validate data type: BelowConfigs
+        if not isinstance(v, BelowConfigs):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `BelowConfigs`")
+        else:
+            match += 1
+        # validate data type: RangeConfigs
+        if not isinstance(v, RangeConfigs):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `RangeConfigs`")
+        else:
+            match += 1
+        # validate data type: AbsoluteConfigsFallback
+        if not isinstance(v, AbsoluteConfigsFallback):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AbsoluteConfigsFallback`")
+        else:
+            match += 1
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when setting `actual_instance` in AbsoluteConfigs with oneOf schemas: AboveConfigs, AbsoluteConfigsFallback, BelowConfigs, RangeConfigs. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when setting `actual_instance` in AbsoluteConfigs with oneOf schemas: AboveConfigs, AbsoluteConfigsFallback, BelowConfigs, RangeConfigs. Details: " + ", ".join(error_messages))
+        else:
+            return v
+
+    @classmethod
+    def from_dict(cls, obj: Union[str, Dict[str, Any]]) -> Self:
+        return cls.from_json(json.dumps(obj))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Returns the object represented by the json string"""
+        instance = cls.model_construct()
+        error_messages = []
+        match = 0
+
+        # use oneOf discriminator to lookup the data type
+        _data_type = json.loads(json_str).get("direction")
+        if not _data_type:
+            raise ValueError("Failed to lookup data type from the field `direction` in the input.")
+
+        # check if data type is `AboveConfigs`
+        if _data_type == "ABOVE":
+            instance.actual_instance = AboveConfigs.from_json(json_str)
+            return instance
+
+        # check if data type is `AbsoluteConfigsFallback`
+        if _data_type == "AbsoluteConfigsFallback":
+            instance.actual_instance = AbsoluteConfigsFallback.from_json(json_str)
+            return instance
+
+        # check if data type is `BelowConfigs`
+        if _data_type == "BELOW":
+            instance.actual_instance = BelowConfigs.from_json(json_str)
+            return instance
+
+        # check if data type is `RangeConfigs`
+        if _data_type == "RANGE":
+            instance.actual_instance = RangeConfigs.from_json(json_str)
+            return instance
+
+        # Unknown discriminator value – forward-compatible fallback
+        instance.actual_instance = AbsoluteConfigsFallback.from_json(json_str)
+        return instance
+
+        # deserialize data into AboveConfigs
+        try:
+            instance.actual_instance = AboveConfigs.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into BelowConfigs
+        try:
+            instance.actual_instance = BelowConfigs.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into RangeConfigs
+        try:
+            instance.actual_instance = RangeConfigs.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into AbsoluteConfigsFallback
+        try:
+            instance.actual_instance = AbsoluteConfigsFallback.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when deserializing the JSON string into AbsoluteConfigs with oneOf schemas: AboveConfigs, AbsoluteConfigsFallback, BelowConfigs, RangeConfigs. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when deserializing the JSON string into AbsoluteConfigs with oneOf schemas: AboveConfigs, AbsoluteConfigsFallback, BelowConfigs, RangeConfigs. Details: " + ", ".join(error_messages))
+        else:
+            return instance
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the actual instance"""
+        if self.actual_instance is None:
+            return "null"
+
+        if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
+            return self.actual_instance.to_json()
+        else:
+            return json.dumps(self.actual_instance)
+
+    def to_dict(self) -> Optional[Union[Dict[str, Any], AboveConfigs, AbsoluteConfigsFallback, BelowConfigs, RangeConfigs]]:
+        """Returns the dict representation of the actual instance"""
+        if self.actual_instance is None:
+            return None
+
+        if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
+            return self.actual_instance.to_dict()
+        else:
+            # primitive type
+            return self.actual_instance
+
+    def to_str(self) -> str:
+        """Returns the string representation of the actual instance"""
+        return pprint.pformat(self.model_dump())
+
+

acceldata/models/api/catalog/absolute_configs_fallback.py

@@ -0,0 +1,101 @@
+# coding: utf-8
+
+"""
+    Catalog 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, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class AbsoluteConfigsFallback(BaseModel):
+    """
+    Forward-compatible payload for `AbsoluteConfigs` when `direction` is not listed in this spec's discriminator mapping (new backend subtype).
+    """ # noqa: E501
+    direction: StrictStr
+    additional_properties: Dict[str, Any] = {}
+    __properties: ClassVar[List[str]] = ["direction"]
+
+    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 AbsoluteConfigsFallback 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.
+        * Fields in `self.additional_properties` are added to the output dict.
+        """
+        excluded_fields: Set[str] = set([
+            "additional_properties",
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # puts key-value pairs in additional_properties in the top level
+        if self.additional_properties is not None:
+            for _key, _value in self.additional_properties.items():
+                _dict[_key] = _value
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AbsoluteConfigsFallback from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "direction": obj.get("direction")
+        })
+        # store additional fields in additional_properties
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                _obj.additional_properties[_key] = obj.get(_key)
+
+        return _obj
+
+

acceldata/models/api/catalog/absolute_threshold_config.py

@@ -0,0 +1,105 @@
+# coding: utf-8
+
+"""
+    Catalog 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, Optional
+from acceldata.models.api.catalog.above_configs_success import AboveConfigsSuccess
+from acceldata.models.api.catalog.absolute_configs import AbsoluteConfigs
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class AbsoluteThresholdConfig(BaseModel):
+    """
+    AbsoluteThresholdConfig
+    """ # noqa: E501
+    config: AbsoluteConfigs
+    success: Optional[AboveConfigsSuccess] = None
+    type: StrictStr = Field(description="Allowed values at generation time: ANOMALY, RELATIVE, TIME_BASED, ABSOLUTE.")
+    warning: Optional[AboveConfigsSuccess] = None
+    __properties: ClassVar[List[str]] = ["config", "success", "type", "warning"]
+
+    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 AbsoluteThresholdConfig 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 config
+        if self.config:
+            _dict['config'] = self.config.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of success
+        if self.success:
+            _dict['success'] = self.success.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of warning
+        if self.warning:
+            _dict['warning'] = self.warning.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AbsoluteThresholdConfig from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "config": AbsoluteConfigs.from_dict(obj["config"]) if obj.get("config") is not None else None,
+            "success": AboveConfigsSuccess.from_dict(obj["success"]) if obj.get("success") is not None else None,
+            "type": obj.get("type"),
+            "warning": AboveConfigsSuccess.from_dict(obj["warning"]) if obj.get("warning") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/catalog/analysis_job_context.py

@@ -0,0 +1,102 @@
+# coding: utf-8
+
+"""
+    Catalog 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, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from acceldata.models.api.catalog.dataplane_job_status import DataplaneJobStatus
+from acceldata.models.api.catalog.spark_job_status import SparkJobStatus
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class AnalysisJobContext(BaseModel):
+    """
+    AnalysisJobContext
+    """ # noqa: E501
+    dataplane_job_status: Optional[DataplaneJobStatus] = Field(default=None, alias="dataplaneJobStatus")
+    spark_job_status: Optional[SparkJobStatus] = Field(default=None, alias="sparkJobStatus")
+    status_retry_count: Optional[StrictInt] = Field(default=None, alias="statusRetryCount")
+    type: StrictStr
+    __properties: ClassVar[List[str]] = ["dataplaneJobStatus", "sparkJobStatus", "statusRetryCount", "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 AnalysisJobContext 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 dataplane_job_status
+        if self.dataplane_job_status:
+            _dict['dataplaneJobStatus'] = self.dataplane_job_status.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of spark_job_status
+        if self.spark_job_status:
+            _dict['sparkJobStatus'] = self.spark_job_status.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of AnalysisJobContext from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "dataplaneJobStatus": DataplaneJobStatus.from_dict(obj["dataplaneJobStatus"]) if obj.get("dataplaneJobStatus") is not None else None,
+            "sparkJobStatus": SparkJobStatus.from_dict(obj["sparkJobStatus"]) if obj.get("sparkJobStatus") is not None else None,
+            "statusRetryCount": obj.get("statusRetryCount"),
+            "type": obj.get("type")
+        })
+        return _obj
+
+