Skip to content

Commit 936df1e

Browse files
committed
[Python] Add database instances
1 parent bb32491 commit 936df1e

File tree

9 files changed

+689
-1
lines changed

9 files changed

+689
-1
lines changed

python/codegen/codegen/packages.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
"resources.Pipeline": "pipelines",
88
"resources.Schema": "schemas",
99
"resources.Volume": "volumes",
10+
"resources.DatabaseInstance": "databaseinstances",
1011
}
1112

1213
RESOURCE_TYPES = list(RESOURCE_NAMESPACE.keys())
@@ -15,11 +16,12 @@
1516
#
1617
# We can't load all types because of errors while loading some of them.
1718
LOADED_NAMESPACES = [
19+
"catalog",
1820
"compute",
21+
"database",
1922
"jobs",
2023
"pipelines",
2124
"resources",
22-
"catalog",
2325
]
2426

2527
RENAMES = {
Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
__all__ = [
2+
"CustomTag",
3+
"CustomTagDict",
4+
"CustomTagParam",
5+
"DatabaseInstance",
6+
"DatabaseInstanceDict",
7+
"DatabaseInstanceParam",
8+
"DatabaseInstancePermission",
9+
"DatabaseInstancePermissionDict",
10+
"DatabaseInstancePermissionLevel",
11+
"DatabaseInstancePermissionLevelParam",
12+
"DatabaseInstancePermissionParam",
13+
"DatabaseInstanceRef",
14+
"DatabaseInstanceRefDict",
15+
"DatabaseInstanceRefParam",
16+
"DatabaseInstanceState",
17+
"DatabaseInstanceStateParam",
18+
"Lifecycle",
19+
"LifecycleDict",
20+
"LifecycleParam",
21+
]
22+
23+
24+
from databricks.bundles.databaseinstances._models.custom_tag import (
25+
CustomTag,
26+
CustomTagDict,
27+
CustomTagParam,
28+
)
29+
from databricks.bundles.databaseinstances._models.database_instance import (
30+
DatabaseInstance,
31+
DatabaseInstanceDict,
32+
DatabaseInstanceParam,
33+
)
34+
from databricks.bundles.databaseinstances._models.database_instance_permission import (
35+
DatabaseInstancePermission,
36+
DatabaseInstancePermissionDict,
37+
DatabaseInstancePermissionParam,
38+
)
39+
from databricks.bundles.databaseinstances._models.database_instance_permission_level import (
40+
DatabaseInstancePermissionLevel,
41+
DatabaseInstancePermissionLevelParam,
42+
)
43+
from databricks.bundles.databaseinstances._models.database_instance_ref import (
44+
DatabaseInstanceRef,
45+
DatabaseInstanceRefDict,
46+
DatabaseInstanceRefParam,
47+
)
48+
from databricks.bundles.databaseinstances._models.database_instance_state import (
49+
DatabaseInstanceState,
50+
DatabaseInstanceStateParam,
51+
)
52+
from databricks.bundles.databaseinstances._models.lifecycle import (
53+
Lifecycle,
54+
LifecycleDict,
55+
LifecycleParam,
56+
)
Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
1+
from dataclasses import dataclass
2+
from typing import TYPE_CHECKING, TypedDict
3+
4+
from databricks.bundles.core._transform import _transform
5+
from databricks.bundles.core._transform_to_json import _transform_to_json_value
6+
from databricks.bundles.core._variable import VariableOrOptional
7+
8+
if TYPE_CHECKING:
9+
from typing_extensions import Self
10+
11+
12+
@dataclass(kw_only=True)
13+
class CustomTag:
14+
""""""
15+
16+
key: VariableOrOptional[str] = None
17+
"""
18+
The key of the custom tag.
19+
"""
20+
21+
value: VariableOrOptional[str] = None
22+
"""
23+
The value of the custom tag.
24+
"""
25+
26+
@classmethod
27+
def from_dict(cls, value: "CustomTagDict") -> "Self":
28+
return _transform(cls, value)
29+
30+
def as_dict(self) -> "CustomTagDict":
31+
return _transform_to_json_value(self) # type:ignore
32+
33+
34+
class CustomTagDict(TypedDict, total=False):
35+
""""""
36+
37+
key: VariableOrOptional[str]
38+
"""
39+
The key of the custom tag.
40+
"""
41+
42+
value: VariableOrOptional[str]
43+
"""
44+
The value of the custom tag.
45+
"""
46+
47+
48+
CustomTagParam = CustomTagDict | CustomTag

0 commit comments

Comments
 (0)