Skip to content

Commit e15c5f6

Browse files
authored
Support writeStatus flag in resource configuration in stack API (databricks#290)
* Update stack api and examples * add test example * Rename RESOURCE_WRITE_STATUS
1 parent c7ba47d commit e15c5f6

File tree

3 files changed

+23
-8
lines changed

3 files changed

+23
-8
lines changed

databricks_cli/stack/api.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@
5050
# Resource Fields
5151
RESOURCE_ID = 'id'
5252
RESOURCE_SERVICE = 'service'
53+
RESOURCE_WRITE_STATUS = 'writeStatus'
5354
RESOURCE_PROPERTIES = 'properties'
5455

5556
# Resource Status Fields
@@ -121,7 +122,9 @@ def deploy(self, stack_config, stack_status=None, headers=None, **kwargs):
121122
resource_status,
122123
headers=headers,
123124
**kwargs)
124-
resource_statuses.append(new_resource_status)
125+
126+
if resource_config.get(RESOURCE_WRITE_STATUS, True):
127+
resource_statuses.append(new_resource_status)
125128
click.echo('#' * 80)
126129

127130
new_stack_status = {STACK_NAME: stack_name,

examples/stack/notebook-job-project/project-stack-config.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
"properties": {
1717
"name": "[Databricks] Stack CLI Example: Hello",
1818
"new_cluster": {
19-
"spark_version": "4.0.x-scala2.11",
20-
"node_type_id": "Standard_F16s",
19+
"spark_version": "6.5.x-scala2.11",
20+
"node_type_id": "Standard_F4s",
2121
"num_workers": 1
2222
},
2323
"timeout_seconds": 1200,

tests/stack/test_api.py

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -126,11 +126,23 @@
126126
}
127127
TEST_STACK = {
128128
api.STACK_NAME: "test-stack",
129-
api.STACK_RESOURCES: [TEST_JOB_RESOURCE,
130-
TEST_WORKSPACE_NB_RESOURCE,
131-
TEST_WORKSPACE_DIR_RESOURCE,
132-
TEST_DBFS_FILE_RESOURCE,
133-
TEST_DBFS_DIR_RESOURCE]
129+
api.STACK_RESOURCES: [
130+
TEST_JOB_RESOURCE,
131+
TEST_WORKSPACE_NB_RESOURCE,
132+
TEST_WORKSPACE_DIR_RESOURCE,
133+
TEST_DBFS_FILE_RESOURCE,
134+
TEST_DBFS_DIR_RESOURCE,
135+
{
136+
api.RESOURCE_ID: "NoStatusResource",
137+
api.RESOURCE_SERVICE: api.DBFS_SERVICE,
138+
api.RESOURCE_WRITE_STATUS: False,
139+
api.RESOURCE_PROPERTIES: {
140+
api.DBFS_RESOURCE_SOURCE_PATH: 'test.jar',
141+
api.DBFS_RESOURCE_PATH: 'dbfs:/test/test-no-status.jar',
142+
api.DBFS_RESOURCE_IS_DIR: False
143+
}
144+
}
145+
]
134146
}
135147
TEST_STATUS = {
136148
api.STACK_NAME: "test-stack",

0 commit comments

Comments
 (0)