databricks-sdk 0.19.1__py3-none-any.whl → 0.21.0__py3-none-any.whl

databricks/sdk/service/oauth2.py

@@ -160,6 +160,48 @@ class CreateServicePrincipalSecretResponse:
                    update_time=d.get('update_time', None))
 
 
+@dataclass
+class DeleteCustomAppIntegrationOutput:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteCustomAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteCustomAppIntegrationOutput:
+        """Deserializes the DeleteCustomAppIntegrationOutput from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeletePublishedAppIntegrationOutput:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeletePublishedAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeletePublishedAppIntegrationOutput:
+        """Deserializes the DeletePublishedAppIntegrationOutput from a dictionary."""
+        return cls()
+
+
+@dataclass
+class DeleteResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeleteResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeleteResponse:
+        """Deserializes the DeleteResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class GetCustomAppIntegrationOutput:
     client_id: Optional[str] = None
@@ -445,6 +487,20 @@ class UpdateCustomAppIntegration:
                    token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy))
 
 
+@dataclass
+class UpdateCustomAppIntegrationOutput:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdateCustomAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdateCustomAppIntegrationOutput:
+        """Deserializes the UpdateCustomAppIntegrationOutput from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UpdatePublishedAppIntegration:
     integration_id: Optional[str] = None
@@ -467,6 +523,20 @@ class UpdatePublishedAppIntegration:
                    token_access_policy=_from_dict(d, 'token_access_policy', TokenAccessPolicy))
 
 
+@dataclass
+class UpdatePublishedAppIntegrationOutput:
+
+    def as_dict(self) -> dict:
+        """Serializes the UpdatePublishedAppIntegrationOutput into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> UpdatePublishedAppIntegrationOutput:
+        """Deserializes the UpdatePublishedAppIntegrationOutput from a dictionary."""
+        return cls()
+
+
 class CustomAppIntegrationAPI:
     """These APIs enable administrators to manage custom oauth app integrations, which is required for
     adding/using Custom OAuth App Integration like Tableau Cloud for Databricks in AWS cloud."""
@@ -508,6 +578,7 @@ class CustomAppIntegrationAPI:
         if scopes is not None: body['scopes'] = [v for v in scopes]
         if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations',
                            body=body,
@@ -527,6 +598,7 @@ class CustomAppIntegrationAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do(
             'DELETE',
             f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}',
@@ -544,6 +616,7 @@ class CustomAppIntegrationAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do(
             'GET',
             f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}',
@@ -559,6 +632,7 @@ class CustomAppIntegrationAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET',
                             f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations',
                             headers=headers)
@@ -588,6 +662,7 @@ class CustomAppIntegrationAPI:
         if redirect_urls is not None: body['redirect_urls'] = [v for v in redirect_urls]
         if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do(
             'PATCH',
             f'/api/2.0/accounts/{self._api.account_id}/oauth2/custom-app-integrations/{integration_id}',
@@ -666,6 +741,7 @@ class PublishedAppIntegrationAPI:
         if app_id is not None: body['app_id'] = app_id
         if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST',
                            f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations',
                            body=body,
@@ -685,6 +761,7 @@ class PublishedAppIntegrationAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do(
             'DELETE',
             f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}',
@@ -702,6 +779,7 @@ class PublishedAppIntegrationAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do(
             'GET',
             f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}',
@@ -717,6 +795,7 @@ class PublishedAppIntegrationAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         json = self._api.do('GET',
                             f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations',
                             headers=headers)
@@ -739,6 +818,7 @@ class PublishedAppIntegrationAPI:
         body = {}
         if token_access_policy is not None: body['token_access_policy'] = token_access_policy.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do(
             'PATCH',
             f'/api/2.0/accounts/{self._api.account_id}/oauth2/published-app-integrations/{integration_id}',
@@ -774,6 +854,7 @@ class ServicePrincipalSecretsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do(
             'POST',
             f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets',
@@ -794,6 +875,7 @@ class ServicePrincipalSecretsAPI:
         """
 
         headers = {}
+
         self._api.do(
             'DELETE',
             f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets/{secret_id}',
@@ -812,6 +894,7 @@ class ServicePrincipalSecretsAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         json = self._api.do(
             'GET',
             f'/api/2.0/accounts/{self._api.account_id}/servicePrincipals/{service_principal_id}/credentials/secrets',

databricks/sdk/service/pipelines.py

@@ -8,7 +8,7 @@ import time
 from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
-from typing import Any, Callable, Dict, Iterator, List, Optional
+from typing import Callable, Dict, Iterator, List, Optional
 
 from ..errors import OperationFailed
 from ._internal import Wait, _enum, _from_dict, _repeated_dict
@@ -178,14 +178,14 @@ class DataPlaneId:
     instance: Optional[str] = None
     """The instance name of the data plane emitting an event."""
 
-    seq_no: Optional[Any] = None
+    seq_no: Optional[int] = None
     """A sequence number, unique and increasing within the data plane instance."""
 
     def as_dict(self) -> dict:
         """Serializes the DataPlaneId into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.instance is not None: body['instance'] = self.instance
-        if self.seq_no: body['seq_no'] = self.seq_no
+        if self.seq_no is not None: body['seq_no'] = self.seq_no
         return body
 
     @classmethod
@@ -194,6 +194,20 @@ class DataPlaneId:
         return cls(instance=d.get('instance', None), seq_no=d.get('seq_no', None))
 
 
+@dataclass
+class DeletePipelineResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the DeletePipelineResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> DeletePipelineResponse:
+        """Deserializes the DeletePipelineResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class EditPipeline:
     allow_duplicate_names: Optional[bool] = None
@@ -313,6 +327,20 @@ class EditPipeline:
                    trigger=_from_dict(d, 'trigger', PipelineTrigger))
 
 
+@dataclass
+class EditPipelineResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the EditPipelineResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> EditPipelineResponse:
+        """Deserializes the EditPipelineResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class ErrorDetail:
     exceptions: Optional[List[SerializedException]] = None
@@ -566,6 +594,20 @@ class ListUpdatesResponse:
                    updates=_repeated_dict(d, 'updates', UpdateInfo))
 
 
+@dataclass
+class ManualTrigger:
+
+    def as_dict(self) -> dict:
+        """Serializes the ManualTrigger into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> ManualTrigger:
+        """Deserializes the ManualTrigger from a dictionary."""
+        return cls()
+
+
 class MaturityLevel(Enum):
     """Maturity level for EventDetails."""
 
@@ -1335,19 +1377,19 @@ class PipelineStateInfo:
 class PipelineTrigger:
     cron: Optional[CronTrigger] = None
 
-    manual: Optional[Any] = None
+    manual: Optional[ManualTrigger] = None
 
     def as_dict(self) -> dict:
         """Serializes the PipelineTrigger into a dictionary suitable for use as a JSON request body."""
         body = {}
         if self.cron: body['cron'] = self.cron.as_dict()
-        if self.manual: body['manual'] = self.manual
+        if self.manual: body['manual'] = self.manual.as_dict()
         return body
 
     @classmethod
     def from_dict(cls, d: Dict[str, any]) -> PipelineTrigger:
         """Deserializes the PipelineTrigger from a dictionary."""
-        return cls(cron=_from_dict(d, 'cron', CronTrigger), manual=d.get('manual', None))
+        return cls(cron=_from_dict(d, 'cron', CronTrigger), manual=_from_dict(d, 'manual', ManualTrigger))
 
 
 @dataclass
@@ -1503,6 +1545,20 @@ class StartUpdateResponse:
         return cls(update_id=d.get('update_id', None))
 
 
+@dataclass
+class StopPipelineResponse:
+
+    def as_dict(self) -> dict:
+        """Serializes the StopPipelineResponse into a dictionary suitable for use as a JSON request body."""
+        body = {}
+        return body
+
+    @classmethod
+    def from_dict(cls, d: Dict[str, any]) -> StopPipelineResponse:
+        """Deserializes the StopPipelineResponse from a dictionary."""
+        return cls()
+
+
 @dataclass
 class UpdateInfo:
     cause: Optional[UpdateInfoCause] = None
@@ -1810,6 +1866,7 @@ class PipelinesAPI:
         if target is not None: body['target'] = target
         if trigger is not None: body['trigger'] = trigger.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', '/api/2.0/pipelines', body=body, headers=headers)
         return CreatePipelineResponse.from_dict(res)
 
@@ -1824,6 +1881,7 @@ class PipelinesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         self._api.do('DELETE', f'/api/2.0/pipelines/{pipeline_id}', headers=headers)
 
     def get(self, pipeline_id: str) -> GetPipelineResponse:
@@ -1835,6 +1893,7 @@ class PipelinesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/pipelines/{pipeline_id}', headers=headers)
         return GetPipelineResponse.from_dict(res)
 
@@ -1850,6 +1909,7 @@ class PipelinesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET',
                            f'/api/2.0/permissions/pipelines/{pipeline_id}/permissionLevels',
                            headers=headers)
@@ -1867,6 +1927,7 @@ class PipelinesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/permissions/pipelines/{pipeline_id}', headers=headers)
         return PipelinePermissions.from_dict(res)
 
@@ -1884,6 +1945,7 @@ class PipelinesAPI:
         """
 
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/pipelines/{pipeline_id}/updates/{update_id}', headers=headers)
         return GetUpdateResponse.from_dict(res)
 
@@ -2015,6 +2077,7 @@ class PipelinesAPI:
         if page_token is not None: query['page_token'] = page_token
         if until_update_id is not None: query['until_update_id'] = until_update_id
         headers = {'Accept': 'application/json', }
+
         res = self._api.do('GET', f'/api/2.0/pipelines/{pipeline_id}/updates', query=query, headers=headers)
         return ListUpdatesResponse.from_dict(res)
 
@@ -2037,6 +2100,7 @@ class PipelinesAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PUT', f'/api/2.0/permissions/pipelines/{pipeline_id}', body=body, headers=headers)
         return PipelinePermissions.from_dict(res)
 
@@ -2079,6 +2143,7 @@ class PipelinesAPI:
         if refresh_selection is not None: body['refresh_selection'] = [v for v in refresh_selection]
         if validate_only is not None: body['validate_only'] = validate_only
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('POST', f'/api/2.0/pipelines/{pipeline_id}/updates', body=body, headers=headers)
         return StartUpdateResponse.from_dict(res)
 
@@ -2096,8 +2161,11 @@ class PipelinesAPI:
         """
 
         headers = {'Accept': 'application/json', }
-        self._api.do('POST', f'/api/2.0/pipelines/{pipeline_id}/stop', headers=headers)
-        return Wait(self.wait_get_pipeline_idle, pipeline_id=pipeline_id)
+
+        op_response = self._api.do('POST', f'/api/2.0/pipelines/{pipeline_id}/stop', headers=headers)
+        return Wait(self.wait_get_pipeline_idle,
+                    response=StopPipelineResponse.from_dict(op_response),
+                    pipeline_id=pipeline_id)
 
     def stop_and_wait(self, pipeline_id: str, timeout=timedelta(minutes=20)) -> GetPipelineResponse:
         return self.stop(pipeline_id=pipeline_id).result(timeout=timeout)
@@ -2196,6 +2264,7 @@ class PipelinesAPI:
         if target is not None: body['target'] = target
         if trigger is not None: body['trigger'] = trigger.as_dict()
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         self._api.do('PUT', f'/api/2.0/pipelines/{pipeline_id}', body=body, headers=headers)
 
     def update_permissions(
@@ -2217,6 +2286,7 @@ class PipelinesAPI:
         if access_control_list is not None:
             body['access_control_list'] = [v.as_dict() for v in access_control_list]
         headers = {'Accept': 'application/json', 'Content-Type': 'application/json', }
+
         res = self._api.do('PATCH',
                            f'/api/2.0/permissions/pipelines/{pipeline_id}',
                            body=body,