pulumi-databricks 1.77.0a1760375482__py3-none-any.whl → 1.79.0a1762839813__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (176) hide show
  1. pulumi_databricks/__init__.py +54 -46
  2. pulumi_databricks/_inputs.py +26079 -32918
  3. pulumi_databricks/account_federation_policy.py +44 -16
  4. pulumi_databricks/account_network_policy.py +56 -2
  5. pulumi_databricks/account_setting_v2.py +46 -83
  6. pulumi_databricks/alert_v2.py +168 -134
  7. pulumi_databricks/app.py +153 -0
  8. pulumi_databricks/apps_settings_custom_template.py +75 -52
  9. pulumi_databricks/budget_policy.py +6 -2
  10. pulumi_databricks/catalog.py +65 -23
  11. pulumi_databricks/cluster.py +47 -168
  12. pulumi_databricks/cluster_policy.py +120 -0
  13. pulumi_databricks/config/__init__.pyi +4 -0
  14. pulumi_databricks/config/vars.py +8 -0
  15. pulumi_databricks/data_quality_monitor.py +453 -0
  16. pulumi_databricks/data_quality_refresh.py +492 -0
  17. pulumi_databricks/database_database_catalog.py +9 -52
  18. pulumi_databricks/database_instance.py +271 -120
  19. pulumi_databricks/database_synced_database_table.py +239 -52
  20. pulumi_databricks/entitlements.py +21 -21
  21. pulumi_databricks/entity_tag_assignment.py +83 -52
  22. pulumi_databricks/external_location.py +131 -7
  23. pulumi_databricks/external_metadata.py +16 -59
  24. pulumi_databricks/feature_engineering_feature.py +480 -0
  25. pulumi_databricks/feature_engineering_materialized_feature.py +397 -0
  26. pulumi_databricks/file.py +2 -2
  27. pulumi_databricks/get_account_federation_policies.py +22 -3
  28. pulumi_databricks/get_account_federation_policy.py +12 -27
  29. pulumi_databricks/get_account_network_policies.py +4 -0
  30. pulumi_databricks/get_account_network_policy.py +9 -18
  31. pulumi_databricks/get_account_setting_v2.py +36 -141
  32. pulumi_databricks/get_alert_v2.py +41 -107
  33. pulumi_databricks/get_alerts_v2.py +29 -31
  34. pulumi_databricks/get_app.py +21 -5
  35. pulumi_databricks/get_apps.py +22 -6
  36. pulumi_databricks/get_apps_settings_custom_template.py +11 -55
  37. pulumi_databricks/get_apps_settings_custom_templates.py +25 -21
  38. pulumi_databricks/get_aws_assume_role_policy.py +14 -14
  39. pulumi_databricks/get_aws_bucket_policy.py +10 -10
  40. pulumi_databricks/get_aws_unity_catalog_assume_role_policy.py +10 -10
  41. pulumi_databricks/get_aws_unity_catalog_policy.py +10 -10
  42. pulumi_databricks/get_budget_policies.py +70 -6
  43. pulumi_databricks/get_budget_policy.py +9 -32
  44. pulumi_databricks/get_catalog.py +21 -4
  45. pulumi_databricks/get_catalogs.py +23 -4
  46. pulumi_databricks/get_cluster.py +75 -4
  47. pulumi_databricks/get_cluster_policy.py +23 -4
  48. pulumi_databricks/get_clusters.py +21 -4
  49. pulumi_databricks/get_current_config.py +4 -4
  50. pulumi_databricks/get_current_metastore.py +21 -4
  51. pulumi_databricks/get_dashboards.py +20 -4
  52. pulumi_databricks/get_data_quality_monitor.py +210 -0
  53. pulumi_databricks/get_data_quality_monitors.py +143 -0
  54. pulumi_databricks/get_data_quality_refresh.py +270 -0
  55. pulumi_databricks/get_data_quality_refreshes.py +207 -0
  56. pulumi_databricks/get_database_database_catalog.py +10 -41
  57. pulumi_databricks/get_database_database_catalogs.py +37 -18
  58. pulumi_databricks/get_database_instance.py +93 -99
  59. pulumi_databricks/get_database_instances.py +20 -16
  60. pulumi_databricks/get_database_synced_database_table.py +11 -49
  61. pulumi_databricks/get_database_synced_database_tables.py +42 -23
  62. pulumi_databricks/get_directory.py +20 -1
  63. pulumi_databricks/get_entity_tag_assignment.py +63 -30
  64. pulumi_databricks/get_entity_tag_assignments.py +69 -23
  65. pulumi_databricks/get_external_location.py +21 -4
  66. pulumi_databricks/get_external_locations.py +23 -4
  67. pulumi_databricks/get_external_metadata.py +18 -73
  68. pulumi_databricks/get_external_metadatas.py +22 -16
  69. pulumi_databricks/get_feature_engineering_feature.py +179 -0
  70. pulumi_databricks/get_feature_engineering_features.py +103 -0
  71. pulumi_databricks/get_feature_engineering_materialized_feature.py +180 -0
  72. pulumi_databricks/get_feature_engineering_materialized_features.py +123 -0
  73. pulumi_databricks/get_functions.py +16 -1
  74. pulumi_databricks/get_instance_profiles.py +21 -4
  75. pulumi_databricks/get_jobs.py +23 -4
  76. pulumi_databricks/get_materialized_features_feature_tag.py +7 -32
  77. pulumi_databricks/get_materialized_features_feature_tags.py +50 -18
  78. pulumi_databricks/get_metastore.py +6 -6
  79. pulumi_databricks/get_mlflow_experiment.py +18 -1
  80. pulumi_databricks/get_mlflow_model.py +18 -1
  81. pulumi_databricks/get_mlflow_models.py +23 -4
  82. pulumi_databricks/get_node_type.py +42 -5
  83. pulumi_databricks/get_notebook.py +20 -1
  84. pulumi_databricks/get_notification_destinations.py +17 -1
  85. pulumi_databricks/get_online_store.py +11 -38
  86. pulumi_databricks/get_online_stores.py +20 -18
  87. pulumi_databricks/get_pipelines.py +23 -4
  88. pulumi_databricks/get_policy_info.py +21 -109
  89. pulumi_databricks/get_policy_infos.py +50 -25
  90. pulumi_databricks/get_quality_monitor_v2.py +8 -21
  91. pulumi_databricks/get_quality_monitors_v2.py +23 -25
  92. pulumi_databricks/get_registered_model.py +19 -4
  93. pulumi_databricks/get_registered_model_versions.py +19 -4
  94. pulumi_databricks/get_rfa_access_request_destinations.py +126 -0
  95. pulumi_databricks/get_schema.py +18 -1
  96. pulumi_databricks/get_schemas.py +23 -4
  97. pulumi_databricks/get_service_principal_federation_policies.py +23 -4
  98. pulumi_databricks/get_service_principal_federation_policy.py +15 -24
  99. pulumi_databricks/get_service_principals.py +64 -0
  100. pulumi_databricks/get_serving_endpoints.py +19 -4
  101. pulumi_databricks/get_share.py +117 -18
  102. pulumi_databricks/get_shares.py +22 -3
  103. pulumi_databricks/get_spark_version.py +20 -1
  104. pulumi_databricks/get_sql_warehouse.py +16 -1
  105. pulumi_databricks/get_sql_warehouses.py +20 -1
  106. pulumi_databricks/get_storage_credential.py +18 -1
  107. pulumi_databricks/get_storage_credentials.py +23 -4
  108. pulumi_databricks/get_table.py +18 -1
  109. pulumi_databricks/get_tables.py +20 -1
  110. pulumi_databricks/get_tag_policies.py +55 -23
  111. pulumi_databricks/get_tag_policy.py +60 -40
  112. pulumi_databricks/get_views.py +20 -1
  113. pulumi_databricks/get_volume.py +18 -1
  114. pulumi_databricks/get_volumes.py +20 -1
  115. pulumi_databricks/get_workspace_network_option.py +8 -16
  116. pulumi_databricks/get_workspace_setting_v2.py +41 -163
  117. pulumi_databricks/get_zones.py +20 -1
  118. pulumi_databricks/git_credential.py +54 -7
  119. pulumi_databricks/group.py +21 -21
  120. pulumi_databricks/instance_profile.py +0 -182
  121. pulumi_databricks/job.py +47 -0
  122. pulumi_databricks/library.py +165 -0
  123. pulumi_databricks/materialized_features_feature_tag.py +9 -52
  124. pulumi_databricks/metastore.py +81 -7
  125. pulumi_databricks/metastore_data_access.py +48 -0
  126. pulumi_databricks/mlflow_webhook.py +4 -4
  127. pulumi_databricks/mws_credentials.py +10 -10
  128. pulumi_databricks/mws_customer_managed_keys.py +0 -288
  129. pulumi_databricks/mws_log_delivery.py +146 -0
  130. pulumi_databricks/mws_storage_configurations.py +16 -16
  131. pulumi_databricks/mws_vpc_endpoint.py +56 -56
  132. pulumi_databricks/mws_workspaces.py +146 -65
  133. pulumi_databricks/notebook.py +49 -0
  134. pulumi_databricks/online_store.py +9 -52
  135. pulumi_databricks/outputs.py +11885 -13474
  136. pulumi_databricks/permission_assignment.py +299 -8
  137. pulumi_databricks/permissions.py +6 -6
  138. pulumi_databricks/pipeline.py +37 -3
  139. pulumi_databricks/policy_info.py +9 -52
  140. pulumi_databricks/provider.py +36 -1
  141. pulumi_databricks/pulumi-plugin.json +1 -1
  142. pulumi_databricks/quality_monitor.py +47 -0
  143. pulumi_databricks/quality_monitor_v2.py +9 -52
  144. pulumi_databricks/recipient.py +74 -0
  145. pulumi_databricks/registered_model.py +308 -36
  146. pulumi_databricks/rfa_access_request_destinations.py +286 -0
  147. pulumi_databricks/schema.py +7 -7
  148. pulumi_databricks/service_principal_federation_policy.py +48 -16
  149. pulumi_databricks/share.py +71 -84
  150. pulumi_databricks/sql_table.py +42 -14
  151. pulumi_databricks/storage_credential.py +15 -2
  152. pulumi_databricks/tag_policy.py +111 -54
  153. pulumi_databricks/user.py +21 -21
  154. pulumi_databricks/volume.py +7 -7
  155. pulumi_databricks/workspace_binding.py +0 -48
  156. pulumi_databricks/workspace_network_option.py +6 -2
  157. pulumi_databricks/workspace_setting_v2.py +49 -133
  158. {pulumi_databricks-1.77.0a1760375482.dist-info → pulumi_databricks-1.79.0a1762839813.dist-info}/METADATA +1 -1
  159. pulumi_databricks-1.79.0a1762839813.dist-info/RECORD +250 -0
  160. pulumi_databricks/clean_room_asset.py +0 -938
  161. pulumi_databricks/clean_room_auto_approval_rule.py +0 -473
  162. pulumi_databricks/clean_rooms_clean_room.py +0 -569
  163. pulumi_databricks/get_clean_room_asset.py +0 -416
  164. pulumi_databricks/get_clean_room_asset_revisions_clean_room_asset.py +0 -392
  165. pulumi_databricks/get_clean_room_asset_revisions_clean_room_assets.py +0 -121
  166. pulumi_databricks/get_clean_room_assets.py +0 -146
  167. pulumi_databricks/get_clean_room_auto_approval_rule.py +0 -217
  168. pulumi_databricks/get_clean_room_auto_approval_rules.py +0 -101
  169. pulumi_databricks/get_clean_rooms_clean_room.py +0 -289
  170. pulumi_databricks/get_clean_rooms_clean_rooms.py +0 -125
  171. pulumi_databricks/get_recipient_federation_policies.py +0 -101
  172. pulumi_databricks/get_recipient_federation_policy.py +0 -182
  173. pulumi_databricks/recipient_federation_policy.py +0 -393
  174. pulumi_databricks-1.77.0a1760375482.dist-info/RECORD +0 -250
  175. {pulumi_databricks-1.77.0a1760375482.dist-info → pulumi_databricks-1.79.0a1762839813.dist-info}/WHEEL +0 -0
  176. {pulumi_databricks-1.77.0a1760375482.dist-info → pulumi_databricks-1.79.0a1762839813.dist-info}/top_level.txt +0 -0
@@ -13,6 +13,8 @@ if sys.version_info >= (3, 11):
13
13
  else:
14
14
  from typing_extensions import NotRequired, TypedDict, TypeAlias
15
15
  from . import _utilities
16
+ from . import outputs
17
+ from ._inputs import *
16
18
 
17
19
  __all__ = [
18
20
  'GetSchemasResult',
@@ -26,7 +28,7 @@ class GetSchemasResult:
26
28
  """
27
29
  A collection of values returned by getSchemas.
28
30
  """
29
- def __init__(__self__, catalog_name=None, id=None, ids=None):
31
+ def __init__(__self__, catalog_name=None, id=None, ids=None, provider_config=None):
30
32
  if catalog_name and not isinstance(catalog_name, str):
31
33
  raise TypeError("Expected argument 'catalog_name' to be a str")
32
34
  pulumi.set(__self__, "catalog_name", catalog_name)
@@ -36,6 +38,9 @@ class GetSchemasResult:
36
38
  if ids and not isinstance(ids, list):
37
39
  raise TypeError("Expected argument 'ids' to be a list")
38
40
  pulumi.set(__self__, "ids", ids)
41
+ if provider_config and not isinstance(provider_config, dict):
42
+ raise TypeError("Expected argument 'provider_config' to be a dict")
43
+ pulumi.set(__self__, "provider_config", provider_config)
39
44
 
40
45
  @_builtins.property
41
46
  @pulumi.getter(name="catalogName")
@@ -58,6 +63,11 @@ class GetSchemasResult:
58
63
  """
59
64
  return pulumi.get(self, "ids")
60
65
 
66
+ @_builtins.property
67
+ @pulumi.getter(name="providerConfig")
68
+ def provider_config(self) -> Optional['outputs.GetSchemasProviderConfigResult']:
69
+ return pulumi.get(self, "provider_config")
70
+
61
71
 
62
72
  class AwaitableGetSchemasResult(GetSchemasResult):
63
73
  # pylint: disable=using-constant-test
@@ -67,11 +77,13 @@ class AwaitableGetSchemasResult(GetSchemasResult):
67
77
  return GetSchemasResult(
68
78
  catalog_name=self.catalog_name,
69
79
  id=self.id,
70
- ids=self.ids)
80
+ ids=self.ids,
81
+ provider_config=self.provider_config)
71
82
 
72
83
 
73
84
  def get_schemas(catalog_name: Optional[_builtins.str] = None,
74
85
  ids: Optional[Sequence[_builtins.str]] = None,
86
+ provider_config: Optional[Union['GetSchemasProviderConfigArgs', 'GetSchemasProviderConfigArgsDict']] = None,
75
87
  opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetSchemasResult:
76
88
  """
77
89
  Retrieves a list of Schema ids, that were created by Pulumi or manually, so that special handling could be applied.
@@ -100,19 +112,23 @@ def get_schemas(catalog_name: Optional[_builtins.str] = None,
100
112
 
101
113
  :param _builtins.str catalog_name: Name of databricks_catalog
102
114
  :param Sequence[_builtins.str] ids: set of Schema full names: *`catalog`.`schema`*
115
+ :param Union['GetSchemasProviderConfigArgs', 'GetSchemasProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
103
116
  """
104
117
  __args__ = dict()
105
118
  __args__['catalogName'] = catalog_name
106
119
  __args__['ids'] = ids
120
+ __args__['providerConfig'] = provider_config
107
121
  opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
108
122
  __ret__ = pulumi.runtime.invoke('databricks:index/getSchemas:getSchemas', __args__, opts=opts, typ=GetSchemasResult).value
109
123
 
110
124
  return AwaitableGetSchemasResult(
111
125
  catalog_name=pulumi.get(__ret__, 'catalog_name'),
112
126
  id=pulumi.get(__ret__, 'id'),
113
- ids=pulumi.get(__ret__, 'ids'))
127
+ ids=pulumi.get(__ret__, 'ids'),
128
+ provider_config=pulumi.get(__ret__, 'provider_config'))
114
129
  def get_schemas_output(catalog_name: Optional[pulumi.Input[_builtins.str]] = None,
115
130
  ids: Optional[pulumi.Input[Optional[Sequence[_builtins.str]]]] = None,
131
+ provider_config: Optional[pulumi.Input[Optional[Union['GetSchemasProviderConfigArgs', 'GetSchemasProviderConfigArgsDict']]]] = None,
116
132
  opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetSchemasResult]:
117
133
  """
118
134
  Retrieves a list of Schema ids, that were created by Pulumi or manually, so that special handling could be applied.
@@ -141,13 +157,16 @@ def get_schemas_output(catalog_name: Optional[pulumi.Input[_builtins.str]] = Non
141
157
 
142
158
  :param _builtins.str catalog_name: Name of databricks_catalog
143
159
  :param Sequence[_builtins.str] ids: set of Schema full names: *`catalog`.`schema`*
160
+ :param Union['GetSchemasProviderConfigArgs', 'GetSchemasProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
144
161
  """
145
162
  __args__ = dict()
146
163
  __args__['catalogName'] = catalog_name
147
164
  __args__['ids'] = ids
165
+ __args__['providerConfig'] = provider_config
148
166
  opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
149
167
  __ret__ = pulumi.runtime.invoke_output('databricks:index/getSchemas:getSchemas', __args__, opts=opts, typ=GetSchemasResult)
150
168
  return __ret__.apply(lambda __response__: GetSchemasResult(
151
169
  catalog_name=pulumi.get(__response__, 'catalog_name'),
152
170
  id=pulumi.get(__response__, 'id'),
153
- ids=pulumi.get(__response__, 'ids')))
171
+ ids=pulumi.get(__response__, 'ids'),
172
+ provider_config=pulumi.get(__response__, 'provider_config')))
@@ -27,10 +27,13 @@ class GetServicePrincipalFederationPoliciesResult:
27
27
  """
28
28
  A collection of values returned by getServicePrincipalFederationPolicies.
29
29
  """
30
- def __init__(__self__, id=None, policies=None, service_principal_id=None):
30
+ def __init__(__self__, id=None, page_size=None, policies=None, service_principal_id=None):
31
31
  if id and not isinstance(id, str):
32
32
  raise TypeError("Expected argument 'id' to be a str")
33
33
  pulumi.set(__self__, "id", id)
34
+ if page_size and not isinstance(page_size, int):
35
+ raise TypeError("Expected argument 'page_size' to be a int")
36
+ pulumi.set(__self__, "page_size", page_size)
34
37
  if policies and not isinstance(policies, list):
35
38
  raise TypeError("Expected argument 'policies' to be a list")
36
39
  pulumi.set(__self__, "policies", policies)
@@ -46,6 +49,11 @@ class GetServicePrincipalFederationPoliciesResult:
46
49
  """
47
50
  return pulumi.get(self, "id")
48
51
 
52
+ @_builtins.property
53
+ @pulumi.getter(name="pageSize")
54
+ def page_size(self) -> Optional[_builtins.int]:
55
+ return pulumi.get(self, "page_size")
56
+
49
57
  @_builtins.property
50
58
  @pulumi.getter
51
59
  def policies(self) -> Sequence['outputs.GetServicePrincipalFederationPoliciesPolicyResult']:
@@ -55,7 +63,7 @@ class GetServicePrincipalFederationPoliciesResult:
55
63
  @pulumi.getter(name="servicePrincipalId")
56
64
  def service_principal_id(self) -> _builtins.int:
57
65
  """
58
- (integer) - The service principal ID that this federation policy applies to. Only set for service principal federation policies
66
+ (integer) - The service principal ID that this federation policy applies to. Output only. Only set for service principal federation policies
59
67
  """
60
68
  return pulumi.get(self, "service_principal_id")
61
69
 
@@ -67,13 +75,17 @@ class AwaitableGetServicePrincipalFederationPoliciesResult(GetServicePrincipalFe
67
75
  yield self
68
76
  return GetServicePrincipalFederationPoliciesResult(
69
77
  id=self.id,
78
+ page_size=self.page_size,
70
79
  policies=self.policies,
71
80
  service_principal_id=self.service_principal_id)
72
81
 
73
82
 
74
- def get_service_principal_federation_policies(service_principal_id: Optional[_builtins.int] = None,
83
+ def get_service_principal_federation_policies(page_size: Optional[_builtins.int] = None,
84
+ service_principal_id: Optional[_builtins.int] = None,
75
85
  opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetServicePrincipalFederationPoliciesResult:
76
86
  """
87
+ [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
88
+
77
89
  This data source can be used to fetch the list of federation policies for a service principal.
78
90
 
79
91
  > **Note** This data source can only be used with an account-level provider!
@@ -93,17 +105,22 @@ def get_service_principal_federation_policies(service_principal_id: Optional[_bu
93
105
  :param _builtins.int service_principal_id: The service principal id for the federation policy
94
106
  """
95
107
  __args__ = dict()
108
+ __args__['pageSize'] = page_size
96
109
  __args__['servicePrincipalId'] = service_principal_id
97
110
  opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
98
111
  __ret__ = pulumi.runtime.invoke('databricks:index/getServicePrincipalFederationPolicies:getServicePrincipalFederationPolicies', __args__, opts=opts, typ=GetServicePrincipalFederationPoliciesResult).value
99
112
 
100
113
  return AwaitableGetServicePrincipalFederationPoliciesResult(
101
114
  id=pulumi.get(__ret__, 'id'),
115
+ page_size=pulumi.get(__ret__, 'page_size'),
102
116
  policies=pulumi.get(__ret__, 'policies'),
103
117
  service_principal_id=pulumi.get(__ret__, 'service_principal_id'))
104
- def get_service_principal_federation_policies_output(service_principal_id: Optional[pulumi.Input[_builtins.int]] = None,
118
+ def get_service_principal_federation_policies_output(page_size: Optional[pulumi.Input[Optional[_builtins.int]]] = None,
119
+ service_principal_id: Optional[pulumi.Input[_builtins.int]] = None,
105
120
  opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetServicePrincipalFederationPoliciesResult]:
106
121
  """
122
+ [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
123
+
107
124
  This data source can be used to fetch the list of federation policies for a service principal.
108
125
 
109
126
  > **Note** This data source can only be used with an account-level provider!
@@ -123,10 +140,12 @@ def get_service_principal_federation_policies_output(service_principal_id: Optio
123
140
  :param _builtins.int service_principal_id: The service principal id for the federation policy
124
141
  """
125
142
  __args__ = dict()
143
+ __args__['pageSize'] = page_size
126
144
  __args__['servicePrincipalId'] = service_principal_id
127
145
  opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
128
146
  __ret__ = pulumi.runtime.invoke_output('databricks:index/getServicePrincipalFederationPolicies:getServicePrincipalFederationPolicies', __args__, opts=opts, typ=GetServicePrincipalFederationPoliciesResult)
129
147
  return __ret__.apply(lambda __response__: GetServicePrincipalFederationPoliciesResult(
130
148
  id=pulumi.get(__response__, 'id'),
149
+ page_size=pulumi.get(__response__, 'page_size'),
131
150
  policies=pulumi.get(__response__, 'policies'),
132
151
  service_principal_id=pulumi.get(__response__, 'service_principal_id')))
@@ -14,7 +14,6 @@ else:
14
14
  from typing_extensions import NotRequired, TypedDict, TypeAlias
15
15
  from . import _utilities
16
16
  from . import outputs
17
- from ._inputs import *
18
17
 
19
18
  __all__ = [
20
19
  'GetServicePrincipalFederationPolicyResult',
@@ -67,7 +66,7 @@ class GetServicePrincipalFederationPolicyResult:
67
66
 
68
67
  @_builtins.property
69
68
  @pulumi.getter
70
- def description(self) -> Optional[_builtins.str]:
69
+ def description(self) -> _builtins.str:
71
70
  """
72
71
  (string) - Description of the federation policy
73
72
  """
@@ -96,7 +95,7 @@ class GetServicePrincipalFederationPolicyResult:
96
95
 
97
96
  @_builtins.property
98
97
  @pulumi.getter(name="oidcPolicy")
99
- def oidc_policy(self) -> Optional['outputs.GetServicePrincipalFederationPolicyOidcPolicyResult']:
98
+ def oidc_policy(self) -> 'outputs.GetServicePrincipalFederationPolicyOidcPolicyResult':
100
99
  """
101
100
  (OidcFederationPolicy)
102
101
  """
@@ -106,7 +105,7 @@ class GetServicePrincipalFederationPolicyResult:
106
105
  @pulumi.getter(name="policyId")
107
106
  def policy_id(self) -> _builtins.str:
108
107
  """
109
- (string) - The ID of the federation policy
108
+ (string) - The ID of the federation policy. Output only
110
109
  """
111
110
  return pulumi.get(self, "policy_id")
112
111
 
@@ -114,7 +113,7 @@ class GetServicePrincipalFederationPolicyResult:
114
113
  @pulumi.getter(name="servicePrincipalId")
115
114
  def service_principal_id(self) -> _builtins.int:
116
115
  """
117
- (integer) - The service principal ID that this federation policy applies to. Only set for service principal federation policies
116
+ (integer) - The service principal ID that this federation policy applies to. Output only. Only set for service principal federation policies
118
117
  """
119
118
  return pulumi.get(self, "service_principal_id")
120
119
 
@@ -152,12 +151,12 @@ class AwaitableGetServicePrincipalFederationPolicyResult(GetServicePrincipalFede
152
151
  update_time=self.update_time)
153
152
 
154
153
 
155
- def get_service_principal_federation_policy(description: Optional[_builtins.str] = None,
156
- oidc_policy: Optional[Union['GetServicePrincipalFederationPolicyOidcPolicyArgs', 'GetServicePrincipalFederationPolicyOidcPolicyArgsDict']] = None,
157
- policy_id: Optional[_builtins.str] = None,
154
+ def get_service_principal_federation_policy(policy_id: Optional[_builtins.str] = None,
158
155
  service_principal_id: Optional[_builtins.int] = None,
159
156
  opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetServicePrincipalFederationPolicyResult:
160
157
  """
158
+ [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
159
+
161
160
  This data source can be used to get a single service principal federation policy.
162
161
 
163
162
  > **Note** This data source can only be used with an account-level provider!
@@ -167,14 +166,10 @@ def get_service_principal_federation_policy(description: Optional[_builtins.str]
167
166
  Referring to a service principal federation policy by id:
168
167
 
169
168
 
170
- :param _builtins.str description: (string) - Description of the federation policy
171
- :param Union['GetServicePrincipalFederationPolicyOidcPolicyArgs', 'GetServicePrincipalFederationPolicyOidcPolicyArgsDict'] oidc_policy: (OidcFederationPolicy)
172
- :param _builtins.str policy_id: The ID of the federation policy
173
- :param _builtins.int service_principal_id: The service principal ID that this federation policy applies to. Only set for service principal federation policies
169
+ :param _builtins.str policy_id: The ID of the federation policy. Output only
170
+ :param _builtins.int service_principal_id: The service principal ID that this federation policy applies to. Output only. Only set for service principal federation policies
174
171
  """
175
172
  __args__ = dict()
176
- __args__['description'] = description
177
- __args__['oidcPolicy'] = oidc_policy
178
173
  __args__['policyId'] = policy_id
179
174
  __args__['servicePrincipalId'] = service_principal_id
180
175
  opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
@@ -190,12 +185,12 @@ def get_service_principal_federation_policy(description: Optional[_builtins.str]
190
185
  service_principal_id=pulumi.get(__ret__, 'service_principal_id'),
191
186
  uid=pulumi.get(__ret__, 'uid'),
192
187
  update_time=pulumi.get(__ret__, 'update_time'))
193
- def get_service_principal_federation_policy_output(description: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
194
- oidc_policy: Optional[pulumi.Input[Optional[Union['GetServicePrincipalFederationPolicyOidcPolicyArgs', 'GetServicePrincipalFederationPolicyOidcPolicyArgsDict']]]] = None,
195
- policy_id: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
196
- service_principal_id: Optional[pulumi.Input[Optional[_builtins.int]]] = None,
188
+ def get_service_principal_federation_policy_output(policy_id: Optional[pulumi.Input[_builtins.str]] = None,
189
+ service_principal_id: Optional[pulumi.Input[_builtins.int]] = None,
197
190
  opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetServicePrincipalFederationPolicyResult]:
198
191
  """
192
+ [![GA](https://img.shields.io/badge/Release_Stage-GA-green)](https://docs.databricks.com/aws/en/release-notes/release-types)
193
+
199
194
  This data source can be used to get a single service principal federation policy.
200
195
 
201
196
  > **Note** This data source can only be used with an account-level provider!
@@ -205,14 +200,10 @@ def get_service_principal_federation_policy_output(description: Optional[pulumi.
205
200
  Referring to a service principal federation policy by id:
206
201
 
207
202
 
208
- :param _builtins.str description: (string) - Description of the federation policy
209
- :param Union['GetServicePrincipalFederationPolicyOidcPolicyArgs', 'GetServicePrincipalFederationPolicyOidcPolicyArgsDict'] oidc_policy: (OidcFederationPolicy)
210
- :param _builtins.str policy_id: The ID of the federation policy
211
- :param _builtins.int service_principal_id: The service principal ID that this federation policy applies to. Only set for service principal federation policies
203
+ :param _builtins.str policy_id: The ID of the federation policy. Output only
204
+ :param _builtins.int service_principal_id: The service principal ID that this federation policy applies to. Output only. Only set for service principal federation policies
212
205
  """
213
206
  __args__ = dict()
214
- __args__['description'] = description
215
- __args__['oidcPolicy'] = oidc_policy
216
207
  __args__['policyId'] = policy_id
217
208
  __args__['servicePrincipalId'] = service_principal_id
218
209
  opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
@@ -78,6 +78,38 @@ def get_service_principals(application_ids: Optional[Sequence[_builtins.str]] =
78
78
 
79
79
  > This data source can be used with an account or workspace-level provider.
80
80
 
81
+ ## Example Usage
82
+
83
+ Adding all service principals of which display name contains `my-spn` to admin group
84
+
85
+ ```python
86
+ import pulumi
87
+ import pulumi_databricks as databricks
88
+ import pulumi_std as std
89
+
90
+ admins = databricks.get_group(display_name="admins")
91
+ spns = databricks.get_service_principals(display_name_contains="my-spn")
92
+ spn = {__key: databricks.get_service_principal(application_id=__value) for __key, __value in std.toset(input=spns.application_ids).result}
93
+ my_member_spn = []
94
+ for range in [{"key": k, "value": v} for [k, v] in enumerate(std.toset(input=spns.application_ids).result)]:
95
+ my_member_spn.append(databricks.GroupMember(f"my_member_spn-{range['key']}",
96
+ group_id=admins.id,
97
+ member_id=spn[range["value"]].sp_id))
98
+ ```
99
+
100
+ ## Related Resources
101
+
102
+ The following resources are used in the same context:
103
+
104
+ - End to end workspace management guide.
105
+ - get_current_user data to retrieve information about User or databricks_service_principal, that is calling Databricks REST API.
106
+ - Group to manage [Account-level](https://docs.databricks.com/aws/en/admin/users-groups/groups) or [Workspace-level](https://docs.databricks.com/aws/en/admin/users-groups/workspace-local-groups) groups.
107
+ - Group data to retrieve information about Group members, entitlements and instance profiles.
108
+ - GroupInstanceProfile to attach InstanceProfile (AWS) to databricks_group.
109
+ - GroupMember to attach users and groups as group members.
110
+ - Permissions to manage [access control](https://docs.databricks.com/security/access-control/index.html) in Databricks workspace.
111
+ - databricks_service principal to manage service principals
112
+
81
113
 
82
114
  :param Sequence[_builtins.str] application_ids: List of `application_ids` of service principals. Individual service principal can be retrieved using ServicePrincipal data source
83
115
  :param _builtins.str display_name_contains: Only return ServicePrincipal display name that match the given name string
@@ -100,6 +132,38 @@ def get_service_principals_output(application_ids: Optional[pulumi.Input[Optiona
100
132
 
101
133
  > This data source can be used with an account or workspace-level provider.
102
134
 
135
+ ## Example Usage
136
+
137
+ Adding all service principals of which display name contains `my-spn` to admin group
138
+
139
+ ```python
140
+ import pulumi
141
+ import pulumi_databricks as databricks
142
+ import pulumi_std as std
143
+
144
+ admins = databricks.get_group(display_name="admins")
145
+ spns = databricks.get_service_principals(display_name_contains="my-spn")
146
+ spn = {__key: databricks.get_service_principal(application_id=__value) for __key, __value in std.toset(input=spns.application_ids).result}
147
+ my_member_spn = []
148
+ for range in [{"key": k, "value": v} for [k, v] in enumerate(std.toset(input=spns.application_ids).result)]:
149
+ my_member_spn.append(databricks.GroupMember(f"my_member_spn-{range['key']}",
150
+ group_id=admins.id,
151
+ member_id=spn[range["value"]].sp_id))
152
+ ```
153
+
154
+ ## Related Resources
155
+
156
+ The following resources are used in the same context:
157
+
158
+ - End to end workspace management guide.
159
+ - get_current_user data to retrieve information about User or databricks_service_principal, that is calling Databricks REST API.
160
+ - Group to manage [Account-level](https://docs.databricks.com/aws/en/admin/users-groups/groups) or [Workspace-level](https://docs.databricks.com/aws/en/admin/users-groups/workspace-local-groups) groups.
161
+ - Group data to retrieve information about Group members, entitlements and instance profiles.
162
+ - GroupInstanceProfile to attach InstanceProfile (AWS) to databricks_group.
163
+ - GroupMember to attach users and groups as group members.
164
+ - Permissions to manage [access control](https://docs.databricks.com/security/access-control/index.html) in Databricks workspace.
165
+ - databricks_service principal to manage service principals
166
+
103
167
 
104
168
  :param Sequence[_builtins.str] application_ids: List of `application_ids` of service principals. Individual service principal can be retrieved using ServicePrincipal data source
105
169
  :param _builtins.str display_name_contains: Only return ServicePrincipal display name that match the given name string
@@ -28,13 +28,16 @@ class GetServingEndpointsResult:
28
28
  """
29
29
  A collection of values returned by getServingEndpoints.
30
30
  """
31
- def __init__(__self__, endpoints=None, id=None):
31
+ def __init__(__self__, endpoints=None, id=None, provider_config=None):
32
32
  if endpoints and not isinstance(endpoints, list):
33
33
  raise TypeError("Expected argument 'endpoints' to be a list")
34
34
  pulumi.set(__self__, "endpoints", endpoints)
35
35
  if id and not isinstance(id, str):
36
36
  raise TypeError("Expected argument 'id' to be a str")
37
37
  pulumi.set(__self__, "id", id)
38
+ if provider_config and not isinstance(provider_config, dict):
39
+ raise TypeError("Expected argument 'provider_config' to be a dict")
40
+ pulumi.set(__self__, "provider_config", provider_config)
38
41
 
39
42
  @_builtins.property
40
43
  @pulumi.getter
@@ -52,6 +55,11 @@ class GetServingEndpointsResult:
52
55
  """
53
56
  return pulumi.get(self, "id")
54
57
 
58
+ @_builtins.property
59
+ @pulumi.getter(name="providerConfig")
60
+ def provider_config(self) -> Optional['outputs.GetServingEndpointsProviderConfigResult']:
61
+ return pulumi.get(self, "provider_config")
62
+
55
63
 
56
64
  class AwaitableGetServingEndpointsResult(GetServingEndpointsResult):
57
65
  # pylint: disable=using-constant-test
@@ -60,10 +68,12 @@ class AwaitableGetServingEndpointsResult(GetServingEndpointsResult):
60
68
  yield self
61
69
  return GetServingEndpointsResult(
62
70
  endpoints=self.endpoints,
63
- id=self.id)
71
+ id=self.id,
72
+ provider_config=self.provider_config)
64
73
 
65
74
 
66
75
  def get_serving_endpoints(endpoints: Optional[Sequence[Union['GetServingEndpointsEndpointArgs', 'GetServingEndpointsEndpointArgsDict']]] = None,
76
+ provider_config: Optional[Union['GetServingEndpointsProviderConfigArgs', 'GetServingEndpointsProviderConfigArgsDict']] = None,
67
77
  opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetServingEndpointsResult:
68
78
  """
69
79
  This resource allows you to get information about [Model Serving](https://docs.databricks.com/machine-learning/model-serving/index.html) endpoints in Databricks.
@@ -108,13 +118,16 @@ def get_serving_endpoints(endpoints: Optional[Sequence[Union['GetServingEndpoint
108
118
  """
109
119
  __args__ = dict()
110
120
  __args__['endpoints'] = endpoints
121
+ __args__['providerConfig'] = provider_config
111
122
  opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
112
123
  __ret__ = pulumi.runtime.invoke('databricks:index/getServingEndpoints:getServingEndpoints', __args__, opts=opts, typ=GetServingEndpointsResult).value
113
124
 
114
125
  return AwaitableGetServingEndpointsResult(
115
126
  endpoints=pulumi.get(__ret__, 'endpoints'),
116
- id=pulumi.get(__ret__, 'id'))
127
+ id=pulumi.get(__ret__, 'id'),
128
+ provider_config=pulumi.get(__ret__, 'provider_config'))
117
129
  def get_serving_endpoints_output(endpoints: Optional[pulumi.Input[Optional[Sequence[Union['GetServingEndpointsEndpointArgs', 'GetServingEndpointsEndpointArgsDict']]]]] = None,
130
+ provider_config: Optional[pulumi.Input[Optional[Union['GetServingEndpointsProviderConfigArgs', 'GetServingEndpointsProviderConfigArgsDict']]]] = None,
118
131
  opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetServingEndpointsResult]:
119
132
  """
120
133
  This resource allows you to get information about [Model Serving](https://docs.databricks.com/machine-learning/model-serving/index.html) endpoints in Databricks.
@@ -159,8 +172,10 @@ def get_serving_endpoints_output(endpoints: Optional[pulumi.Input[Optional[Seque
159
172
  """
160
173
  __args__ = dict()
161
174
  __args__['endpoints'] = endpoints
175
+ __args__['providerConfig'] = provider_config
162
176
  opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
163
177
  __ret__ = pulumi.runtime.invoke_output('databricks:index/getServingEndpoints:getServingEndpoints', __args__, opts=opts, typ=GetServingEndpointsResult)
164
178
  return __ret__.apply(lambda __response__: GetServingEndpointsResult(
165
179
  endpoints=pulumi.get(__response__, 'endpoints'),
166
- id=pulumi.get(__response__, 'id')))
180
+ id=pulumi.get(__response__, 'id'),
181
+ provider_config=pulumi.get(__response__, 'provider_config')))