make python

Author: andrewnester

experimental/python/databricks/bundles/jobs/__init__.py

@@ -134,6 +134,9 @@
     "Library",
     "LibraryDict",
     "LibraryParam",
+    "Lifecycle",
+    "LifecycleDict",
+    "LifecycleParam",
     "LocalFileInfo",
     "LocalFileInfoDict",
     "LocalFileInfoParam",
@@ -482,6 +485,11 @@
     JobsHealthRulesParam,
 )
 from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam
+from databricks.bundles.jobs._models.lifecycle import (
+    Lifecycle,
+    LifecycleDict,
+    LifecycleParam,
+)
 from databricks.bundles.jobs._models.local_file_info import (
     LocalFileInfo,
     LocalFileInfoDict,

experimental/python/databricks/bundles/jobs/_models/job.py

@@ -44,6 +44,7 @@
     JobsHealthRules,
     JobsHealthRulesParam,
 )
+from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleParam
 from databricks.bundles.jobs._models.performance_target import (
     PerformanceTarget,
     PerformanceTargetParam,
@@ -116,6 +117,8 @@ class Job(Resource):
     A list of job cluster specifications that can be shared and reused by tasks of this job. Libraries cannot be declared in a shared job cluster. You must declare dependent libraries in task settings.
     """
 
+    lifecycle: VariableOrOptional[Lifecycle] = None
+
     max_concurrent_runs: VariableOrOptional[int] = None
     """
     An optional maximum allowed number of concurrent runs of the job.
@@ -256,6 +259,8 @@ class JobDict(TypedDict, total=False):
     A list of job cluster specifications that can be shared and reused by tasks of this job. Libraries cannot be declared in a shared job cluster. You must declare dependent libraries in task settings.
     """
 
+    lifecycle: VariableOrOptional[LifecycleParam]
+
     max_concurrent_runs: VariableOrOptional[int]
     """
     An optional maximum allowed number of concurrent runs of the job.

experimental/python/databricks/bundles/jobs/_models/lifecycle.py

@@ -0,0 +1,32 @@
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, TypedDict
+
+from databricks.bundles.core._transform import _transform
+from databricks.bundles.core._transform_to_json import _transform_to_json_value
+from databricks.bundles.core._variable import VariableOrOptional
+
+if TYPE_CHECKING:
+    from typing_extensions import Self
+
+
+@dataclass(kw_only=True)
+class Lifecycle:
+    """"""
+
+    prevent_destroy: VariableOrOptional[bool] = None
+
+    @classmethod
+    def from_dict(cls, value: "LifecycleDict") -> "Self":
+        return _transform(cls, value)
+
+    def as_dict(self) -> "LifecycleDict":
+        return _transform_to_json_value(self)  # type:ignore
+
+
+class LifecycleDict(TypedDict, total=False):
+    """"""
+
+    prevent_destroy: VariableOrOptional[bool]
+
+
+LifecycleParam = LifecycleDict | Lifecycle

experimental/python/databricks/bundles/pipelines/__init__.py

@@ -56,6 +56,9 @@
     "InitScriptInfo",
     "InitScriptInfoDict",
     "InitScriptInfoParam",
+    "Lifecycle",
+    "LifecycleDict",
+    "LifecycleParam",
     "LocalFileInfo",
     "LocalFileInfoDict",
     "LocalFileInfoParam",
@@ -236,6 +239,11 @@
     InitScriptInfoDict,
     InitScriptInfoParam,
 )
+from databricks.bundles.pipelines._models.lifecycle import (
+    Lifecycle,
+    LifecycleDict,
+    LifecycleParam,
+)
 from databricks.bundles.pipelines._models.local_file_info import (
     LocalFileInfo,
     LocalFileInfoDict,

experimental/python/databricks/bundles/pipelines/_models/lifecycle.py

@@ -0,0 +1,32 @@
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, TypedDict
+
+from databricks.bundles.core._transform import _transform
+from databricks.bundles.core._transform_to_json import _transform_to_json_value
+from databricks.bundles.core._variable import VariableOrOptional
+
+if TYPE_CHECKING:
+    from typing_extensions import Self
+
+
+@dataclass(kw_only=True)
+class Lifecycle:
+    """"""
+
+    prevent_destroy: VariableOrOptional[bool] = None
+
+    @classmethod
+    def from_dict(cls, value: "LifecycleDict") -> "Self":
+        return _transform(cls, value)
+
+    def as_dict(self) -> "LifecycleDict":
+        return _transform_to_json_value(self)  # type:ignore
+
+
+class LifecycleDict(TypedDict, total=False):
+    """"""
+
+    prevent_destroy: VariableOrOptional[bool]
+
+
+LifecycleParam = LifecycleDict | Lifecycle

experimental/python/databricks/bundles/pipelines/_models/pipeline.py

@@ -25,6 +25,10 @@
     IngestionPipelineDefinition,
     IngestionPipelineDefinitionParam,
 )
+from databricks.bundles.pipelines._models.lifecycle import (
+    Lifecycle,
+    LifecycleParam,
+)
 from databricks.bundles.pipelines._models.notifications import (
     Notifications,
     NotificationsParam,
@@ -143,6 +147,8 @@ class Pipeline(Resource):
     Libraries or code needed by this deployment.
     """
 
+    lifecycle: VariableOrOptional[Lifecycle] = None
+
     name: VariableOrOptional[str] = None
     """
     Friendly identifier for this pipeline.
@@ -301,6 +307,8 @@ class PipelineDict(TypedDict, total=False):
     Libraries or code needed by this deployment.
     """
 
+    lifecycle: VariableOrOptional[LifecycleParam]
+
     name: VariableOrOptional[str]
     """
     Friendly identifier for this pipeline.

experimental/python/databricks/bundles/volumes/__init__.py

@@ -1,4 +1,7 @@
 __all__ = [
+    "Lifecycle",
+    "LifecycleDict",
+    "LifecycleParam",
     "Volume",
     "VolumeDict",
     "VolumeGrant",
@@ -12,6 +15,11 @@
 ]
 
 
+from databricks.bundles.volumes._models.lifecycle import (
+    Lifecycle,
+    LifecycleDict,
+    LifecycleParam,
+)
 from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam
 from databricks.bundles.volumes._models.volume_grant import (
     VolumeGrant,

experimental/python/databricks/bundles/volumes/_models/lifecycle.py

@@ -0,0 +1,32 @@
+from dataclasses import dataclass
+from typing import TYPE_CHECKING, TypedDict
+
+from databricks.bundles.core._transform import _transform
+from databricks.bundles.core._transform_to_json import _transform_to_json_value
+from databricks.bundles.core._variable import VariableOrOptional
+
+if TYPE_CHECKING:
+    from typing_extensions import Self
+
+
+@dataclass(kw_only=True)
+class Lifecycle:
+    """"""
+
+    prevent_destroy: VariableOrOptional[bool] = None
+
+    @classmethod
+    def from_dict(cls, value: "LifecycleDict") -> "Self":
+        return _transform(cls, value)
+
+    def as_dict(self) -> "LifecycleDict":
+        return _transform_to_json_value(self)  # type:ignore
+
+
+class LifecycleDict(TypedDict, total=False):
+    """"""
+
+    prevent_destroy: VariableOrOptional[bool]
+
+
+LifecycleParam = LifecycleDict | Lifecycle

experimental/python/databricks/bundles/volumes/_models/volume.py

@@ -9,6 +9,7 @@
     VariableOrList,
     VariableOrOptional,
 )
+from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleParam
 from databricks.bundles.volumes._models.volume_grant import (
     VolumeGrant,
     VolumeGrantParam,
@@ -45,6 +46,8 @@ class Volume(Resource):
 
     grants: VariableOrList[VolumeGrant] = field(default_factory=list)
 
+    lifecycle: VariableOrOptional[Lifecycle] = None
+
     storage_location: VariableOrOptional[str] = None
     """
     The storage location on the cloud
@@ -85,6 +88,8 @@ class VolumeDict(TypedDict, total=False):
 
     grants: VariableOrList[VolumeGrantParam]
 
+    lifecycle: VariableOrOptional[LifecycleParam]
+
     storage_location: VariableOrOptional[str]
     """
     The storage location on the cloud