unitycatalog-client 0.2.1__py3-none-any.whl

unitycatalog/client/models/table_info.py

@@ -0,0 +1,125 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    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 unitycatalog.client.models.column_info import ColumnInfo
+from unitycatalog.client.models.data_source_format import DataSourceFormat
+from unitycatalog.client.models.table_type import TableType
+from typing import Optional, Set
+from typing_extensions import Self
+
+class TableInfo(BaseModel):
+    """
+    TableInfo
+    """ # noqa: E501
+    name: Optional[StrictStr] = Field(default=None, description="Name of table, relative to parent schema.")
+    catalog_name: Optional[StrictStr] = Field(default=None, description="Name of parent catalog.")
+    schema_name: Optional[StrictStr] = Field(default=None, description="Name of parent schema relative to its parent catalog.")
+    table_type: Optional[TableType] = None
+    data_source_format: Optional[DataSourceFormat] = None
+    columns: Optional[List[ColumnInfo]] = Field(default=None, description="The array of __ColumnInfo__ definitions of the table's columns.")
+    storage_location: Optional[StrictStr] = Field(default=None, description="Storage root URL for table (for **MANAGED**, **EXTERNAL** tables)")
+    comment: Optional[StrictStr] = Field(default=None, description="User-provided free-form text description.")
+    properties: Optional[Dict[str, StrictStr]] = Field(default=None, description="A map of key-value properties attached to the securable.")
+    owner: Optional[StrictStr] = Field(default=None, description="Username of current owner of table.")
+    created_at: Optional[StrictInt] = Field(default=None, description="Time at which this table was created, in epoch milliseconds.")
+    created_by: Optional[StrictStr] = Field(default=None, description="Username of table creator.")
+    updated_at: Optional[StrictInt] = Field(default=None, description="Time at which this table was last modified, in epoch milliseconds.")
+    updated_by: Optional[StrictStr] = Field(default=None, description="Username of user who last modified the table.")
+    table_id: Optional[StrictStr] = Field(default=None, description="Unique identifier for the table.")
+    __properties: ClassVar[List[str]] = ["name", "catalog_name", "schema_name", "table_type", "data_source_format", "columns", "storage_location", "comment", "properties", "owner", "created_at", "created_by", "updated_at", "updated_by", "table_id"]
+
+    model_config = ConfigDict(
+        populate_by_name=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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of TableInfo 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 columns (list)
+        _items = []
+        if self.columns:
+            for _item in self.columns:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['columns'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of TableInfo from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name"),
+            "catalog_name": obj.get("catalog_name"),
+            "schema_name": obj.get("schema_name"),
+            "table_type": obj.get("table_type"),
+            "data_source_format": obj.get("data_source_format"),
+            "columns": [ColumnInfo.from_dict(_item) for _item in obj["columns"]] if obj.get("columns") is not None else None,
+            "storage_location": obj.get("storage_location"),
+            "comment": obj.get("comment"),
+            "properties": obj.get("properties"),
+            "owner": obj.get("owner"),
+            "created_at": obj.get("created_at"),
+            "created_by": obj.get("created_by"),
+            "updated_at": obj.get("updated_at"),
+            "updated_by": obj.get("updated_by"),
+            "table_id": obj.get("table_id")
+        })
+        return _obj
+
+

unitycatalog/client/models/table_operation.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class TableOperation(str, Enum):
+    """
+    TableOperation
+    """
+
+    """
+    allowed enum values
+    """
+    UNKNOWN_TABLE_OPERATION = 'UNKNOWN_TABLE_OPERATION'
+    READ = 'READ'
+    READ_WRITE = 'READ_WRITE'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of TableOperation from a JSON string"""
+        return cls(json.loads(json_str))
+
+

unitycatalog/client/models/table_type.py

@@ -0,0 +1,37 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class TableType(str, Enum):
+    """
+    TableType
+    """
+
+    """
+    allowed enum values
+    """
+    MANAGED = 'MANAGED'
+    EXTERNAL = 'EXTERNAL'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of TableType from a JSON string"""
+        return cls(json.loads(json_str))
+
+

unitycatalog/client/models/temporary_credentials.py

@@ -0,0 +1,105 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    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, Optional
+from unitycatalog.client.models.aws_credentials import AwsCredentials
+from unitycatalog.client.models.azure_user_delegation_sas import AzureUserDelegationSAS
+from unitycatalog.client.models.gcp_oauth_token import GcpOauthToken
+from typing import Optional, Set
+from typing_extensions import Self
+
+class TemporaryCredentials(BaseModel):
+    """
+    TemporaryCredentials
+    """ # noqa: E501
+    aws_temp_credentials: Optional[AwsCredentials] = None
+    azure_user_delegation_sas: Optional[AzureUserDelegationSAS] = None
+    gcp_oauth_token: Optional[GcpOauthToken] = None
+    expiration_time: Optional[StrictInt] = Field(default=None, description="Server time when the credential will expire, in epoch milliseconds. The API client is advised to cache the credential given this expiration time. ")
+    __properties: ClassVar[List[str]] = ["aws_temp_credentials", "azure_user_delegation_sas", "gcp_oauth_token", "expiration_time"]
+
+    model_config = ConfigDict(
+        populate_by_name=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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of TemporaryCredentials 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 aws_temp_credentials
+        if self.aws_temp_credentials:
+            _dict['aws_temp_credentials'] = self.aws_temp_credentials.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of azure_user_delegation_sas
+        if self.azure_user_delegation_sas:
+            _dict['azure_user_delegation_sas'] = self.azure_user_delegation_sas.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of gcp_oauth_token
+        if self.gcp_oauth_token:
+            _dict['gcp_oauth_token'] = self.gcp_oauth_token.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of TemporaryCredentials from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "aws_temp_credentials": AwsCredentials.from_dict(obj["aws_temp_credentials"]) if obj.get("aws_temp_credentials") is not None else None,
+            "azure_user_delegation_sas": AzureUserDelegationSAS.from_dict(obj["azure_user_delegation_sas"]) if obj.get("azure_user_delegation_sas") is not None else None,
+            "gcp_oauth_token": GcpOauthToken.from_dict(obj["gcp_oauth_token"]) if obj.get("gcp_oauth_token") is not None else None,
+            "expiration_time": obj.get("expiration_time")
+        })
+        return _obj
+
+

unitycatalog/client/models/update_catalog.py

@@ -0,0 +1,91 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    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
+
+class UpdateCatalog(BaseModel):
+    """
+    UpdateCatalog
+    """ # noqa: E501
+    comment: Optional[StrictStr] = Field(default=None, description="User-provided free-form text description.")
+    properties: Optional[Dict[str, StrictStr]] = Field(default=None, description="A map of key-value properties attached to the securable.")
+    new_name: Optional[StrictStr] = Field(default=None, description="New name for the catalog.")
+    __properties: ClassVar[List[str]] = ["comment", "properties", "new_name"]
+
+    model_config = ConfigDict(
+        populate_by_name=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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of UpdateCatalog 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 UpdateCatalog from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "comment": obj.get("comment"),
+            "properties": obj.get("properties"),
+            "new_name": obj.get("new_name")
+        })
+        return _obj
+
+

unitycatalog/client/models/update_model_version.py

@@ -0,0 +1,87 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    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
+
+class UpdateModelVersion(BaseModel):
+    """
+    UpdateModelVersion
+    """ # noqa: E501
+    comment: Optional[StrictStr] = Field(default=None, description="User-provided free-form text description.")
+    __properties: ClassVar[List[str]] = ["comment"]
+
+    model_config = ConfigDict(
+        populate_by_name=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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of UpdateModelVersion 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 UpdateModelVersion from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "comment": obj.get("comment")
+        })
+        return _obj
+
+

unitycatalog/client/models/update_permissions.py

@@ -0,0 +1,95 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    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
+from unitycatalog.client.models.permissions_change import PermissionsChange
+from typing import Optional, Set
+from typing_extensions import Self
+
+class UpdatePermissions(BaseModel):
+    """
+    UpdatePermissions
+    """ # noqa: E501
+    changes: List[PermissionsChange] = Field(description="Array of permissions change objects.")
+    __properties: ClassVar[List[str]] = ["changes"]
+
+    model_config = ConfigDict(
+        populate_by_name=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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of UpdatePermissions 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 changes (list)
+        _items = []
+        if self.changes:
+            for _item in self.changes:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['changes'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of UpdatePermissions from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "changes": [PermissionsChange.from_dict(_item) for _item in obj["changes"]] if obj.get("changes") is not None else None
+        })
+        return _obj
+
+

unitycatalog/client/models/update_registered_model.py

@@ -0,0 +1,89 @@
+# coding: utf-8
+
+"""
+    Unity Catalog API
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: 0.1
+    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
+
+class UpdateRegisteredModel(BaseModel):
+    """
+    UpdateRegisteredModel
+    """ # noqa: E501
+    comment: Optional[StrictStr] = Field(default=None, description="User-provided free-form text description.")
+    new_name: Optional[StrictStr] = Field(default=None, description="New name for the model.")
+    __properties: ClassVar[List[str]] = ["comment", "new_name"]
+
+    model_config = ConfigDict(
+        populate_by_name=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"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of UpdateRegisteredModel 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 UpdateRegisteredModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "comment": obj.get("comment"),
+            "new_name": obj.get("new_name")
+        })
+        return _obj
+
+