Add codegen

Author: kanterov

experimental/python/codegen/codegen/packages.py

@@ -8,12 +8,15 @@
     "resources.JobPermissionLevel": "jobs",
     "resources.PipelinePermission": "pipelines",
     "resources.PipelinePermissionLevel": "pipelines",
+    "resources.Schema": "catalog",
+    "resources.Grant": "catalog",
 }
 
 # All supported resource types
 RESOURCE_TYPES = [
     "resources.Job",
     "resources.Pipeline",
+    "resources.Schema",
 ]
 
 # Namespaces to load from OpenAPI spec.
@@ -24,6 +27,7 @@
     "jobs",
     "pipelines",
     "resources",
+    "catalog",
 ]
 
 RENAMES = {

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

@@ -0,0 +1,12 @@
+__all__ = [
+    "Grant",
+    "GrantDict",
+    "GrantParam",
+    "Schema",
+    "SchemaDict",
+    "SchemaParam",
+]
+
+
+from databricks.bundles.catalog._models.grant import Grant, GrantDict, GrantParam
+from databricks.bundles.catalog._models.schema import Schema, SchemaDict, SchemaParam

experimental/python/databricks/bundles/catalog/_models/grant.py

@@ -0,0 +1,48 @@
+from dataclasses import dataclass, field
+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 VariableOr, VariableOrList
+
+if TYPE_CHECKING:
+    from typing_extensions import Self
+
+
+@dataclass(kw_only=True)
+class Grant:
+    """"""
+
+    principal: VariableOr[str]
+    """
+    The name of the principal that will be granted privileges
+    """
+
+    privileges: VariableOrList[str] = field(default_factory=list)
+    """
+    The privileges to grant to the specified entity
+    """
+
+    @classmethod
+    def from_dict(cls, value: "GrantDict") -> "Self":
+        return _transform(cls, value)
+
+    def as_dict(self) -> "GrantDict":
+        return _transform_to_json_value(self)  # type:ignore
+
+
+class GrantDict(TypedDict, total=False):
+    """"""
+
+    principal: VariableOr[str]
+    """
+    The name of the principal that will be granted privileges
+    """
+
+    privileges: VariableOrList[str]
+    """
+    The privileges to grant to the specified entity
+    """
+
+
+GrantParam = GrantDict | Grant

experimental/python/databricks/bundles/catalog/_models/schema.py

@@ -0,0 +1,83 @@
+from dataclasses import dataclass, field
+from typing import TYPE_CHECKING, TypedDict
+
+from databricks.bundles.catalog._models.grant import Grant, GrantParam
+from databricks.bundles.core._resource import Resource
+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 (
+    VariableOr,
+    VariableOrDict,
+    VariableOrList,
+    VariableOrOptional,
+)
+
+if TYPE_CHECKING:
+    from typing_extensions import Self
+
+
+@dataclass(kw_only=True)
+class Schema(Resource):
+    """"""
+
+    catalog_name: VariableOr[str]
+    """
+    Name of parent catalog.
+    """
+
+    name: VariableOr[str]
+    """
+    Name of schema, relative to parent catalog.
+    """
+
+    comment: VariableOrOptional[str] = None
+    """
+    User-provided free-form text description.
+    """
+
+    grants: VariableOrList[Grant] = field(default_factory=list)
+
+    properties: VariableOrDict[str] = field(default_factory=dict)
+
+    storage_root: VariableOrOptional[str] = None
+    """
+    Storage root URL for managed tables within schema.
+    """
+
+    @classmethod
+    def from_dict(cls, value: "SchemaDict") -> "Self":
+        return _transform(cls, value)
+
+    def as_dict(self) -> "SchemaDict":
+        return _transform_to_json_value(self)  # type:ignore
+
+
+class SchemaDict(TypedDict, total=False):
+    """"""
+
+    catalog_name: VariableOr[str]
+    """
+    Name of parent catalog.
+    """
+
+    name: VariableOr[str]
+    """
+    Name of schema, relative to parent catalog.
+    """
+
+    comment: VariableOrOptional[str]
+    """
+    User-provided free-form text description.
+    """
+
+    grants: VariableOrList[GrantParam]
+
+    properties: VariableOrDict[str]
+
+    storage_root: VariableOrOptional[str]
+    """
+    Storage root URL for managed tables within schema.
+    """
+
+
+SchemaParam = SchemaDict | Schema