databricks-sdk 0.48.0__py3-none-any.whl → 0.49.0__py3-none-any.whl

databricks/sdk/service/sql.py

@@ -3161,6 +3161,123 @@ class ExternalLink:
         )
 
 
+@dataclass
+class ExternalQuerySource:
+    alert_id: Optional[str] = None
+    """The canonical identifier for this SQL alert"""
+
+    dashboard_id: Optional[str] = None
+    """The canonical identifier for this Lakeview dashboard"""
+
+    genie_space_id: Optional[str] = None
+    """The canonical identifier for this Genie space"""
+
+    job_info: Optional[ExternalQuerySourceJobInfo] = None
+
+    legacy_dashboard_id: Optional[str] = None
+    """The canonical identifier for this legacy dashboard"""
+
+    notebook_id: Optional[str] = None
+    """The canonical identifier for this notebook"""
+
+    sql_query_id: Optional[str] = None
+    """The canonical identifier for this SQL query"""
+
+    def as_dict(self) -> dict:
+        """Serializes the ExternalQuerySource into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.alert_id is not None:
+            body["alert_id"] = self.alert_id
+        if self.dashboard_id is not None:
+            body["dashboard_id"] = self.dashboard_id
+        if self.genie_space_id is not None:
+            body["genie_space_id"] = self.genie_space_id
+        if self.job_info:
+            body["job_info"] = self.job_info.as_dict()
+        if self.legacy_dashboard_id is not None:
+            body["legacy_dashboard_id"] = self.legacy_dashboard_id
+        if self.notebook_id is not None:
+            body["notebook_id"] = self.notebook_id
+        if self.sql_query_id is not None:
+            body["sql_query_id"] = self.sql_query_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the ExternalQuerySource into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.alert_id is not None:
+            body["alert_id"] = self.alert_id
+        if self.dashboard_id is not None:
+            body["dashboard_id"] = self.dashboard_id
+        if self.genie_space_id is not None:
+            body["genie_space_id"] = self.genie_space_id
+        if self.job_info:
+            body["job_info"] = self.job_info
+        if self.legacy_dashboard_id is not None:
+            body["legacy_dashboard_id"] = self.legacy_dashboard_id
+        if self.notebook_id is not None:
+            body["notebook_id"] = self.notebook_id
+        if self.sql_query_id is not None:
+            body["sql_query_id"] = self.sql_query_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> ExternalQuerySource:
+        """Deserializes the ExternalQuerySource from a dictionary."""
+        return cls(
+            alert_id=d.get("alert_id", None),
+            dashboard_id=d.get("dashboard_id", None),
+            genie_space_id=d.get("genie_space_id", None),
+            job_info=_from_dict(d, "job_info", ExternalQuerySourceJobInfo),
+            legacy_dashboard_id=d.get("legacy_dashboard_id", None),
+            notebook_id=d.get("notebook_id", None),
+            sql_query_id=d.get("sql_query_id", None),
+        )
+
+
+@dataclass
+class ExternalQuerySourceJobInfo:
+    job_id: Optional[str] = None
+    """The canonical identifier for this job."""
+
+    job_run_id: Optional[str] = None
+    """The canonical identifier of the run. This ID is unique across all runs of all jobs."""
+
+    job_task_run_id: Optional[str] = None
+    """The canonical identifier of the task run."""
+
+    def as_dict(self) -> dict:
+        """Serializes the ExternalQuerySourceJobInfo into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        if self.job_id is not None:
+            body["job_id"] = self.job_id
+        if self.job_run_id is not None:
+            body["job_run_id"] = self.job_run_id
+        if self.job_task_run_id is not None:
+            body["job_task_run_id"] = self.job_task_run_id
+        return body
+
+    def as_shallow_dict(self) -> dict:
+        """Serializes the ExternalQuerySourceJobInfo into a shallow dictionary of its immediate attributes."""
+        body = {}
+        if self.job_id is not None:
+            body["job_id"] = self.job_id
+        if self.job_run_id is not None:
+            body["job_run_id"] = self.job_run_id
+        if self.job_task_run_id is not None:
+            body["job_task_run_id"] = self.job_task_run_id
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, Any]) -> ExternalQuerySourceJobInfo:
+        """Deserializes the ExternalQuerySourceJobInfo from a dictionary."""
+        return cls(
+            job_id=d.get("job_id", None),
+            job_run_id=d.get("job_run_id", None),
+            job_task_run_id=d.get("job_task_run_id", None),
+        )
+
+
 class Format(Enum):
 
     ARROW_STREAM = "ARROW_STREAM"
@@ -5113,6 +5230,11 @@ class QueryInfo:
     query_id: Optional[str] = None
     """The query ID."""
 
+    query_source: Optional[ExternalQuerySource] = None
+    """A struct that contains key-value pairs representing Databricks entities that were involved in
+    the execution of this statement, such as jobs, notebooks, or dashboards. This field only records
+    Databricks entities."""
+
     query_start_time_ms: Optional[int] = None
     """The time the query started."""
 
@@ -5173,6 +5295,8 @@ class QueryInfo:
             body["query_end_time_ms"] = self.query_end_time_ms
         if self.query_id is not None:
             body["query_id"] = self.query_id
+        if self.query_source:
+            body["query_source"] = self.query_source.as_dict()
         if self.query_start_time_ms is not None:
             body["query_start_time_ms"] = self.query_start_time_ms
         if self.query_text is not None:
@@ -5222,6 +5346,8 @@ class QueryInfo:
             body["query_end_time_ms"] = self.query_end_time_ms
         if self.query_id is not None:
             body["query_id"] = self.query_id
+        if self.query_source:
+            body["query_source"] = self.query_source
         if self.query_start_time_ms is not None:
             body["query_start_time_ms"] = self.query_start_time_ms
         if self.query_text is not None:
@@ -5259,6 +5385,7 @@ class QueryInfo:
             plans_state=_enum(d, "plans_state", PlansState),
             query_end_time_ms=d.get("query_end_time_ms", None),
             query_id=d.get("query_id", None),
+            query_source=_from_dict(d, "query_source", ExternalQuerySource),
             query_start_time_ms=d.get("query_start_time_ms", None),
             query_text=d.get("query_text", None),
             rows_produced=d.get("rows_produced", None),

databricks/sdk/version.py

@@ -1 +1 @@
-__version__ = "0.48.0"
+__version__ = "0.49.0"

{databricks_sdk-0.48.0.dist-info → databricks_sdk-0.49.0.dist-info}/METADATA

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

{databricks_sdk-0.48.0.dist-info → databricks_sdk-0.49.0.dist-info}/RECORD

@@ -15,7 +15,7 @@ databricks/sdk/oauth.py,sha256=8Cod3Fa4gPuVhd7zFBkVOVLB1fwiAKkyibU6_wnpqDE,28411
 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=bkYe4lEQZCEmFm0XRZaZkxTV1niMqR_lbp-tzKL6s6c,23
+databricks/sdk/version.py,sha256=09FNUq8XbPsGZin0IRhNT7qwi97ZXQafNn7GixZiqAw,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
@@ -42,29 +42,29 @@ 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/apps.py,sha256=NdPhpycOpsYlXlBB5JTvt-AmiaCxIqFKMz7370CKKQo,54550
-databricks/sdk/service/billing.py,sha256=s-QN57TyMMfJSA40CfIjjno8iWHBssjpxcy0xcEcwyw,98590
-databricks/sdk/service/catalog.py,sha256=eu1zLzoeQxirFPMbthS5d5Mt0EjdbuqYP4sESotNwHc,602387
+databricks/sdk/service/apps.py,sha256=Qe5WLmtvGJQdMXRQm4-huR7_wudAfW0971IBE4_dezg,56078
+databricks/sdk/service/billing.py,sha256=s78zrScWy3lGAzxgpWbHGZO0GOfbpS9u9C7jq1Nalbs,98668
+databricks/sdk/service/catalog.py,sha256=MMfwK42yyU13KPvaLVa4ZmWtEzb23crY7tVLv7UoAmA,602472
 databricks/sdk/service/cleanrooms.py,sha256=TON2V0-T_TOk31K9bUypuRJaAJhHMM0bQDbMjJWlCDQ,58613
-databricks/sdk/service/compute.py,sha256=R8GilWUVYVljZHfdNuiP4AJiAFWqKG4_XKn6Wg-um68,553503
-databricks/sdk/service/dashboards.py,sha256=MhKFjglM4odyOiNzagII4pprkPMh8FOfzvwm9Ntlyr8,105263
+databricks/sdk/service/compute.py,sha256=NaZitVVTvLZNb_7Fl8I3mlwp5PJo8O5zgZtks4HlQEM,565066
+databricks/sdk/service/dashboards.py,sha256=pmvLwPn6pL87WH5r-17phS2d69XHsJMiWgDQVhO7yBU,108290
 databricks/sdk/service/files.py,sha256=gWjtORuRe7B1VZol0rr_dcSHq1gMmuKszGHaMqunslU,46234
-databricks/sdk/service/iam.py,sha256=TFKlhh7A1INGVcfAZQ1f0ORpAIzWLJ78l23vkerTe4M,174846
-databricks/sdk/service/jobs.py,sha256=2DQIqaeYq2tOeRzeeq23Q8_ntXaeJNK0LHhlcbd_8A4,443424
-databricks/sdk/service/marketplace.py,sha256=h-hGo2GHrjbVe0lXuengvqIDEN5VGhBdna4di9at5hA,175736
-databricks/sdk/service/ml.py,sha256=q5vdOguyLiTpGqT6R5w41l_cCendUx4Hm-fQeW1-08k,316482
-databricks/sdk/service/oauth2.py,sha256=LLp1uMvxfPOXzuekLjnLH6si6YOCvie7qZSpjSNDYd4,79788
-databricks/sdk/service/pipelines.py,sha256=_2RpczckyhexbCtLoB_ewVKgZNFseN8bCwABod6YUqs,166666
+databricks/sdk/service/iam.py,sha256=23mpLuwFwNAwK632y11dI7TH3jXmShJLiDx_KxKt6JQ,173744
+databricks/sdk/service/jobs.py,sha256=iipYTJyIULZW1IIhpkYfgvVqhPgQRD6UBAa2S60dUjE,443450
+databricks/sdk/service/marketplace.py,sha256=NLCJ7zb0Oj5nhLth9R1Szuz6fSSoxAnsKvFhKgx994s,175741
+databricks/sdk/service/ml.py,sha256=cnBxWeSjXwNWwojT90d7aXI1cXu9QocB9NWleK6vFU0,314696
+databricks/sdk/service/oauth2.py,sha256=EKptQcQnTX0M4JDLfzwTKekjWLDS_LPIwZdTXv6-YwU,80467
+databricks/sdk/service/pipelines.py,sha256=tTwC6_Benn8CEzstWvx08XGZa5tugkNouS3_i6sHQwU,169459
 databricks/sdk/service/provisioning.py,sha256=-Ly2o02i-jhNmiP9zLPeYF8H2usoB-oTG0RLF5gkIpc,169311
-databricks/sdk/service/serving.py,sha256=eHPRoxeOpqV-dyVf7V8_6_FHBoTTwXqr5W9Y6b_iZtI,204406
-databricks/sdk/service/settings.py,sha256=9I8NHnVTtTn_u91fYQi-QpDMjKPF2c2Sec0gEub6on0,323112
-databricks/sdk/service/sharing.py,sha256=jP3XWl3qojFWmZE9xkgCixqEZPuGZAJC_M-eGXMrPQs,142473
-databricks/sdk/service/sql.py,sha256=2t8QhrEUyLekRYHnRYKkxVY8qpDbxE7JayfMSds6OKM,409593
+databricks/sdk/service/serving.py,sha256=O6gE7xgLhhXPlrSgbN--7EdBhFI9RxMOdFW-ONk8Rew,213071
+databricks/sdk/service/settings.py,sha256=fHWRL_eIFifePgFC8L7lQdpjY_WEr2ce270DzPq6Nvw,323366
+databricks/sdk/service/sharing.py,sha256=IV-rws5EzsHFfyjSBlbPEDQKZvbH4OgDp9Yxg9Hyrwk,142557
+databricks/sdk/service/sql.py,sha256=uaG9IA7ngz36EaDB0zV1YVAn4SJTEWEXJiR7jNPTpmI,414703
 databricks/sdk/service/vectorsearch.py,sha256=stdRmV-iE8L0rGzhSSG5tmoVRXb9bPItSE_ss6HQVDo,79897
 databricks/sdk/service/workspace.py,sha256=T0ZbnG1qcPjKysGO_tBzl5x1PyalydeYJRBZbooYNm0,130893
-databricks_sdk-0.48.0.dist-info/licenses/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
-databricks_sdk-0.48.0.dist-info/licenses/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
-databricks_sdk-0.48.0.dist-info/METADATA,sha256=QVB1xlNzOVnFrOEVSKUVtW-ZIf9GAoGBzH_HdPwl3sA,38334
-databricks_sdk-0.48.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-databricks_sdk-0.48.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
-databricks_sdk-0.48.0.dist-info/RECORD,,
+databricks_sdk-0.49.0.dist-info/licenses/LICENSE,sha256=afBgTZo-JsYqj4VOjnejBetMuHKcFR30YobDdpVFkqY,11411
+databricks_sdk-0.49.0.dist-info/licenses/NOTICE,sha256=tkRcQYA1k68wDLcnOWbg2xJDsUOJw8G8DGBhb8dnI3w,1588
+databricks_sdk-0.49.0.dist-info/METADATA,sha256=AFKsQih93i-pdkWS9ZmIPKwwzqyCOqx0B5SuTnQ0vEM,38334
+databricks_sdk-0.49.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+databricks_sdk-0.49.0.dist-info/top_level.txt,sha256=7kRdatoSgU0EUurRQJ_3F1Nv4EOSHWAr6ng25tJOJKU,11
+databricks_sdk-0.49.0.dist-info/RECORD,,