databricks-sdk 0.55.0__py3-none-any.whl → 0.56.0__py3-none-any.whl

databricks/sdk/service/sharing.py

@@ -22,6 +22,7 @@ class AuthenticationType(Enum):
 
     DATABRICKS = "DATABRICKS"
     OAUTH_CLIENT_CREDENTIALS = "OAUTH_CLIENT_CREDENTIALS"
+    OIDC_FEDERATION = "OIDC_FEDERATION"
     TOKEN = "TOKEN"
 
 
@@ -2763,7 +2764,7 @@ class UpdateShare:
 @dataclass
 class UpdateSharePermissions:
     changes: Optional[List[PermissionsChange]] = None
-    """Array of permission changes."""
+    """Array of permissions change objects."""
 
     name: Optional[str] = None
     """The name of the share."""
@@ -3999,7 +4000,7 @@ class SharesAPI:
         :param name: str
           The name of the share.
         :param changes: List[:class:`PermissionsChange`] (optional)
-          Array of permission changes.
+          Array of permissions change objects.
         :param omit_permissions_list: bool (optional)
           Optional. Whether to return the latest permissions list of the share in the response.
 

databricks/sdk/service/sql.py

@@ -1696,6 +1696,68 @@ class CreateQueryRequestQuery:
         )
 
 
+@dataclass
+class CreateQueryVisualizationsLegacyRequest:
+    """Add visualization to a query"""
+
+    query_id: str
+    """The identifier returned by :method:queries/create"""
+
+    type: str
+    """The type of visualization: chart, table, pivot table, and so on."""
+
+    options: Any
+    """The options object varies widely from one visualization type to the next and is unsupported.
+    Databricks does not recommend modifying visualization settings in JSON."""
+
+    description: Optional[str] = None
+    """A short description of this visualization. This is not displayed in the UI."""
+
+    name: Optional[str] = None
+    """The name of the visualization that appears on dashboards and the query screen."""
+
+    def as_dict(self) -> dict:
+        """Serializes the CreateQueryVisualizationsLegacyRequest into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.description is not None:
+            body["description"] = self.description
+        if self.name is not None:
+            body["name"] = self.name
+        if self.options:
+            body["options"] = self.options
+        if self.query_id is not None:
+            body["query_id"] = self.query_id
+        if self.type is not None:
+            body["type"] = self.type
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the CreateQueryVisualizationsLegacyRequest into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.description is not None:
+            body["description"] = self.description
+        if self.name is not None:
+            body["name"] = self.name
+        if self.options:
+            body["options"] = self.options
+        if self.query_id is not None:
+            body["query_id"] = self.query_id
+        if self.type is not None:
+            body["type"] = self.type
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> CreateQueryVisualizationsLegacyRequest:
+        """Deserializes the CreateQueryVisualizationsLegacyRequest from a dictionary."""
+        return cls(
+            description=d.get("description", None),
+            name=d.get("name", None),
+            options=d.get("options", None),
+            query_id=d.get("query_id", None),
+            type=d.get("type", None),
+        )
+
+
 @dataclass
 class CreateVisualizationRequest:
     visualization: Optional[CreateVisualizationRequestVisualization] = None
@@ -6047,6 +6109,10 @@ class QueryMetrics:
     spill_to_disk_bytes: Optional[int] = None
     """Size of data temporarily written to disk while executing the query, in bytes."""
 
+    task_time_over_time_range: Optional[TaskTimeOverRange] = None
+    """sum of task times completed in a range of wall clock time, approximated to a configurable number
+    of points aggregated over all stages and jobs in the query (based on task_total_time_ms)"""
+
     task_total_time_ms: Optional[int] = None
     """Sum of execution time for all of the query’s tasks, in milliseconds."""
 
@@ -6097,6 +6163,8 @@ class QueryMetrics:
             body["rows_read_count"] = self.rows_read_count
         if self.spill_to_disk_bytes is not None:
             body["spill_to_disk_bytes"] = self.spill_to_disk_bytes
+        if self.task_time_over_time_range:
+            body["task_time_over_time_range"] = self.task_time_over_time_range.as_dict()
         if self.task_total_time_ms is not None:
             body["task_total_time_ms"] = self.task_total_time_ms
         if self.total_time_ms is not None:
@@ -6146,6 +6214,8 @@ class QueryMetrics:
             body["rows_read_count"] = self.rows_read_count
         if self.spill_to_disk_bytes is not None:
             body["spill_to_disk_bytes"] = self.spill_to_disk_bytes
+        if self.task_time_over_time_range:
+            body["task_time_over_time_range"] = self.task_time_over_time_range
         if self.task_total_time_ms is not None:
             body["task_total_time_ms"] = self.task_total_time_ms
         if self.total_time_ms is not None:
@@ -6177,6 +6247,7 @@ class QueryMetrics:
             rows_produced_count=d.get("rows_produced_count", None),
             rows_read_count=d.get("rows_read_count", None),
             spill_to_disk_bytes=d.get("spill_to_disk_bytes", None),
+            task_time_over_time_range=_from_dict(d, "task_time_over_time_range", TaskTimeOverRange),
             task_total_time_ms=d.get("task_total_time_ms", None),
             total_time_ms=d.get("total_time_ms", None),
             write_remote_bytes=d.get("write_remote_bytes", None),
@@ -6763,6 +6834,50 @@ class ServiceErrorCode(Enum):
     WORKSPACE_TEMPORARILY_UNAVAILABLE = "WORKSPACE_TEMPORARILY_UNAVAILABLE"
 
 
+@dataclass
+class SetRequest:
+    """Set object ACL"""
+
+    access_control_list: Optional[List[AccessControl]] = None
+
+    object_id: Optional[str] = None
+    """Object ID. The ACL for the object with this UUID is overwritten by this request's POST content."""
+
+    object_type: Optional[ObjectTypePlural] = None
+    """The type of object permission to set."""
+
+    def as_dict(self) -> dict:
+        """Serializes the SetRequest into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.access_control_list:
+            body["access_control_list"] = [v.as_dict() for v in self.access_control_list]
+        if self.object_id is not None:
+            body["objectId"] = self.object_id
+        if self.object_type is not None:
+            body["objectType"] = self.object_type.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the SetRequest into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.access_control_list:
+            body["access_control_list"] = self.access_control_list
+        if self.object_id is not None:
+            body["objectId"] = self.object_id
+        if self.object_type is not None:
+            body["objectType"] = self.object_type
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> SetRequest:
+        """Deserializes the SetRequest from a dictionary."""
+        return cls(
+            access_control_list=_repeated_dict(d, "access_control_list", AccessControl),
+            object_id=d.get("objectId", None),
+            object_type=_enum(d, "objectType", ObjectTypePlural),
+        )
+
+
 @dataclass
 class SetResponse:
     access_control_list: Optional[List[AccessControl]] = None
@@ -7167,6 +7282,63 @@ class SuccessMessage(Enum):
     SUCCESS = "Success"
 
 
+@dataclass
+class TaskTimeOverRange:
+    entries: Optional[List[TaskTimeOverRangeEntry]] = None
+
+    interval: Optional[int] = None
+    """interval length for all entries (difference in start time and end time of an entry range) the
+    same for all entries start time of first interval is query_start_time_ms"""
+
+    def as_dict(self) -> dict:
+        """Serializes the TaskTimeOverRange into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.entries:
+            body["entries"] = [v.as_dict() for v in self.entries]
+        if self.interval is not None:
+            body["interval"] = self.interval
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TaskTimeOverRange into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.entries:
+            body["entries"] = self.entries
+        if self.interval is not None:
+            body["interval"] = self.interval
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TaskTimeOverRange:
+        """Deserializes the TaskTimeOverRange from a dictionary."""
+        return cls(entries=_repeated_dict(d, "entries", TaskTimeOverRangeEntry), interval=d.get("interval", None))
+
+
+@dataclass
+class TaskTimeOverRangeEntry:
+    task_completed_time_ms: Optional[int] = None
+    """total task completion time in this time range, aggregated over all stages and jobs in the query"""
+
+    def as_dict(self) -> dict:
+        """Serializes the TaskTimeOverRangeEntry into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.task_completed_time_ms is not None:
+            body["task_completed_time_ms"] = self.task_completed_time_ms
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TaskTimeOverRangeEntry into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.task_completed_time_ms is not None:
+            body["task_completed_time_ms"] = self.task_completed_time_ms
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TaskTimeOverRangeEntry:
+        """Deserializes the TaskTimeOverRangeEntry from a dictionary."""
+        return cls(task_completed_time_ms=d.get("task_completed_time_ms", None))
+
+
 @dataclass
 class TerminationReason:
     code: Optional[TerminationReasonCode] = None
@@ -7384,6 +7556,51 @@ class TransferOwnershipObjectId:
         return cls(new_owner=d.get("new_owner", None))
 
 
+@dataclass
+class TransferOwnershipRequest:
+    """Transfer object ownership"""
+
+    new_owner: Optional[str] = None
+    """Email address for the new owner, who must exist in the workspace."""
+
+    object_id: Optional[TransferOwnershipObjectId] = None
+    """The ID of the object on which to change ownership."""
+
+    object_type: Optional[OwnableObjectType] = None
+    """The type of object on which to change ownership."""
+
+    def as_dict(self) -> dict:
+        """Serializes the TransferOwnershipRequest into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.new_owner is not None:
+            body["new_owner"] = self.new_owner
+        if self.object_id:
+            body["objectId"] = self.object_id.as_dict()
+        if self.object_type is not None:
+            body["objectType"] = self.object_type.value
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the TransferOwnershipRequest into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.new_owner is not None:
+            body["new_owner"] = self.new_owner
+        if self.object_id:
+            body["objectId"] = self.object_id
+        if self.object_type is not None:
+            body["objectType"] = self.object_type
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> TransferOwnershipRequest:
+        """Deserializes the TransferOwnershipRequest from a dictionary."""
+        return cls(
+            new_owner=d.get("new_owner", None),
+            object_id=_from_dict(d, "objectId", TransferOwnershipObjectId),
+            object_type=_enum(d, "objectType", OwnableObjectType),
+        )
+
+
 @dataclass
 class UpdateAlertRequest:
     update_mask: str
@@ -8760,7 +8977,7 @@ class AlertsLegacyAPI:
 
 
 class AlertsV2API:
-    """TODO: Add description"""
+    """New version of SQL Alerts"""
 
     def __init__(self, api_client):
         self._api = api_client

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = "0.55.0"
+__version__ = "0.56.0"

{databricks_sdk-0.55.0.dist-info → databricks_sdk-0.56.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-sdk
-Version: 0.55.0
+Version: 0.56.0
 Summary: Databricks SDK for Python (Beta)
 Project-URL: Documentation, https://databricks-sdk-py.readthedocs.io
 Keywords: databricks,sdk

{databricks_sdk-0.55.0.dist-info → databricks_sdk-0.56.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 databricks/__init__.py,sha256=CF2MJcZFwbpn9TwQER8qnCDhkPooBGQNVkX4v7g6p3g,537
-databricks/sdk/__init__.py,sha256=lliePy2zXmwEuf6KYDV6LjEl3ehc-SwCCZNvipK0xtg,57869
+databricks/sdk/__init__.py,sha256=PNn6lVdcn-IpZyM2Ll_VCmL1GapmIcpLWd3uZ7RXGSw,58527
 databricks/sdk/_base_client.py,sha256=IMHtzC5BhWt-lBVjifewR1Ah5fegGDMv0__-O1hCxWI,15850
 databricks/sdk/_property.py,sha256=ccbxhkXZmZOxbx2sqKMTzhVZDuvWXG0WPHFRgac6JAM,1701
 databricks/sdk/azure.py,sha256=sN_ARpmP9h1JovtiHIsDLtrVQP_K11eNDDtHS6PD19k,1015
@@ -17,7 +17,7 @@ databricks/sdk/oidc_token_supplier.py,sha256=QrO6J0QY4yFfcdQDL5h2OfxMxvBZJPtPmPe
 databricks/sdk/py.typed,sha256=pSvaHpbY1UPNEXyVFUjlgBhjPFZMmVC_UNrPC7eMOHI,74
 databricks/sdk/retries.py,sha256=7k2kEexGqGKXHNAWHbPFSZSugU8UIU0qtyly_hix22Q,2581
 databricks/sdk/useragent.py,sha256=boEgzTv-Zmo6boipZKjSopNy0CXg4GShC1_lTKpJgqs,7361
-databricks/sdk/version.py,sha256=UQLHFqXC6UhDcoKB4MFAazCsPAeQEfBN68BEvb4D2Tk,23
+databricks/sdk/version.py,sha256=pLF99dIRilisWyZd9CliPPWqIlV81Xy8XtGEMuThTaM,23
 databricks/sdk/_widgets/__init__.py,sha256=VhI-VvLlr3rKUT1nbROslHJIbmZX_tPJ9rRhrdFsYUA,2811
 databricks/sdk/_widgets/default_widgets_utils.py,sha256=_hwCbptLbRzWEmknco0H1wQNAYcuy2pjFO9NiRbvFeo,1127
 databricks/sdk/_widgets/ipywidgets_utils.py,sha256=mg3rEPG9z76e0yVjGgcLybUvd_zSuN5ziGeKiZ-c8Ew,2927
@@ -44,29 +44,32 @@ databricks/sdk/runtime/__init__.py,sha256=6nthZxeYY1HjHieQcP7kXVLIId7w2yfHpZRXXt
 databricks/sdk/runtime/dbutils_stub.py,sha256=S_pgWyGmwp3Ay-pMDEXccYsPwNVqCtz7MpD3fZVlHUA,11408
 databricks/sdk/service/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databricks/sdk/service/_internal.py,sha256=PY83MPehEwGuMzCnyvolqglnfZeQ7-eS38kedTa7KDU,1985
+databricks/sdk/service/aibuilder.py,sha256=u3WTVBm0sw8aeC9p4kfZAlHtl_6GU2DiNK_vUAMNgfk,12436
 databricks/sdk/service/apps.py,sha256=az06ij0_I4G3UcdRs5Pob4uXn5EjJoysXl1F8rR_NQ4,58080
-databricks/sdk/service/billing.py,sha256=2_6QVp6KPTKqFHgoaa8dGSOrD1P8i-UjKaupWOxpMW8,99177
-databricks/sdk/service/catalog.py,sha256=8zJ3NqN90r1GhxeqpvY3rhUiEW3rJ5c89220B2nG2Lk,646689
-databricks/sdk/service/cleanrooms.py,sha256=ZrlsMFJ8QSXSztwwQ-qa1QKfe-y--0yAOrfzhRjgyYI,63372
-databricks/sdk/service/compute.py,sha256=okznn__wKCZDaDIarf2Qp7SySK6wMa3z9gu5mxPLsCc,571739
-databricks/sdk/service/dashboards.py,sha256=NAG1DOPqp8IWcqLVTpB27FeFI56jC7VeX6uoEpJug5o,118497
-databricks/sdk/service/files.py,sha256=nAbN1US56vJYk-YDPShfT_mgk_KIhgbidtSGTraCqPo,46335
-databricks/sdk/service/iam.py,sha256=E4_FQT0eumpBfR_qy8IK1qL-ssSEhYrJJXsPb6GUzF4,179229
-databricks/sdk/service/jobs.py,sha256=tDWwAtN54DvoHIxLld1oKe6DpywSjvbjA-Jl9g3vF8s,466539
+databricks/sdk/service/billing.py,sha256=fAJQNJcFSMwIYlAiX-s9S4FvWZfXE5niwNw7HI9nMUo,96079
+databricks/sdk/service/catalog.py,sha256=ZcHscrNVi0T0gU0uLJXD0PpeEtz9_JXi0DuzoRuPwsk,625998
+databricks/sdk/service/cleanrooms.py,sha256=mFg0hp7GtO_ZZBZ6p38Gx-ggMNUCJpuvITuh6n-zTLs,63702
+databricks/sdk/service/compute.py,sha256=ZNERuPZGZJkNT54Ye9WJhh4ctEssIpcPIRo7eOO8RPg,570795
+databricks/sdk/service/dashboards.py,sha256=aDDf23c7O4hAMlMzpAz4m1eSLgI2OjEjhE04PxpSTGc,103473
+databricks/sdk/service/database.py,sha256=i-CPsMqX8RJnaa_Nr_J6CxDuoyhls-6rfsKxYToQ8D0,51164
+databricks/sdk/service/files.py,sha256=rMu5rmlnhGr1Sic1XWOZi_Pnv-pARQfbumO50JxS4DA,46401
+databricks/sdk/service/iam.py,sha256=ZURErK_Nd9aIuU1b2kGZZJg9tXBSfNBRC7QpozJBvwc,179241
+databricks/sdk/service/jobs.py,sha256=KfYPBWfgll_LKXg8KgB_bhGiMqoAx2h7WfJoKsA1Efw,475787
 databricks/sdk/service/marketplace.py,sha256=NLCJ7zb0Oj5nhLth9R1Szuz6fSSoxAnsKvFhKgx994s,175741
-databricks/sdk/service/ml.py,sha256=ovv33tW8aS28smK8IrDo6J_hxfOeA2Qe_49tWPT6DzY,374308
+databricks/sdk/service/ml.py,sha256=orS3-xEJWtPibxshGDvyso4MtLdrc9UKGcff6bKlB3E,363830
 databricks/sdk/service/oauth2.py,sha256=OMXkb5eo4D7WfCrxBlJeY_jVTHXdpZNJl-Sv8o25jxc,81101
-databricks/sdk/service/pipelines.py,sha256=ha9WnaZguTiS-gXdQ9UFptxCFkGdfKcy41t-umRi3ww,173948
-databricks/sdk/service/provisioning.py,sha256=-Ly2o02i-jhNmiP9zLPeYF8H2usoB-oTG0RLF5gkIpc,169311
-databricks/sdk/service/serving.py,sha256=ugI94MriielW2X8bjm5C_cGqEwMNFsGVcsiqOY89RLQ,226865
-databricks/sdk/service/settings.py,sha256=GJ68sZMDPSOTZqc2-Nd1gEYgWM3dul80zS94mU620vQ,400566
-databricks/sdk/service/sharing.py,sha256=i7d5ggmOnHJ2gVp4PfPmf14M-AUDRIU-XZb4SqkrVCo,160173
-databricks/sdk/service/sql.py,sha256=v6uspo4o0VoUW0T2eHhPDlJYeYah1vhR_WMxVwEhu2Y,442747
+databricks/sdk/service/pipelines.py,sha256=OUdYMS9h1gxiwpK3OqaXNqJochCaByOuvQ_qBxxO4zA,176094
+databricks/sdk/service/provisioning.py,sha256=CEzgilTXUNbLjDsvVPxC3mYsAVKHO4v0TnwSIgI373Q,169281
+databricks/sdk/service/qualitymonitorv2.py,sha256=lhnw_7wyr3Uj0ywCNE15SUQXg-Qnhz_sJ_wFcXP5w9w,9991
+databricks/sdk/service/serving.py,sha256=hrGH4qCSJooHSdUhqyMUXJKfPg126q53cpHpXuWMBUI,231597
+databricks/sdk/service/settings.py,sha256=KmPLdnoJoOK7flmuMvrPQvgPCaPiEuQqTj9G4emIFSM,441251
+databricks/sdk/service/sharing.py,sha256=ugzmeXN-1PMZS58JHaQZzUSyxmSMebO3dHCsFGbrmkc,160229
+databricks/sdk/service/sql.py,sha256=xCj080qniFiVafkt-pne71vfd45sKgLARpObCIDyt90,451364
 databricks/sdk/service/vectorsearch.py,sha256=Y1yUJwoY7rg6HWKh7_90aJzIWNiWEG2WUw0i1n3KH7I,86701
 databricks/sdk/service/workspace.py,sha256=T0ZbnG1qcPjKysGO_tBzl5x1PyalydeYJRBZbooYNm0,130893
-databricks_sdk-0.55.0.dist-info/licenses/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
-databricks_sdk-0.55.0.dist-info/licenses/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
-databricks_sdk-0.55.0.dist-info/METADATA,sha256=ufiMw6V7EtmZD-5ZuCoanbezk94CfykKLbtNASbNFDI,39397
-databricks_sdk-0.55.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-databricks_sdk-0.55.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
-databricks_sdk-0.55.0.dist-info/RECORD,,
+databricks_sdk-0.56.0.dist-info/licenses/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
+databricks_sdk-0.56.0.dist-info/licenses/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
+databricks_sdk-0.56.0.dist-info/METADATA,sha256=XmtTcmgu2d7o6z-RTevL5mQ5D_FsyTVa-oCXcat2ChQ,39397
+databricks_sdk-0.56.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+databricks_sdk-0.56.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
+databricks_sdk-0.56.0.dist-info/RECORD,,