codegen python

Author: andrewnester

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

@@ -1,4 +1,7 @@
 __all__ = [
+    "Lifecycle",
+    "LifecycleDict",
+    "LifecycleParam",
     "Schema",
     "SchemaDict",
     "SchemaGrant",
@@ -10,6 +13,11 @@
 ]
 
 
+from databricks.bundles.schemas._models.lifecycle import (
+    Lifecycle,
+    LifecycleDict,
+    LifecycleParam,
+)
 from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam
 from databricks.bundles.schemas._models.schema_grant import (
     SchemaGrant,

experimental/python/databricks/bundles/schemas/_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/schemas/_models/schema.py

@@ -10,6 +10,7 @@
     VariableOrList,
     VariableOrOptional,
 )
+from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleParam
 from databricks.bundles.schemas._models.schema_grant import (
     SchemaGrant,
     SchemaGrantParam,
@@ -40,6 +41,8 @@ class Schema(Resource):
 
     grants: VariableOrList[SchemaGrant] = field(default_factory=list)
 
+    lifecycle: VariableOrOptional[Lifecycle] = None
+
     properties: VariableOrDict[str] = field(default_factory=dict)
 
     storage_root: VariableOrOptional[str] = None
@@ -75,6 +78,8 @@ class SchemaDict(TypedDict, total=False):
 
     grants: VariableOrList[SchemaGrantParam]
 
+    lifecycle: VariableOrOptional[LifecycleParam]
+
     properties: VariableOrDict[str]
 
     storage_root: VariableOrOptional[str]