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

acceldata/models/api/catalog/drift.py

@@ -0,0 +1,121 @@
+# 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.asset_type import AssetType
+from acceldata.models.api.catalog.snapshot_data import SnapshotData
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class Drift(BaseModel):
+    """
+    Drift
+    """ # noqa: E501
+    added_count: Optional[StrictInt] = Field(default=None, alias="addedCount")
+    added_names: List[StrictStr] = Field(alias="addedNames")
+    backing_asset_id: Optional[StrictInt] = Field(default=None, alias="backingAssetId")
+    child_asset_types: List[AssetType] = Field(alias="childAssetTypes")
+    deleted_count: Optional[StrictInt] = Field(default=None, alias="deletedCount")
+    deleted_names: List[StrictStr] = Field(alias="deletedNames")
+    left_snapshot_data: Optional[SnapshotData] = Field(default=None, alias="leftSnapshotData")
+    modified_count: Optional[StrictInt] = Field(default=None, alias="modifiedCount")
+    modified_names: List[StrictStr] = Field(alias="modifiedNames")
+    right_snapshot_data: Optional[SnapshotData] = Field(default=None, alias="rightSnapshotData")
+    __properties: ClassVar[List[str]] = ["addedCount", "addedNames", "backingAssetId", "childAssetTypes", "deletedCount", "deletedNames", "leftSnapshotData", "modifiedCount", "modifiedNames", "rightSnapshotData"]
+
+    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 Drift 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 child_asset_types (list)
+        _items = []
+        if self.child_asset_types:
+            for _item_child_asset_types in self.child_asset_types:
+                if _item_child_asset_types:
+                    _items.append(_item_child_asset_types.to_dict())
+            _dict['childAssetTypes'] = _items
+        # override the default output from pydantic by calling `to_dict()` of left_snapshot_data
+        if self.left_snapshot_data:
+            _dict['leftSnapshotData'] = self.left_snapshot_data.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of right_snapshot_data
+        if self.right_snapshot_data:
+            _dict['rightSnapshotData'] = self.right_snapshot_data.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of Drift from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "addedCount": obj.get("addedCount"),
+            "addedNames": obj.get("addedNames"),
+            "backingAssetId": obj.get("backingAssetId"),
+            "childAssetTypes": [AssetType.from_dict(_item) for _item in obj["childAssetTypes"]] if obj.get("childAssetTypes") is not None else None,
+            "deletedCount": obj.get("deletedCount"),
+            "deletedNames": obj.get("deletedNames"),
+            "leftSnapshotData": SnapshotData.from_dict(obj["leftSnapshotData"]) if obj.get("leftSnapshotData") is not None else None,
+            "modifiedCount": obj.get("modifiedCount"),
+            "modifiedNames": obj.get("modifiedNames"),
+            "rightSnapshotData": SnapshotData.from_dict(obj["rightSnapshotData"]) if obj.get("rightSnapshotData") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/catalog/enable_rule_request.py

@@ -0,0 +1,88 @@
+# 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, StrictBool
+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 EnableRuleRequest(BaseModel):
+    """
+    EnableRuleRequest
+    """ # noqa: E501
+    enable: StrictBool
+    __properties: ClassVar[List[str]] = ["enable"]
+
+    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 EnableRuleRequest 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 EnableRuleRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "enable": obj.get("enable")
+        })
+        return _obj
+
+

acceldata/models/api/catalog/engine.py

@@ -0,0 +1,90 @@
+# 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 typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class Engine(BaseModel):
+    """
+    Engine
+    """ # noqa: E501
+    config: Optional[Dict[str, Any]] = None
+    type: StrictStr = Field(description="Allowed values at generation time: SPARK_ASYNC, POD_ASYNC, STANDALONE_ASYNC, PUSHDOWN_ASYNC.")
+    __properties: ClassVar[List[str]] = ["config", "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 Engine 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 Engine from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "config": obj.get("config"),
+            "type": obj.get("type")
+        })
+        return _obj
+
+

acceldata/models/api/catalog/execution_cancellation_response.py

@@ -0,0 +1,92 @@
+# 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
+from typing import Any, ClassVar, Dict, List
+from acceldata.models.api.catalog.connection_test_result import ConnectionTestResult
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class ExecutionCancellationResponse(BaseModel):
+    """
+    ExecutionCancellationResponse
+    """ # noqa: E501
+    result: ConnectionTestResult
+    __properties: ClassVar[List[str]] = ["result"]
+
+    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 ExecutionCancellationResponse 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 result
+        if self.result:
+            _dict['result'] = self.result.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ExecutionCancellationResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "result": ConnectionTestResult.from_dict(obj["result"]) if obj.get("result") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/catalog/execution_response.py

@@ -0,0 +1,144 @@
+# 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.rule_execution import RuleExecution
+from acceldata.models.api.catalog.rule_execution_result import RuleExecutionResult
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class ExecutionResponse(BaseModel):
+    """
+    ExecutionResponse
+    """ # noqa: E501
+    execution: RuleExecution
+    left_backing_asset_id: Optional[StrictInt] = Field(default=None, alias="leftBackingAssetId")
+    left_backing_asset_uid: Optional[StrictStr] = Field(default=None, alias="leftBackingAssetUid")
+    left_data_source: Optional[StrictStr] = Field(default=None, alias="leftDataSource")
+    left_data_source_type: Optional[StrictStr] = Field(default=None, alias="leftDataSourceType")
+    result: Optional[RuleExecutionResult] = None
+    right_backing_asset_id: Optional[StrictInt] = Field(default=None, alias="rightBackingAssetId")
+    right_backing_asset_uid: Optional[StrictStr] = Field(default=None, alias="rightBackingAssetUid")
+    right_data_source: Optional[StrictStr] = Field(default=None, alias="rightDataSource")
+    right_data_source_type: Optional[StrictStr] = Field(default=None, alias="rightDataSourceType")
+    __properties: ClassVar[List[str]] = ["execution", "leftBackingAssetId", "leftBackingAssetUid", "leftDataSource", "leftDataSourceType", "result", "rightBackingAssetId", "rightBackingAssetUid", "rightDataSource", "rightDataSourceType"]
+
+    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 ExecutionResponse 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 execution
+        if self.execution:
+            _dict['execution'] = self.execution.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of result
+        if self.result:
+            _dict['result'] = self.result.to_dict()
+        # set to None if left_backing_asset_uid (nullable) is None
+        # and model_fields_set contains the field
+        if self.left_backing_asset_uid is None and "left_backing_asset_uid" in self.model_fields_set:
+            _dict['leftBackingAssetUid'] = None
+
+        # set to None if left_data_source (nullable) is None
+        # and model_fields_set contains the field
+        if self.left_data_source is None and "left_data_source" in self.model_fields_set:
+            _dict['leftDataSource'] = None
+
+        # set to None if left_data_source_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.left_data_source_type is None and "left_data_source_type" in self.model_fields_set:
+            _dict['leftDataSourceType'] = None
+
+        # set to None if right_backing_asset_uid (nullable) is None
+        # and model_fields_set contains the field
+        if self.right_backing_asset_uid is None and "right_backing_asset_uid" in self.model_fields_set:
+            _dict['rightBackingAssetUid'] = None
+
+        # set to None if right_data_source (nullable) is None
+        # and model_fields_set contains the field
+        if self.right_data_source is None and "right_data_source" in self.model_fields_set:
+            _dict['rightDataSource'] = None
+
+        # set to None if right_data_source_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.right_data_source_type is None and "right_data_source_type" in self.model_fields_set:
+            _dict['rightDataSourceType'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ExecutionResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "execution": RuleExecution.from_dict(obj["execution"]) if obj.get("execution") is not None else None,
+            "leftBackingAssetId": obj.get("leftBackingAssetId"),
+            "leftBackingAssetUid": obj.get("leftBackingAssetUid"),
+            "leftDataSource": obj.get("leftDataSource"),
+            "leftDataSourceType": obj.get("leftDataSourceType"),
+            "result": RuleExecutionResult.from_dict(obj["result"]) if obj.get("result") is not None else None,
+            "rightBackingAssetId": obj.get("rightBackingAssetId"),
+            "rightBackingAssetUid": obj.get("rightBackingAssetUid"),
+            "rightDataSource": obj.get("rightDataSource"),
+            "rightDataSourceType": obj.get("rightDataSourceType")
+        })
+        return _obj
+
+

acceldata/models/api/catalog/execution_result.py

@@ -0,0 +1,126 @@
+# 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
+from typing import Any, ClassVar, Dict, List, Optional
+from acceldata.models.api.catalog.generic_rule_execution import GenericRuleExecution
+from acceldata.models.api.catalog.rule_execution_result import RuleExecutionResult
+from acceldata.models.api.catalog.rule_item_result import RuleItemResult
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class ExecutionResult(BaseModel):
+    """
+    ExecutionResult
+    """ # noqa: E501
+    child_executions: Optional[List[ExecutionResult]] = Field(default=None, alias="childExecutions")
+    execution: GenericRuleExecution
+    items: List[RuleItemResult]
+    meta: Dict[str, Any]
+    result: RuleExecutionResult
+    __properties: ClassVar[List[str]] = ["childExecutions", "execution", "items", "meta", "result"]
+
+    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 ExecutionResult 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 child_executions (list)
+        _items = []
+        if self.child_executions:
+            for _item_child_executions in self.child_executions:
+                if _item_child_executions:
+                    _items.append(_item_child_executions.to_dict())
+            _dict['childExecutions'] = _items
+        # override the default output from pydantic by calling `to_dict()` of execution
+        if self.execution:
+            _dict['execution'] = self.execution.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in items (list)
+        _items = []
+        if self.items:
+            for _item_items in self.items:
+                if _item_items:
+                    _items.append(_item_items.to_dict())
+            _dict['items'] = _items
+        # override the default output from pydantic by calling `to_dict()` of result
+        if self.result:
+            _dict['result'] = self.result.to_dict()
+        # set to None if child_executions (nullable) is None
+        # and model_fields_set contains the field
+        if self.child_executions is None and "child_executions" in self.model_fields_set:
+            _dict['childExecutions'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ExecutionResult from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "childExecutions": [ExecutionResult.from_dict(_item) for _item in obj["childExecutions"]] if obj.get("childExecutions") is not None else None,
+            "execution": GenericRuleExecution.from_dict(obj["execution"]) if obj.get("execution") is not None else None,
+            "items": [RuleItemResult.from_dict(_item) for _item in obj["items"]] if obj.get("items") is not None else None,
+            "meta": obj.get("meta"),
+            "result": RuleExecutionResult.from_dict(obj["result"]) if obj.get("result") is not None else None
+        })
+        return _obj
+
+# TODO: Rewrite to not use raise_errors
+ExecutionResult.model_rebuild(raise_errors=False)
+