remove andwait functions

Author: parthban-db

databricks/sdk/apps/v2/impl.py

@@ -4,7 +4,6 @@
 
 import logging
 from dataclasses import dataclass
-from datetime import timedelta
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
@@ -1094,11 +1093,6 @@ def create(self, *, app: Optional[App] = None, no_compute: Optional[bool] = None
         op_response = self._api.do("POST", "/api/2.0/apps", query=query, body=body, headers=headers)
         return Wait(self.WaitGetAppActive, response=App.from_dict(op_response), name=op_response["name"])
 
-    def create_and_wait(
-        self, *, app: Optional[App] = None, no_compute: Optional[bool] = None, timeout=timedelta(minutes=20)
-    ) -> App:
-        return self.create(app=app, no_compute=no_compute).result(timeout=timeout)
-
     def delete(self, name: str) -> App:
         """Delete an app.
 
@@ -1144,11 +1138,6 @@ def deploy(self, app_name: str, *, app_deployment: Optional[AppDeployment] = Non
             deployment_id=op_response["deployment_id"],
         )
 
-    def deploy_and_wait(
-        self, app_name: str, *, app_deployment: Optional[AppDeployment] = None, timeout=timedelta(minutes=20)
-    ) -> AppDeployment:
-        return self.deploy(app_deployment=app_deployment, app_name=app_name).result(timeout=timeout)
-
     def get(self, name: str) -> App:
         """Get an app.
 
@@ -1335,9 +1324,6 @@ def start(self, name: str) -> Wait[App]:
         op_response = self._api.do("POST", f"/api/2.0/apps/{name}/start", headers=headers)
         return Wait(self.WaitGetAppActive, response=App.from_dict(op_response), name=op_response["name"])
 
-    def start_and_wait(self, name: str, timeout=timedelta(minutes=20)) -> App:
-        return self.start(name=name).result(timeout=timeout)
-
     def stop(self, name: str) -> Wait[App]:
         """Stop an app.
 
@@ -1359,9 +1345,6 @@ def stop(self, name: str) -> Wait[App]:
         op_response = self._api.do("POST", f"/api/2.0/apps/{name}/stop", headers=headers)
         return Wait(self.WaitGetAppStopped, response=App.from_dict(op_response), name=op_response["name"])
 
-    def stop_and_wait(self, name: str, timeout=timedelta(minutes=20)) -> App:
-        return self.stop(name=name).result(timeout=timeout)
-
     def update(self, name: str, *, app: Optional[App] = None) -> App:
         """Update an app.
 

databricks/sdk/catalog/v2/impl.py

@@ -4,7 +4,6 @@
 
 import logging
 from dataclasses import dataclass
-from datetime import timedelta
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
@@ -12311,9 +12310,6 @@ def create(self, *, table: Optional[OnlineTable] = None) -> Wait[OnlineTable]:
             self.WaitGetOnlineTableActive, response=OnlineTable.from_dict(op_response), name=op_response["name"]
         )
 
-    def create_and_wait(self, *, table: Optional[OnlineTable] = None, timeout=timedelta(minutes=20)) -> OnlineTable:
-        return self.create(table=table).result(timeout=timeout)
-
     def delete(self, name: str):
         """Delete an Online Table.
 

databricks/sdk/compute/v2/impl.py

@@ -4,7 +4,6 @@
 
 import logging
 from dataclasses import dataclass
-from datetime import timedelta
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
@@ -10307,76 +10306,6 @@ def create(
             cluster_id=op_response["cluster_id"],
         )
 
-    def create_and_wait(
-        self,
-        spark_version: str,
-        *,
-        apply_policy_default_values: Optional[bool] = None,
-        autoscale: Optional[AutoScale] = None,
-        autotermination_minutes: Optional[int] = None,
-        aws_attributes: Optional[AwsAttributes] = None,
-        azure_attributes: Optional[AzureAttributes] = None,
-        clone_from: Optional[CloneCluster] = None,
-        cluster_log_conf: Optional[ClusterLogConf] = None,
-        cluster_name: Optional[str] = None,
-        custom_tags: Optional[Dict[str, str]] = None,
-        data_security_mode: Optional[DataSecurityMode] = None,
-        docker_image: Optional[DockerImage] = None,
-        driver_instance_pool_id: Optional[str] = None,
-        driver_node_type_id: Optional[str] = None,
-        enable_elastic_disk: Optional[bool] = None,
-        enable_local_disk_encryption: Optional[bool] = None,
-        gcp_attributes: Optional[GcpAttributes] = None,
-        init_scripts: Optional[List[InitScriptInfo]] = None,
-        instance_pool_id: Optional[str] = None,
-        is_single_node: Optional[bool] = None,
-        kind: Optional[Kind] = None,
-        node_type_id: Optional[str] = None,
-        num_workers: Optional[int] = None,
-        policy_id: Optional[str] = None,
-        runtime_engine: Optional[RuntimeEngine] = None,
-        single_user_name: Optional[str] = None,
-        spark_conf: Optional[Dict[str, str]] = None,
-        spark_env_vars: Optional[Dict[str, str]] = None,
-        ssh_public_keys: Optional[List[str]] = None,
-        use_ml_runtime: Optional[bool] = None,
-        workload_type: Optional[WorkloadType] = None,
-        timeout=timedelta(minutes=20),
-    ) -> ClusterDetails:
-        return self.create(
-            apply_policy_default_values=apply_policy_default_values,
-            autoscale=autoscale,
-            autotermination_minutes=autotermination_minutes,
-            aws_attributes=aws_attributes,
-            azure_attributes=azure_attributes,
-            clone_from=clone_from,
-            cluster_log_conf=cluster_log_conf,
-            cluster_name=cluster_name,
-            custom_tags=custom_tags,
-            data_security_mode=data_security_mode,
-            docker_image=docker_image,
-            driver_instance_pool_id=driver_instance_pool_id,
-            driver_node_type_id=driver_node_type_id,
-            enable_elastic_disk=enable_elastic_disk,
-            enable_local_disk_encryption=enable_local_disk_encryption,
-            gcp_attributes=gcp_attributes,
-            init_scripts=init_scripts,
-            instance_pool_id=instance_pool_id,
-            is_single_node=is_single_node,
-            kind=kind,
-            node_type_id=node_type_id,
-            num_workers=num_workers,
-            policy_id=policy_id,
-            runtime_engine=runtime_engine,
-            single_user_name=single_user_name,
-            spark_conf=spark_conf,
-            spark_env_vars=spark_env_vars,
-            spark_version=spark_version,
-            ssh_public_keys=ssh_public_keys,
-            use_ml_runtime=use_ml_runtime,
-            workload_type=workload_type,
-        ).result(timeout=timeout)
-
     def delete(self, cluster_id: str) -> Wait[ClusterDetails]:
         """Terminate cluster.
 
@@ -10404,9 +10333,6 @@ def delete(self, cluster_id: str) -> Wait[ClusterDetails]:
             self.WaitGetClusterTerminated, response=DeleteClusterResponse.from_dict(op_response), cluster_id=cluster_id
         )
 
-    def delete_and_wait(self, cluster_id: str, timeout=timedelta(minutes=20)) -> ClusterDetails:
-        return self.delete(cluster_id=cluster_id).result(timeout=timeout)
-
     def edit(
         self,
         cluster_id: str,
@@ -10695,76 +10621,6 @@ def edit(
             self.WaitGetClusterRunning, response=EditClusterResponse.from_dict(op_response), cluster_id=cluster_id
         )
 
-    def edit_and_wait(
-        self,
-        cluster_id: str,
-        spark_version: str,
-        *,
-        apply_policy_default_values: Optional[bool] = None,
-        autoscale: Optional[AutoScale] = None,
-        autotermination_minutes: Optional[int] = None,
-        aws_attributes: Optional[AwsAttributes] = None,
-        azure_attributes: Optional[AzureAttributes] = None,
-        cluster_log_conf: Optional[ClusterLogConf] = None,
-        cluster_name: Optional[str] = None,
-        custom_tags: Optional[Dict[str, str]] = None,
-        data_security_mode: Optional[DataSecurityMode] = None,
-        docker_image: Optional[DockerImage] = None,
-        driver_instance_pool_id: Optional[str] = None,
-        driver_node_type_id: Optional[str] = None,
-        enable_elastic_disk: Optional[bool] = None,
-        enable_local_disk_encryption: Optional[bool] = None,
-        gcp_attributes: Optional[GcpAttributes] = None,
-        init_scripts: Optional[List[InitScriptInfo]] = None,
-        instance_pool_id: Optional[str] = None,
-        is_single_node: Optional[bool] = None,
-        kind: Optional[Kind] = None,
-        node_type_id: Optional[str] = None,
-        num_workers: Optional[int] = None,
-        policy_id: Optional[str] = None,
-        runtime_engine: Optional[RuntimeEngine] = None,
-        single_user_name: Optional[str] = None,
-        spark_conf: Optional[Dict[str, str]] = None,
-        spark_env_vars: Optional[Dict[str, str]] = None,
-        ssh_public_keys: Optional[List[str]] = None,
-        use_ml_runtime: Optional[bool] = None,
-        workload_type: Optional[WorkloadType] = None,
-        timeout=timedelta(minutes=20),
-    ) -> ClusterDetails:
-        return self.edit(
-            apply_policy_default_values=apply_policy_default_values,
-            autoscale=autoscale,
-            autotermination_minutes=autotermination_minutes,
-            aws_attributes=aws_attributes,
-            azure_attributes=azure_attributes,
-            cluster_id=cluster_id,
-            cluster_log_conf=cluster_log_conf,
-            cluster_name=cluster_name,
-            custom_tags=custom_tags,
-            data_security_mode=data_security_mode,
-            docker_image=docker_image,
-            driver_instance_pool_id=driver_instance_pool_id,
-            driver_node_type_id=driver_node_type_id,
-            enable_elastic_disk=enable_elastic_disk,
-            enable_local_disk_encryption=enable_local_disk_encryption,
-            gcp_attributes=gcp_attributes,
-            init_scripts=init_scripts,
-            instance_pool_id=instance_pool_id,
-            is_single_node=is_single_node,
-            kind=kind,
-            node_type_id=node_type_id,
-            num_workers=num_workers,
-            policy_id=policy_id,
-            runtime_engine=runtime_engine,
-            single_user_name=single_user_name,
-            spark_conf=spark_conf,
-            spark_env_vars=spark_env_vars,
-            spark_version=spark_version,
-            ssh_public_keys=ssh_public_keys,
-            use_ml_runtime=use_ml_runtime,
-            workload_type=workload_type,
-        ).result(timeout=timeout)
-
     def events(
         self,
         cluster_id: str,
@@ -11055,16 +10911,6 @@ def resize(
             self.WaitGetClusterRunning, response=ResizeClusterResponse.from_dict(op_response), cluster_id=cluster_id
         )
 
-    def resize_and_wait(
-        self,
-        cluster_id: str,
-        *,
-        autoscale: Optional[AutoScale] = None,
-        num_workers: Optional[int] = None,
-        timeout=timedelta(minutes=20),
-    ) -> ClusterDetails:
-        return self.resize(autoscale=autoscale, cluster_id=cluster_id, num_workers=num_workers).result(timeout=timeout)
-
     def restart(self, cluster_id: str, *, restart_user: Optional[str] = None) -> Wait[ClusterDetails]:
         """Restart cluster.
 
@@ -11094,11 +10940,6 @@ def restart(self, cluster_id: str, *, restart_user: Optional[str] = None) -> Wai
             self.WaitGetClusterRunning, response=RestartClusterResponse.from_dict(op_response), cluster_id=cluster_id
         )
 
-    def restart_and_wait(
-        self, cluster_id: str, *, restart_user: Optional[str] = None, timeout=timedelta(minutes=20)
-    ) -> ClusterDetails:
-        return self.restart(cluster_id=cluster_id, restart_user=restart_user).result(timeout=timeout)
-
     def set_permissions(
         self, cluster_id: str, *, access_control_list: Optional[List[ClusterAccessControlRequest]] = None
     ) -> ClusterPermissions:
@@ -11168,9 +11009,6 @@ def start(self, cluster_id: str) -> Wait[ClusterDetails]:
             self.WaitGetClusterRunning, response=StartClusterResponse.from_dict(op_response), cluster_id=cluster_id
         )
 
-    def start_and_wait(self, cluster_id: str, timeout=timedelta(minutes=20)) -> ClusterDetails:
-        return self.start(cluster_id=cluster_id).result(timeout=timeout)
-
     def unpin(self, cluster_id: str):
         """Unpin cluster.
 
@@ -11245,16 +11083,6 @@ def update(
             self.WaitGetClusterRunning, response=UpdateClusterResponse.from_dict(op_response), cluster_id=cluster_id
         )
 
-    def update_and_wait(
-        self,
-        cluster_id: str,
-        update_mask: str,
-        *,
-        cluster: Optional[UpdateClusterResource] = None,
-        timeout=timedelta(minutes=20),
-    ) -> ClusterDetails:
-        return self.update(cluster=cluster, cluster_id=cluster_id, update_mask=update_mask).result(timeout=timeout)
-
     def update_permissions(
         self, cluster_id: str, *, access_control_list: Optional[List[ClusterAccessControlRequest]] = None
     ) -> ClusterPermissions:
@@ -11325,16 +11153,6 @@ def cancel(
             context_id=context_id,
         )
 
-    def cancel_and_wait(
-        self,
-        *,
-        cluster_id: Optional[str] = None,
-        command_id: Optional[str] = None,
-        context_id: Optional[str] = None,
-        timeout=timedelta(minutes=20),
-    ) -> CommandStatusResponse:
-        return self.cancel(cluster_id=cluster_id, command_id=command_id, context_id=context_id).result(timeout=timeout)
-
     def command_status(self, cluster_id: str, context_id: str, command_id: str) -> CommandStatusResponse:
         """Get command info.
 
@@ -11421,11 +11239,6 @@ def create(
             context_id=op_response["id"],
         )
 
-    def create_and_wait(
-        self, *, cluster_id: Optional[str] = None, language: Optional[Language] = None, timeout=timedelta(minutes=20)
-    ) -> ContextStatusResponse:
-        return self.create(cluster_id=cluster_id, language=language).result(timeout=timeout)
-
     def destroy(self, cluster_id: str, context_id: str):
         """Delete an execution context.
 
@@ -11497,19 +11310,6 @@ def execute(
             context_id=context_id,
         )
 
-    def execute_and_wait(
-        self,
-        *,
-        cluster_id: Optional[str] = None,
-        command: Optional[str] = None,
-        context_id: Optional[str] = None,
-        language: Optional[Language] = None,
-        timeout=timedelta(minutes=20),
-    ) -> CommandStatusResponse:
-        return self.execute(cluster_id=cluster_id, command=command, context_id=context_id, language=language).result(
-            timeout=timeout
-        )
-
 
 class GlobalInitScriptsAPI:
     """The Global Init Scripts API enables Workspace administrators to configure global initialization scripts

databricks/sdk/dashboards/v2/impl.py

@@ -4,7 +4,6 @@
 
 import logging
 from dataclasses import dataclass
-from datetime import timedelta
 from enum import Enum
 from typing import Any, Dict, Iterator, List, Optional
 
@@ -2574,13 +2573,6 @@ def create_message(self, space_id: str, conversation_id: str, content: str) -> W
             space_id=space_id,
         )
 
-    def create_message_and_wait(
-        self, space_id: str, conversation_id: str, content: str, timeout=timedelta(minutes=20)
-    ) -> GenieMessage:
-        return self.create_message(content=content, conversation_id=conversation_id, space_id=space_id).result(
-            timeout=timeout
-        )
-
     def execute_message_attachment_query(
         self, space_id: str, conversation_id: str, message_id: str, attachment_id: str
     ) -> GenieGetMessageQueryResultResponse:
@@ -2887,9 +2879,6 @@ def start_conversation(self, space_id: str, content: str) -> Wait[GenieMessage]:
             space_id=space_id,
         )
 
-    def start_conversation_and_wait(self, space_id: str, content: str, timeout=timedelta(minutes=20)) -> GenieMessage:
-        return self.start_conversation(content=content, space_id=space_id).result(timeout=timeout)
-
 
 class LakeviewAPI:
     """These APIs provide specific management operations for Lakeview dashboards. Generic resource management can