databricks-sdk 0.41.0__py3-none-any.whl → 0.42.0__py3-none-any.whl

databricks/sdk/service/catalog.py

@@ -787,9 +787,6 @@ class CatalogInfo:
     provisioning_info: Optional[ProvisioningInfo] = None
     """Status of an asynchronously provisioned resource."""
 
-    securable_kind: Optional[CatalogInfoSecurableKind] = None
-    """Kind of catalog securable."""
-
     securable_type: Optional[str] = None
 
     share_name: Optional[str] = None
@@ -831,7 +828,6 @@ class CatalogInfo:
         if self.properties: body['properties'] = self.properties
         if self.provider_name is not None: body['provider_name'] = self.provider_name
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info.as_dict()
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind.value
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.share_name is not None: body['share_name'] = self.share_name
         if self.storage_location is not None: body['storage_location'] = self.storage_location
@@ -862,7 +858,6 @@ class CatalogInfo:
         if self.properties: body['properties'] = self.properties
         if self.provider_name is not None: body['provider_name'] = self.provider_name
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.share_name is not None: body['share_name'] = self.share_name
         if self.storage_location is not None: body['storage_location'] = self.storage_location
@@ -893,7 +888,6 @@ class CatalogInfo:
                    properties=d.get('properties', None),
                    provider_name=d.get('provider_name', None),
                    provisioning_info=_from_dict(d, 'provisioning_info', ProvisioningInfo),
-                   securable_kind=_enum(d, 'securable_kind', CatalogInfoSecurableKind),
                    securable_type=d.get('securable_type', None),
                    share_name=d.get('share_name', None),
                    storage_location=d.get('storage_location', None),
@@ -902,24 +896,6 @@ class CatalogInfo:
                    updated_by=d.get('updated_by', None))
 
 
-class CatalogInfoSecurableKind(Enum):
-    """Kind of catalog securable."""
-
-    CATALOG_DELTASHARING = 'CATALOG_DELTASHARING'
-    CATALOG_FOREIGN_BIGQUERY = 'CATALOG_FOREIGN_BIGQUERY'
-    CATALOG_FOREIGN_DATABRICKS = 'CATALOG_FOREIGN_DATABRICKS'
-    CATALOG_FOREIGN_MYSQL = 'CATALOG_FOREIGN_MYSQL'
-    CATALOG_FOREIGN_POSTGRESQL = 'CATALOG_FOREIGN_POSTGRESQL'
-    CATALOG_FOREIGN_REDSHIFT = 'CATALOG_FOREIGN_REDSHIFT'
-    CATALOG_FOREIGN_SNOWFLAKE = 'CATALOG_FOREIGN_SNOWFLAKE'
-    CATALOG_FOREIGN_SQLDW = 'CATALOG_FOREIGN_SQLDW'
-    CATALOG_FOREIGN_SQLSERVER = 'CATALOG_FOREIGN_SQLSERVER'
-    CATALOG_INTERNAL = 'CATALOG_INTERNAL'
-    CATALOG_STANDARD = 'CATALOG_STANDARD'
-    CATALOG_SYSTEM = 'CATALOG_SYSTEM'
-    CATALOG_SYSTEM_DELTASHARING = 'CATALOG_SYSTEM_DELTASHARING'
-
-
 class CatalogIsolationMode(Enum):
     """Whether the current securable is accessible from all workspaces or a specific set of workspaces."""
 
@@ -1158,9 +1134,6 @@ class ConnectionInfo:
     read_only: Optional[bool] = None
     """If the connection is read only."""
 
-    securable_kind: Optional[ConnectionInfoSecurableKind] = None
-    """Kind of connection securable."""
-
     securable_type: Optional[str] = None
 
     updated_at: Optional[int] = None
@@ -1189,7 +1162,6 @@ class ConnectionInfo:
         if self.properties: body['properties'] = self.properties
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info.as_dict()
         if self.read_only is not None: body['read_only'] = self.read_only
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind.value
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.updated_at is not None: body['updated_at'] = self.updated_at
         if self.updated_by is not None: body['updated_by'] = self.updated_by
@@ -1213,7 +1185,6 @@ class ConnectionInfo:
         if self.properties: body['properties'] = self.properties
         if self.provisioning_info: body['provisioning_info'] = self.provisioning_info
         if self.read_only is not None: body['read_only'] = self.read_only
-        if self.securable_kind is not None: body['securable_kind'] = self.securable_kind
         if self.securable_type is not None: body['securable_type'] = self.securable_type
         if self.updated_at is not None: body['updated_at'] = self.updated_at
         if self.updated_by is not None: body['updated_by'] = self.updated_by
@@ -1237,31 +1208,12 @@ class ConnectionInfo:
                    properties=d.get('properties', None),
                    provisioning_info=_from_dict(d, 'provisioning_info', ProvisioningInfo),
                    read_only=d.get('read_only', None),
-                   securable_kind=_enum(d, 'securable_kind', ConnectionInfoSecurableKind),
                    securable_type=d.get('securable_type', None),
                    updated_at=d.get('updated_at', None),
                    updated_by=d.get('updated_by', None),
                    url=d.get('url', None))
 
 
-class ConnectionInfoSecurableKind(Enum):
-    """Kind of connection securable."""
-
-    CONNECTION_BIGQUERY = 'CONNECTION_BIGQUERY'
-    CONNECTION_BUILTIN_HIVE_METASTORE = 'CONNECTION_BUILTIN_HIVE_METASTORE'
-    CONNECTION_DATABRICKS = 'CONNECTION_DATABRICKS'
-    CONNECTION_EXTERNAL_HIVE_METASTORE = 'CONNECTION_EXTERNAL_HIVE_METASTORE'
-    CONNECTION_GLUE = 'CONNECTION_GLUE'
-    CONNECTION_HTTP_BEARER = 'CONNECTION_HTTP_BEARER'
-    CONNECTION_MYSQL = 'CONNECTION_MYSQL'
-    CONNECTION_ONLINE_CATALOG = 'CONNECTION_ONLINE_CATALOG'
-    CONNECTION_POSTGRESQL = 'CONNECTION_POSTGRESQL'
-    CONNECTION_REDSHIFT = 'CONNECTION_REDSHIFT'
-    CONNECTION_SNOWFLAKE = 'CONNECTION_SNOWFLAKE'
-    CONNECTION_SQLDW = 'CONNECTION_SQLDW'
-    CONNECTION_SQLSERVER = 'CONNECTION_SQLSERVER'
-
-
 class ConnectionType(Enum):
     """The type of connection."""
 
@@ -6269,20 +6221,21 @@ SecurablePropertiesMap = Dict[str, str]
 class SecurableType(Enum):
     """The type of Unity Catalog securable"""
 
-    CATALOG = 'catalog'
-    CONNECTION = 'connection'
-    CREDENTIAL = 'credential'
-    EXTERNAL_LOCATION = 'external_location'
-    FUNCTION = 'function'
-    METASTORE = 'metastore'
-    PIPELINE = 'pipeline'
-    PROVIDER = 'provider'
-    RECIPIENT = 'recipient'
-    SCHEMA = 'schema'
-    SHARE = 'share'
-    STORAGE_CREDENTIAL = 'storage_credential'
-    TABLE = 'table'
-    VOLUME = 'volume'
+    CATALOG = 'CATALOG'
+    CLEAN_ROOM = 'CLEAN_ROOM'
+    CONNECTION = 'CONNECTION'
+    CREDENTIAL = 'CREDENTIAL'
+    EXTERNAL_LOCATION = 'EXTERNAL_LOCATION'
+    FUNCTION = 'FUNCTION'
+    METASTORE = 'METASTORE'
+    PIPELINE = 'PIPELINE'
+    PROVIDER = 'PROVIDER'
+    RECIPIENT = 'RECIPIENT'
+    SCHEMA = 'SCHEMA'
+    SHARE = 'SHARE'
+    STORAGE_CREDENTIAL = 'STORAGE_CREDENTIAL'
+    TABLE = 'TABLE'
+    VOLUME = 'VOLUME'
 
 
 @dataclass

databricks/sdk/service/cleanrooms.py

@@ -289,11 +289,24 @@ class CleanRoomAssetNotebook:
     """Base 64 representation of the notebook contents. This is the same format as returned by
     :method:workspace/export with the format of **HTML**."""
 
+    review_state: Optional[CleanRoomNotebookReviewNotebookReviewState] = None
+    """top-level status derived from all reviews"""
+
+    reviews: Optional[List[CleanRoomNotebookReview]] = None
+    """All existing approvals or rejections"""
+
+    runner_collaborators: Optional[List[CleanRoomCollaborator]] = None
+    """collaborators that can run the notebook"""
+
     def as_dict(self) -> dict:
         """Serializes the CleanRoomAssetNotebook into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.etag is not None: body['etag'] = self.etag
         if self.notebook_content is not None: body['notebook_content'] = self.notebook_content
+        if self.review_state is not None: body['review_state'] = self.review_state.value
+        if self.reviews: body['reviews'] = [v.as_dict() for v in self.reviews]
+        if self.runner_collaborators:
+            body['runner_collaborators'] = [v.as_dict() for v in self.runner_collaborators]
         return body
 
     def as_shallow_dict(self) -> dict:
@@ -301,12 +314,19 @@ class CleanRoomAssetNotebook:
         body = {}
         if self.etag is not None: body['etag'] = self.etag
         if self.notebook_content is not None: body['notebook_content'] = self.notebook_content
+        if self.review_state is not None: body['review_state'] = self.review_state
+        if self.reviews: body['reviews'] = self.reviews
+        if self.runner_collaborators: body['runner_collaborators'] = self.runner_collaborators
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> CleanRoomAssetNotebook:
         """Deserializes the CleanRoomAssetNotebook from a dictionary."""
-        return cls(etag=d.get('etag', None), notebook_content=d.get('notebook_content', None))
+        return cls(etag=d.get('etag', None),
+                   notebook_content=d.get('notebook_content', None),
+                   review_state=_enum(d, 'review_state', CleanRoomNotebookReviewNotebookReviewState),
+                   reviews=_repeated_dict(d, 'reviews', CleanRoomNotebookReview),
+                   runner_collaborators=_repeated_dict(d, 'runner_collaborators', CleanRoomCollaborator))
 
 
 class CleanRoomAssetStatusEnum(Enum):
@@ -511,6 +531,56 @@ class CleanRoomCollaborator:
                    organization_name=d.get('organization_name', None))
 
 
+@dataclass
+class CleanRoomNotebookReview:
+    comment: Optional[str] = None
+    """review comment"""
+
+    created_at_millis: Optional[int] = None
+    """timestamp of when the review was submitted"""
+
+    review_state: Optional[CleanRoomNotebookReviewNotebookReviewState] = None
+    """review outcome"""
+
+    reviewer_collaborator_alias: Optional[str] = None
+    """collaborator alias of the reviewer"""
+
+    def as_dict(self) -> dict:
+        """Serializes the CleanRoomNotebookReview into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.comment is not None: body['comment'] = self.comment
+        if self.created_at_millis is not None: body['created_at_millis'] = self.created_at_millis
+        if self.review_state is not None: body['review_state'] = self.review_state.value
+        if self.reviewer_collaborator_alias is not None:
+            body['reviewer_collaborator_alias'] = self.reviewer_collaborator_alias
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CleanRoomNotebookReview into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.comment is not None: body['comment'] = self.comment
+        if self.created_at_millis is not None: body['created_at_millis'] = self.created_at_millis
+        if self.review_state is not None: body['review_state'] = self.review_state
+        if self.reviewer_collaborator_alias is not None:
+            body['reviewer_collaborator_alias'] = self.reviewer_collaborator_alias
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> CleanRoomNotebookReview:
+        """Deserializes the CleanRoomNotebookReview from a dictionary."""
+        return cls(comment=d.get('comment', None),
+                   created_at_millis=d.get('created_at_millis', None),
+                   review_state=_enum(d, 'review_state', CleanRoomNotebookReviewNotebookReviewState),
+                   reviewer_collaborator_alias=d.get('reviewer_collaborator_alias', None))
+
+
+class CleanRoomNotebookReviewNotebookReviewState(Enum):
+
+    APPROVED = 'APPROVED'
+    PENDING = 'PENDING'
+    REJECTED = 'REJECTED'
+
+
 @dataclass
 class CleanRoomNotebookTaskRun:
     """Stores information about a single task run."""

databricks/sdk/service/compute.py

@@ -2937,6 +2937,42 @@ class Created:
         return cls(id=d.get('id', None))
 
 
+@dataclass
+class CustomPolicyTag:
+    key: str
+    """The key of the tag. - Must be unique among all custom tags of the same policy - Cannot be
+    “budget-policy-name”, “budget-policy-id” or "budget-policy-resolution-result" - these
+    tags are preserved.
+    
+    - Follows the regex pattern defined in cluster-common/conf/src/ClusterTagConstraints.scala
+    (https://src.dev.databricks.com/databricks/universe@1647196627c8dc7b4152ad098a94b86484b93a6c/-/blob/cluster-common/conf/src/ClusterTagConstraints.scala?L17)"""
+
+    value: Optional[str] = None
+    """The value of the tag.
+    
+    - Follows the regex pattern defined in cluster-common/conf/src/ClusterTagConstraints.scala
+    (https://src.dev.databricks.com/databricks/universe@1647196627c8dc7b4152ad098a94b86484b93a6c/-/blob/cluster-common/conf/src/ClusterTagConstraints.scala?L24)"""
+
+    def as_dict(self) -> dict:
+        """Serializes the CustomPolicyTag into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.key is not None: body['key'] = self.key
+        if self.value is not None: body['value'] = self.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CustomPolicyTag into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.key is not None: body['key'] = self.key
+        if self.value is not None: body['value'] = self.value
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> CustomPolicyTag:
+        """Deserializes the CustomPolicyTag from a dictionary."""
+        return cls(key=d.get('key', None), value=d.get('value', None))
+
+
 @dataclass
 class DataPlaneEventDetails:
     event_type: Optional[DataPlaneEventDetailsEventType] = None

databricks/sdk/service/dashboards.py

@@ -846,6 +846,8 @@ class QueryAttachment:
     query: Optional[str] = None
     """AI generated SQL query"""
 
+    statement_id: Optional[str] = None
+
     title: Optional[str] = None
     """Name of the query"""
 
@@ -860,6 +862,7 @@ class QueryAttachment:
         if self.last_updated_timestamp is not None:
             body['last_updated_timestamp'] = self.last_updated_timestamp
         if self.query is not None: body['query'] = self.query
+        if self.statement_id is not None: body['statement_id'] = self.statement_id
         if self.title is not None: body['title'] = self.title
         return body
 
@@ -874,6 +877,7 @@ class QueryAttachment:
         if self.last_updated_timestamp is not None:
             body['last_updated_timestamp'] = self.last_updated_timestamp
         if self.query is not None: body['query'] = self.query
+        if self.statement_id is not None: body['statement_id'] = self.statement_id
         if self.title is not None: body['title'] = self.title
         return body
 
@@ -887,6 +891,7 @@ class QueryAttachment:
                    instruction_title=d.get('instruction_title', None),
                    last_updated_timestamp=d.get('last_updated_timestamp', None),
                    query=d.get('query', None),
+                   statement_id=d.get('statement_id', None),
                    title=d.get('title', None))