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

acceldata/models/api/pipeline/pipeline_details.py

@@ -0,0 +1,118 @@
+# 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
+from typing import Any, ClassVar, Dict, List, Optional
+from acceldata.models.api.pipeline.edge import Edge
+from acceldata.models.api.pipeline.node import Node
+from acceldata.models.api.pipeline.pipeline import Pipeline
+from acceldata.models.api.pipeline.pipeline_run import PipelineRun
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PipelineDetails(BaseModel):
+    """
+    PipelineDetails
+    """ # noqa: E501
+    edges: List[Edge]
+    nodes: List[Node]
+    pipeline: Pipeline
+    pipeline_run: Optional[PipelineRun] = Field(default=None, alias="pipelineRun")
+    __properties: ClassVar[List[str]] = ["edges", "nodes", "pipeline", "pipelineRun"]
+
+    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 PipelineDetails 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 edges (list)
+        _items = []
+        if self.edges:
+            for _item_edges in self.edges:
+                if _item_edges:
+                    _items.append(_item_edges.to_dict())
+            _dict['edges'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in nodes (list)
+        _items = []
+        if self.nodes:
+            for _item_nodes in self.nodes:
+                if _item_nodes:
+                    _items.append(_item_nodes.to_dict())
+            _dict['nodes'] = _items
+        # override the default output from pydantic by calling `to_dict()` of pipeline
+        if self.pipeline:
+            _dict['pipeline'] = self.pipeline.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of pipeline_run
+        if self.pipeline_run:
+            _dict['pipelineRun'] = self.pipeline_run.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PipelineDetails from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "edges": [Edge.from_dict(_item) for _item in obj["edges"]] if obj.get("edges") is not None else None,
+            "nodes": [Node.from_dict(_item) for _item in obj["nodes"]] if obj.get("nodes") is not None else None,
+            "pipeline": Pipeline.from_dict(obj["pipeline"]) if obj.get("pipeline") is not None else None,
+            "pipelineRun": PipelineRun.from_dict(obj["pipelineRun"]) if obj.get("pipelineRun") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/pipeline/pipeline_details_response.py

@@ -0,0 +1,102 @@
+# 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.metric import Metric
+from acceldata.models.api.pipeline.pipeline_details import PipelineDetails
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PipelineDetailsResponse(BaseModel):
+    """
+    PipelineDetailsResponse
+    """ # noqa: E501
+    details: PipelineDetails
+    metrics: List[Metric]
+    __properties: ClassVar[List[str]] = ["details", "metrics"]
+
+    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 PipelineDetailsResponse 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 details
+        if self.details:
+            _dict['details'] = self.details.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in metrics (list)
+        _items = []
+        if self.metrics:
+            for _item_metrics in self.metrics:
+                if _item_metrics:
+                    _items.append(_item_metrics.to_dict())
+            _dict['metrics'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PipelineDetailsResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "details": PipelineDetails.from_dict(obj["details"]) if obj.get("details") is not None else None,
+            "metrics": [Metric.from_dict(_item) for _item in obj["metrics"]] if obj.get("metrics") is not None else None
+        })
+        return _obj
+
+

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

acceldata/models/api/pipeline/pipeline_listing_info.py

@@ -0,0 +1,118 @@
+# 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.pipeline_summary import PipelineSummary
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PipelineListingInfo(BaseModel):
+    """
+    PipelineListingInfo
+    """ # noqa: E501
+    asset_nodes_count: StrictInt = Field(alias="assetNodesCount")
+    functional_nodes_count: StrictInt = Field(alias="functionalNodesCount")
+    latest_run_finished_at: Optional[datetime] = Field(default=None, alias="latestRunFinishedAt")
+    latest_run_id: Optional[StrictInt] = Field(default=None, alias="latestRunId")
+    latest_run_result: Optional[StrictStr] = Field(default=None, alias="latestRunResult")
+    latest_run_started_at: Optional[datetime] = Field(default=None, alias="latestRunStartedAt")
+    latest_run_version_id: Optional[StrictInt] = Field(default=None, alias="latestRunVersionId")
+    pipeline_summary: PipelineSummary = Field(alias="pipelineSummary")
+    policy_counts: Dict[str, StrictInt] = Field(alias="policyCounts")
+    source_type: StrictStr = Field(description="Allowed values at generation time: AIRFLOW, AZURE_DATA_FACTORY, TRINO, OTHERS, AUTOSYS, SPARK, DBT, FIVETRAN, SNAPLOGIC.", alias="sourceType")
+    total_runs_count: StrictInt = Field(alias="totalRunsCount")
+    __properties: ClassVar[List[str]] = ["assetNodesCount", "functionalNodesCount", "latestRunFinishedAt", "latestRunId", "latestRunResult", "latestRunStartedAt", "latestRunVersionId", "pipelineSummary", "policyCounts", "sourceType", "totalRunsCount"]
+
+    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 PipelineListingInfo 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 pipeline_summary
+        if self.pipeline_summary:
+            _dict['pipelineSummary'] = self.pipeline_summary.to_dict()
+        # set to None if latest_run_result (nullable) is None
+        # and model_fields_set contains the field
+        if self.latest_run_result is None and "latest_run_result" in self.model_fields_set:
+            _dict['latestRunResult'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PipelineListingInfo from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "assetNodesCount": obj.get("assetNodesCount") if obj.get("assetNodesCount") is not None else 0,
+            "functionalNodesCount": obj.get("functionalNodesCount") if obj.get("functionalNodesCount") is not None else 0,
+            "latestRunFinishedAt": obj.get("latestRunFinishedAt"),
+            "latestRunId": obj.get("latestRunId"),
+            "latestRunResult": obj.get("latestRunResult"),
+            "latestRunStartedAt": obj.get("latestRunStartedAt"),
+            "latestRunVersionId": obj.get("latestRunVersionId"),
+            "pipelineSummary": PipelineSummary.from_dict(obj["pipelineSummary"]) if obj.get("pipelineSummary") is not None else None,
+            "policyCounts": obj.get("policyCounts"),
+            "sourceType": obj.get("sourceType"),
+            "totalRunsCount": obj.get("totalRunsCount") if obj.get("totalRunsCount") is not None else 0
+        })
+        return _obj
+
+

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

acceldata/models/api/pipeline/pipeline_run.py

@@ -0,0 +1,136 @@
+# 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.big_decimal import BigDecimal
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PipelineRun(BaseModel):
+    """
+    PipelineRun
+    """ # noqa: E501
+    args: Optional[Dict[str, Any]] = None
+    avg_execution_time: Optional[BigDecimal] = Field(default=None, alias="avgExecutionTime")
+    continuation_id: Optional[StrictStr] = Field(default=None, alias="continuationId")
+    error_events: StrictInt = Field(alias="errorEvents")
+    external_run_id: Optional[StrictStr] = Field(default=None, alias="externalRunId")
+    finished_at: Optional[datetime] = Field(default=None, alias="finishedAt")
+    id: Optional[StrictInt] = None
+    pipeline_id: Optional[StrictInt] = Field(default=None, alias="pipelineId")
+    result: StrictStr = Field(description="Allowed values at generation time: RUNNING, SUCCESS, FAILURE, CANCELLED.")
+    started_at: Optional[datetime] = Field(default=None, alias="startedAt")
+    status: StrictStr = Field(description="Allowed values at generation time: STARTED, COMPLETED, FAILED, ABORTED.")
+    success_events: StrictInt = Field(alias="successEvents")
+    tenant_id: Optional[StrictStr] = Field(default=None, alias="tenantId")
+    version_id: Optional[StrictInt] = Field(default=None, alias="versionId")
+    warning_events: StrictInt = Field(alias="warningEvents")
+    __properties: ClassVar[List[str]] = ["args", "avgExecutionTime", "continuationId", "errorEvents", "externalRunId", "finishedAt", "id", "pipelineId", "result", "startedAt", "status", "successEvents", "tenantId", "versionId", "warningEvents"]
+
+    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 PipelineRun 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 avg_execution_time
+        if self.avg_execution_time:
+            _dict['avgExecutionTime'] = self.avg_execution_time.to_dict()
+        # set to None if continuation_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.continuation_id is None and "continuation_id" in self.model_fields_set:
+            _dict['continuationId'] = None
+
+        # set to None if external_run_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.external_run_id is None and "external_run_id" in self.model_fields_set:
+            _dict['externalRunId'] = None
+
+        # set to None if tenant_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.tenant_id is None and "tenant_id" in self.model_fields_set:
+            _dict['tenantId'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PipelineRun from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "args": obj.get("args"),
+            "avgExecutionTime": BigDecimal.from_dict(obj["avgExecutionTime"]) if obj.get("avgExecutionTime") is not None else None,
+            "continuationId": obj.get("continuationId"),
+            "errorEvents": obj.get("errorEvents"),
+            "externalRunId": obj.get("externalRunId"),
+            "finishedAt": obj.get("finishedAt"),
+            "id": obj.get("id"),
+            "pipelineId": obj.get("pipelineId"),
+            "result": obj.get("result"),
+            "startedAt": obj.get("startedAt"),
+            "status": obj.get("status"),
+            "successEvents": obj.get("successEvents"),
+            "tenantId": obj.get("tenantId"),
+            "versionId": obj.get("versionId"),
+            "warningEvents": obj.get("warningEvents")
+        })
+        return _obj
+
+