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

acceldata/models/api/catalog/pipeline_compute_config.py

@@ -0,0 +1,110 @@
+# 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, Optional
+from acceldata.models.api.catalog.databricks_compute_config import DatabricksComputeConfig
+from acceldata.models.api.catalog.dataproc_compute_config import DataprocComputeConfig
+from acceldata.models.api.catalog.livy_compute_config import LivyComputeConfig
+from acceldata.models.api.catalog.yunikorn_compute_config import YunikornComputeConfig
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PipelineComputeConfig(BaseModel):
+    """
+    PipelineComputeConfig
+    """ # noqa: E501
+    databricks: Optional[DatabricksComputeConfig] = None
+    dataproc: Optional[DataprocComputeConfig] = None
+    livy: Optional[LivyComputeConfig] = None
+    yunikorn: Optional[YunikornComputeConfig] = None
+    __properties: ClassVar[List[str]] = ["databricks", "dataproc", "livy", "yunikorn"]
+
+    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 PipelineComputeConfig 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 databricks
+        if self.databricks:
+            _dict['databricks'] = self.databricks.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of dataproc
+        if self.dataproc:
+            _dict['dataproc'] = self.dataproc.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of livy
+        if self.livy:
+            _dict['livy'] = self.livy.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of yunikorn
+        if self.yunikorn:
+            _dict['yunikorn'] = self.yunikorn.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PipelineComputeConfig from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "databricks": DatabricksComputeConfig.from_dict(obj["databricks"]) if obj.get("databricks") is not None else None,
+            "dataproc": DataprocComputeConfig.from_dict(obj["dataproc"]) if obj.get("dataproc") is not None else None,
+            "livy": LivyComputeConfig.from_dict(obj["livy"]) if obj.get("livy") is not None else None,
+            "yunikorn": YunikornComputeConfig.from_dict(obj["yunikorn"]) if obj.get("yunikorn") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/catalog/policy_execution_marker_config.py

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

acceldata/models/api/catalog/policy_execution_request.py

@@ -0,0 +1,229 @@
+# 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, StrictBool, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from acceldata.models.api.catalog.column_variable import ColumnVariable
+from acceldata.models.api.catalog.policy_execution_marker_config import PolicyExecutionMarkerConfig
+from acceldata.models.api.catalog.rule_spark_sql_dynamic_filter_variable_mapping import RuleSparkSQLDynamicFilterVariableMapping
+from acceldata.models.api.catalog.spark_resource_config import SparkResourceConfig
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PolicyExecutionRequest(BaseModel):
+    """
+    PolicyExecutionRequest
+    """ # noqa: E501
+    analysis_job_request_id: Optional[StrictStr] = Field(default=None, alias="analysisJobRequestId")
+    analytics_pipeline_id: Optional[StrictInt] = Field(default=None, alias="analyticsPipelineId")
+    asset_direction: Optional[StrictStr] = Field(default=None, alias="assetDirection")
+    auto_retry_enabled: Optional[StrictBool] = Field(default=False, alias="autoRetryEnabled")
+    column_variables: List[ColumnVariable] = Field(alias="columnVariables")
+    engine_type: Optional[StrictStr] = Field(default='SPARK', description="Allowed values at generation time: SPARK, JDBC_SQL, POD.", alias="engineType")
+    execution_type: StrictStr = Field(description="Allowed values at generation time: SELECTIVE, FULL, INCREMENTAL.", alias="executionType")
+    include_in_quality_score: StrictBool = Field(alias="includeInQualityScore")
+    marker_configs: Optional[List[PolicyExecutionMarkerConfig]] = Field(default=None, alias="markerConfigs")
+    monitor_execution_id: Optional[StrictInt] = Field(default=None, alias="monitorExecutionId")
+    pipeline_automation_execution_id: Optional[StrictInt] = Field(default=None, alias="pipelineAutomationExecutionId")
+    pipeline_run_id: Optional[StrictInt] = Field(default=None, alias="pipelineRunId")
+    profile_id: Optional[StrictInt] = Field(default=None, alias="profileId")
+    resource_strategy_type: StrictStr = Field(description="Allowed values at generation time: INVENTORY, CUSTOM.", alias="resourceStrategyType")
+    right_engine_type: Optional[StrictStr] = Field(default='SPARK', description="Allowed values at generation time: SPARK, JDBC_SQL, POD.", alias="rightEngineType")
+    right_spark_filter_selected_columns: Optional[List[StrictStr]] = Field(default=None, alias="rightSparkFilterSelectedColumns")
+    right_spark_sql_dynamic_filter_variable_mapping: Optional[List[RuleSparkSQLDynamicFilterVariableMapping]] = Field(default=None, alias="rightSparkSQLDynamicFilterVariableMapping")
+    rule_item_selections: Optional[List[StrictInt]] = Field(default=None, alias="ruleItemSelections")
+    selected_resource_inventory: Optional[StrictStr] = Field(default=None, alias="selectedResourceInventory")
+    source_execution_id: Optional[StrictInt] = Field(default=None, alias="sourceExecutionId")
+    spark_app_name: Optional[StrictStr] = Field(default=None, alias="sparkAppName")
+    spark_filter_selected_columns: Optional[List[StrictStr]] = Field(default=None, alias="sparkFilterSelectedColumns")
+    spark_resource_config: Optional[SparkResourceConfig] = Field(default=None, alias="sparkResourceConfig")
+    spark_sql_dynamic_filter_variable_mapping: Optional[List[RuleSparkSQLDynamicFilterVariableMapping]] = Field(default=None, alias="sparkSQLDynamicFilterVariableMapping")
+    __properties: ClassVar[List[str]] = ["analysisJobRequestId", "analyticsPipelineId", "assetDirection", "autoRetryEnabled", "columnVariables", "engineType", "executionType", "includeInQualityScore", "markerConfigs", "monitorExecutionId", "pipelineAutomationExecutionId", "pipelineRunId", "profileId", "resourceStrategyType", "rightEngineType", "rightSparkFilterSelectedColumns", "rightSparkSQLDynamicFilterVariableMapping", "ruleItemSelections", "selectedResourceInventory", "sourceExecutionId", "sparkAppName", "sparkFilterSelectedColumns", "sparkResourceConfig", "sparkSQLDynamicFilterVariableMapping"]
+
+    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 PolicyExecutionRequest 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 column_variables (list)
+        _items = []
+        if self.column_variables:
+            for _item_column_variables in self.column_variables:
+                if _item_column_variables:
+                    _items.append(_item_column_variables.to_dict())
+            _dict['columnVariables'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in marker_configs (list)
+        _items = []
+        if self.marker_configs:
+            for _item_marker_configs in self.marker_configs:
+                if _item_marker_configs:
+                    _items.append(_item_marker_configs.to_dict())
+            _dict['markerConfigs'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in right_spark_sql_dynamic_filter_variable_mapping (list)
+        _items = []
+        if self.right_spark_sql_dynamic_filter_variable_mapping:
+            for _item_right_spark_sql_dynamic_filter_variable_mapping in self.right_spark_sql_dynamic_filter_variable_mapping:
+                if _item_right_spark_sql_dynamic_filter_variable_mapping:
+                    _items.append(_item_right_spark_sql_dynamic_filter_variable_mapping.to_dict())
+            _dict['rightSparkSQLDynamicFilterVariableMapping'] = _items
+        # override the default output from pydantic by calling `to_dict()` of spark_resource_config
+        if self.spark_resource_config:
+            _dict['sparkResourceConfig'] = self.spark_resource_config.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in spark_sql_dynamic_filter_variable_mapping (list)
+        _items = []
+        if self.spark_sql_dynamic_filter_variable_mapping:
+            for _item_spark_sql_dynamic_filter_variable_mapping in self.spark_sql_dynamic_filter_variable_mapping:
+                if _item_spark_sql_dynamic_filter_variable_mapping:
+                    _items.append(_item_spark_sql_dynamic_filter_variable_mapping.to_dict())
+            _dict['sparkSQLDynamicFilterVariableMapping'] = _items
+        # set to None if analysis_job_request_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.analysis_job_request_id is None and "analysis_job_request_id" in self.model_fields_set:
+            _dict['analysisJobRequestId'] = None
+
+        # set to None if asset_direction (nullable) is None
+        # and model_fields_set contains the field
+        if self.asset_direction is None and "asset_direction" in self.model_fields_set:
+            _dict['assetDirection'] = None
+
+        # set to None if engine_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.engine_type is None and "engine_type" in self.model_fields_set:
+            _dict['engineType'] = None
+
+        # set to None if marker_configs (nullable) is None
+        # and model_fields_set contains the field
+        if self.marker_configs is None and "marker_configs" in self.model_fields_set:
+            _dict['markerConfigs'] = None
+
+        # set to None if right_engine_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.right_engine_type is None and "right_engine_type" in self.model_fields_set:
+            _dict['rightEngineType'] = None
+
+        # set to None if right_spark_filter_selected_columns (nullable) is None
+        # and model_fields_set contains the field
+        if self.right_spark_filter_selected_columns is None and "right_spark_filter_selected_columns" in self.model_fields_set:
+            _dict['rightSparkFilterSelectedColumns'] = None
+
+        # set to None if right_spark_sql_dynamic_filter_variable_mapping (nullable) is None
+        # and model_fields_set contains the field
+        if self.right_spark_sql_dynamic_filter_variable_mapping is None and "right_spark_sql_dynamic_filter_variable_mapping" in self.model_fields_set:
+            _dict['rightSparkSQLDynamicFilterVariableMapping'] = None
+
+        # set to None if rule_item_selections (nullable) is None
+        # and model_fields_set contains the field
+        if self.rule_item_selections is None and "rule_item_selections" in self.model_fields_set:
+            _dict['ruleItemSelections'] = None
+
+        # set to None if selected_resource_inventory (nullable) is None
+        # and model_fields_set contains the field
+        if self.selected_resource_inventory is None and "selected_resource_inventory" in self.model_fields_set:
+            _dict['selectedResourceInventory'] = None
+
+        # set to None if spark_app_name (nullable) is None
+        # and model_fields_set contains the field
+        if self.spark_app_name is None and "spark_app_name" in self.model_fields_set:
+            _dict['sparkAppName'] = None
+
+        # set to None if spark_filter_selected_columns (nullable) is None
+        # and model_fields_set contains the field
+        if self.spark_filter_selected_columns is None and "spark_filter_selected_columns" in self.model_fields_set:
+            _dict['sparkFilterSelectedColumns'] = None
+
+        # set to None if spark_sql_dynamic_filter_variable_mapping (nullable) is None
+        # and model_fields_set contains the field
+        if self.spark_sql_dynamic_filter_variable_mapping is None and "spark_sql_dynamic_filter_variable_mapping" in self.model_fields_set:
+            _dict['sparkSQLDynamicFilterVariableMapping'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PolicyExecutionRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "analysisJobRequestId": obj.get("analysisJobRequestId"),
+            "analyticsPipelineId": obj.get("analyticsPipelineId"),
+            "assetDirection": obj.get("assetDirection"),
+            "autoRetryEnabled": obj.get("autoRetryEnabled") if obj.get("autoRetryEnabled") is not None else False,
+            "columnVariables": [ColumnVariable.from_dict(_item) for _item in obj["columnVariables"]] if obj.get("columnVariables") is not None else None,
+            "engineType": obj.get("engineType") if obj.get("engineType") is not None else 'SPARK',
+            "executionType": obj.get("executionType"),
+            "includeInQualityScore": obj.get("includeInQualityScore") if obj.get("includeInQualityScore") is not None else True,
+            "markerConfigs": [PolicyExecutionMarkerConfig.from_dict(_item) for _item in obj["markerConfigs"]] if obj.get("markerConfigs") is not None else None,
+            "monitorExecutionId": obj.get("monitorExecutionId"),
+            "pipelineAutomationExecutionId": obj.get("pipelineAutomationExecutionId"),
+            "pipelineRunId": obj.get("pipelineRunId"),
+            "profileId": obj.get("profileId"),
+            "resourceStrategyType": obj.get("resourceStrategyType") if obj.get("resourceStrategyType") is not None else 'CUSTOM',
+            "rightEngineType": obj.get("rightEngineType") if obj.get("rightEngineType") is not None else 'SPARK',
+            "rightSparkFilterSelectedColumns": obj.get("rightSparkFilterSelectedColumns"),
+            "rightSparkSQLDynamicFilterVariableMapping": [RuleSparkSQLDynamicFilterVariableMapping.from_dict(_item) for _item in obj["rightSparkSQLDynamicFilterVariableMapping"]] if obj.get("rightSparkSQLDynamicFilterVariableMapping") is not None else None,
+            "ruleItemSelections": obj.get("ruleItemSelections"),
+            "selectedResourceInventory": obj.get("selectedResourceInventory"),
+            "sourceExecutionId": obj.get("sourceExecutionId"),
+            "sparkAppName": obj.get("sparkAppName"),
+            "sparkFilterSelectedColumns": obj.get("sparkFilterSelectedColumns"),
+            "sparkResourceConfig": SparkResourceConfig.from_dict(obj["sparkResourceConfig"]) if obj.get("sparkResourceConfig") is not None else None,
+            "sparkSQLDynamicFilterVariableMapping": [RuleSparkSQLDynamicFilterVariableMapping.from_dict(_item) for _item in obj["sparkSQLDynamicFilterVariableMapping"]] if obj.get("sparkSQLDynamicFilterVariableMapping") is not None else None
+        })
+        return _obj
+
+

acceldata/models/api/catalog/policy_group.py

@@ -0,0 +1,114 @@
+# 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 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 PolicyGroup(BaseModel):
+    """
+    PolicyGroup
+    """ # noqa: E501
+    created_at: Optional[datetime] = Field(default=None, alias="createdAt")
+    description: Optional[StrictStr] = None
+    id: Optional[StrictInt] = None
+    name: Optional[StrictStr] = None
+    tenant_id: Optional[StrictStr] = Field(default=None, alias="tenantId")
+    updated_at: Optional[datetime] = Field(default=None, alias="updatedAt")
+    __properties: ClassVar[List[str]] = ["createdAt", "description", "id", "name", "tenantId", "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 PolicyGroup 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,
+        )
+        # set to None if description (nullable) is None
+        # and model_fields_set contains the field
+        if self.description is None and "description" in self.model_fields_set:
+            _dict['description'] = None
+
+        # set to None if name (nullable) is None
+        # and model_fields_set contains the field
+        if self.name is None and "name" in self.model_fields_set:
+            _dict['name'] = 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 PolicyGroup 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"),
+            "description": obj.get("description"),
+            "id": obj.get("id"),
+            "name": obj.get("name"),
+            "tenantId": obj.get("tenantId"),
+            "updatedAt": obj.get("updatedAt")
+        })
+        return _obj
+
+

acceldata/models/api/catalog/policy_transform_udf.py

@@ -0,0 +1,113 @@
+# 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.udf_variable import UdfVariable
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class PolicyTransformUDF(BaseModel):
+    """
+    PolicyTransformUDF
+    """ # noqa: E501
+    asset_id: StrictInt = Field(alias="assetId")
+    column_names: Optional[List[StrictStr]] = Field(default=None, alias="columnNames")
+    id: Optional[StrictInt] = None
+    name: StrictStr
+    parameters: List[UdfVariable]
+    rule_id: Optional[StrictInt] = Field(default=None, alias="ruleId")
+    udf_id: StrictInt = Field(alias="udfId")
+    __properties: ClassVar[List[str]] = ["assetId", "columnNames", "id", "name", "parameters", "ruleId", "udfId"]
+
+    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 PolicyTransformUDF 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 parameters (list)
+        _items = []
+        if self.parameters:
+            for _item_parameters in self.parameters:
+                if _item_parameters:
+                    _items.append(_item_parameters.to_dict())
+            _dict['parameters'] = _items
+        # set to None if column_names (nullable) is None
+        # and model_fields_set contains the field
+        if self.column_names is None and "column_names" in self.model_fields_set:
+            _dict['columnNames'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of PolicyTransformUDF from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "assetId": obj.get("assetId"),
+            "columnNames": obj.get("columnNames"),
+            "id": obj.get("id"),
+            "name": obj.get("name"),
+            "parameters": [UdfVariable.from_dict(_item) for _item in obj["parameters"]] if obj.get("parameters") is not None else None,
+            "ruleId": obj.get("ruleId"),
+            "udfId": obj.get("udfId")
+        })
+        return _obj
+
+