unitycatalog-client 0.2.1__py3-none-any.whl

unitycatalog/client/models/list_schemas_response.py

@@ -0,0 +1,97 @@
+# 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 unitycatalog.client.models.schema_info import SchemaInfo
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ListSchemasResponse(BaseModel):
+    """
+    ListSchemasResponse
+    """ # noqa: E501
+    schemas: Optional[List[SchemaInfo]] = Field(default=None, description="An array of schema information objects.")
+    next_page_token: Optional[StrictStr] = Field(default=None, description="Opaque token to retrieve the next page of results. Absent if there are no more pages. __page_token__ should be set to this value for the next request (for the next page of results). ")
+    __properties: ClassVar[List[str]] = ["schemas", "next_page_token"]
+
+    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 ListSchemasResponse 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 schemas (list)
+        _items = []
+        if self.schemas:
+            for _item in self.schemas:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['schemas'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ListSchemasResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "schemas": [SchemaInfo.from_dict(_item) for _item in obj["schemas"]] if obj.get("schemas") is not None else None,
+            "next_page_token": obj.get("next_page_token")
+        })
+        return _obj
+
+

unitycatalog/client/models/list_tables_response.py

@@ -0,0 +1,97 @@
+# 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 unitycatalog.client.models.table_info import TableInfo
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ListTablesResponse(BaseModel):
+    """
+    ListTablesResponse
+    """ # noqa: E501
+    tables: Optional[List[TableInfo]] = Field(default=None, description="An array of table information objects.")
+    next_page_token: Optional[StrictStr] = Field(default=None, description="Opaque token to retrieve the next page of results. Absent if there are no more pages. __page_token__ should be set to this value for the next request (for the next page of results). ")
+    __properties: ClassVar[List[str]] = ["tables", "next_page_token"]
+
+    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 ListTablesResponse 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 tables (list)
+        _items = []
+        if self.tables:
+            for _item in self.tables:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['tables'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ListTablesResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "tables": [TableInfo.from_dict(_item) for _item in obj["tables"]] if obj.get("tables") is not None else None,
+            "next_page_token": obj.get("next_page_token")
+        })
+        return _obj
+
+

unitycatalog/client/models/list_volumes_response_content.py

@@ -0,0 +1,97 @@
+# 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 unitycatalog.client.models.volume_info import VolumeInfo
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ListVolumesResponseContent(BaseModel):
+    """
+    ListVolumesResponseContent
+    """ # noqa: E501
+    volumes: Optional[List[VolumeInfo]] = None
+    next_page_token: Optional[StrictStr] = Field(default=None, description="Opaque token to retrieve the next page of results. Absent if there are no more pages. __page_token__ should be set to this value for the next request to retrieve the next page of results. ")
+    __properties: ClassVar[List[str]] = ["volumes", "next_page_token"]
+
+    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 ListVolumesResponseContent 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 volumes (list)
+        _items = []
+        if self.volumes:
+            for _item in self.volumes:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['volumes'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ListVolumesResponseContent from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "volumes": [VolumeInfo.from_dict(_item) for _item in obj["volumes"]] if obj.get("volumes") is not None else None,
+            "next_page_token": obj.get("next_page_token")
+        })
+        return _obj
+
+

unitycatalog/client/models/model_version_info.py

@@ -0,0 +1,114 @@
+# 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.model_version_status import ModelVersionStatus
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ModelVersionInfo(BaseModel):
+    """
+    ModelVersionInfo
+    """ # noqa: E501
+    model_name: Optional[StrictStr] = Field(default=None, description="The name of the parent registered model of the model version, relative to parent schema")
+    catalog_name: Optional[StrictStr] = Field(default=None, description="The name of the catalog containing the model version")
+    schema_name: Optional[StrictStr] = Field(default=None, description="The name of the schema containing the model version, relative to parent catalog")
+    version: Optional[StrictInt] = Field(default=None, description="Integer model version number, used to reference the model version in API requests.")
+    source: Optional[StrictStr] = Field(default=None, description="URI indicating the location of the source artifacts (files) for the model version")
+    run_id: Optional[StrictStr] = Field(default=None, description="The run id used by the ML package that generated the model.")
+    status: Optional[ModelVersionStatus] = None
+    storage_location: Optional[StrictStr] = Field(default=None, description="The storage location on the cloud under which model version data files are stored")
+    comment: Optional[StrictStr] = Field(default=None, description="The comment attached to the model version")
+    created_at: Optional[StrictInt] = Field(default=None, description="Time at which this model version was created, in epoch milliseconds.")
+    created_by: Optional[StrictStr] = Field(default=None, description="The identifier of the user who created the model version")
+    updated_at: Optional[StrictInt] = Field(default=None, description="Time at which this model version was last modified, in epoch milliseconds.")
+    updated_by: Optional[StrictStr] = Field(default=None, description="The identifier of the user who updated the model version last time")
+    id: Optional[StrictStr] = Field(default=None, description="Unique identifier for the model version.")
+    __properties: ClassVar[List[str]] = ["model_name", "catalog_name", "schema_name", "version", "source", "run_id", "status", "storage_location", "comment", "created_at", "created_by", "updated_at", "updated_by", "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 ModelVersionInfo 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 ModelVersionInfo from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "model_name": obj.get("model_name"),
+            "catalog_name": obj.get("catalog_name"),
+            "schema_name": obj.get("schema_name"),
+            "version": obj.get("version"),
+            "source": obj.get("source"),
+            "run_id": obj.get("run_id"),
+            "status": obj.get("status"),
+            "storage_location": obj.get("storage_location"),
+            "comment": obj.get("comment"),
+            "created_at": obj.get("created_at"),
+            "created_by": obj.get("created_by"),
+            "updated_at": obj.get("updated_at"),
+            "updated_by": obj.get("updated_by"),
+            "id": obj.get("id")
+        })
+        return _obj
+
+

unitycatalog/client/models/model_version_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 ModelVersionOperation(str, Enum):
+    """
+    ModelVersionOperation
+    """
+
+    """
+    allowed enum values
+    """
+    UNKNOWN_MODEL_VERSION_OPERATION = 'UNKNOWN_MODEL_VERSION_OPERATION'
+    READ_MODEL_VERSION = 'READ_MODEL_VERSION'
+    READ_WRITE_MODEL_VERSION = 'READ_WRITE_MODEL_VERSION'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ModelVersionOperation from a JSON string"""
+        return cls(json.loads(json_str))
+
+

unitycatalog/client/models/model_version_status.py

@@ -0,0 +1,39 @@
+# 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 ModelVersionStatus(str, Enum):
+    """
+    All possible model version statuses (MODEL_VERSION_STATUS_UNKNOWN, PENDING_REGISTRATION, FAILED_REGISTRATION, READY). PENDING_REGISTRATION status indicates that the client has not completely written all model artifacts to external storage. FAILED_REGISTRATION status indicates that the client has failed to write all model artifacts to external storage. READY status indicates that the client has successfully written all model artifacts to external storage. 
+    """
+
+    """
+    allowed enum values
+    """
+    MODEL_VERSION_STATUS_UNKNOWN = 'MODEL_VERSION_STATUS_UNKNOWN'
+    PENDING_REGISTRATION = 'PENDING_REGISTRATION'
+    FAILED_REGISTRATION = 'FAILED_REGISTRATION'
+    READY = 'READY'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of ModelVersionStatus from a JSON string"""
+        return cls(json.loads(json_str))
+
+

unitycatalog/client/models/path_operation.py

@@ -0,0 +1,39 @@
+# 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 PathOperation(str, Enum):
+    """
+    PathOperation
+    """
+
+    """
+    allowed enum values
+    """
+    UNKNOWN_PATH_OPERATION = 'UNKNOWN_PATH_OPERATION'
+    PATH_READ = 'PATH_READ'
+    PATH_READ_WRITE = 'PATH_READ_WRITE'
+    PATH_CREATE_TABLE = 'PATH_CREATE_TABLE'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of PathOperation from a JSON string"""
+        return cls(json.loads(json_str))
+
+

unitycatalog/client/models/permissions_change.py

@@ -0,0 +1,92 @@
+# 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
+from unitycatalog.client.models.privilege import Privilege
+from typing import Optional, Set
+from typing_extensions import Self
+
+class PermissionsChange(BaseModel):
+    """
+    PermissionsChange
+    """ # noqa: E501
+    principal: StrictStr = Field(description="The principal whose privileges we are changing.")
+    add: List[Privilege] = Field(description="The set of privileges to add.")
+    remove: List[Privilege] = Field(description="The set of privileges to remove.")
+    __properties: ClassVar[List[str]] = ["principal", "add", "remove"]
+
+    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 PermissionsChange 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 PermissionsChange from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "principal": obj.get("principal"),
+            "add": obj.get("add"),
+            "remove": obj.get("remove")
+        })
+        return _obj
+
+