pulumi-databricks 1.78.0a1762407761__py3-none-any.whl → 1.79.0__py3-none-any.whl

pulumi_databricks/get_service_principals.py

@@ -13,6 +13,8 @@ if sys.version_info >= (3, 11):
 else:
     from typing_extensions import NotRequired, TypedDict, TypeAlias
 from . import _utilities
+from . import outputs
+from ._inputs import *
 
 __all__ = [
     'GetServicePrincipalsResult',
@@ -26,7 +28,7 @@ class GetServicePrincipalsResult:
     """
     A collection of values returned by getServicePrincipals.
     """
-    def __init__(__self__, application_ids=None, display_name_contains=None, id=None):
+    def __init__(__self__, application_ids=None, display_name_contains=None, id=None, service_principals=None):
         if application_ids and not isinstance(application_ids, list):
             raise TypeError("Expected argument 'application_ids' to be a list")
         pulumi.set(__self__, "application_ids", application_ids)
@@ -36,12 +38,15 @@ class GetServicePrincipalsResult:
         if id and not isinstance(id, str):
             raise TypeError("Expected argument 'id' to be a str")
         pulumi.set(__self__, "id", id)
+        if service_principals and not isinstance(service_principals, list):
+            raise TypeError("Expected argument 'service_principals' to be a list")
+        pulumi.set(__self__, "service_principals", service_principals)
 
     @_builtins.property
     @pulumi.getter(name="applicationIds")
     def application_ids(self) -> Sequence[_builtins.str]:
         """
-        List of `application_ids` of service principals.  Individual service principal can be retrieved using ServicePrincipal data source
+        List of `application_ids` of service principals.  Individual service principal can be retrieved using ServicePrincipal data source or from `service_principals` attribute.
         """
         return pulumi.get(self, "application_ids")
 
@@ -58,6 +63,14 @@ class GetServicePrincipalsResult:
         """
         return pulumi.get(self, "id")
 
+    @_builtins.property
+    @pulumi.getter(name="servicePrincipals")
+    def service_principals(self) -> Sequence['outputs.GetServicePrincipalsServicePrincipalResult']:
+        """
+        List of objects describing individual service principals. Each object has the following attributes:
+        """
+        return pulumi.get(self, "service_principals")
+
 
 class AwaitableGetServicePrincipalsResult(GetServicePrincipalsResult):
     # pylint: disable=using-constant-test
@@ -67,49 +80,122 @@ class AwaitableGetServicePrincipalsResult(GetServicePrincipalsResult):
         return GetServicePrincipalsResult(
             application_ids=self.application_ids,
             display_name_contains=self.display_name_contains,
-            id=self.id)
+            id=self.id,
+            service_principals=self.service_principals)
 
 
 def get_service_principals(application_ids: Optional[Sequence[_builtins.str]] = None,
                            display_name_contains: Optional[_builtins.str] = None,
+                           service_principals: Optional[Sequence[Union['GetServicePrincipalsServicePrincipalArgs', 'GetServicePrincipalsServicePrincipalArgsDict']]] = None,
                            opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetServicePrincipalsResult:
     """
     Retrieves `application_ids` of all ServicePrincipal based on their `display_name`
 
     > This data source can be used with an account or workspace-level provider.
 
+    ## Example Usage
+
+    Adding all service principals of which display name contains `my-spn` to admin group
+
+    ```python
+    import pulumi
+    import pulumi_databricks as databricks
+    import pulumi_std as std
+
+    admins = databricks.get_group(display_name="admins")
+    spns = databricks.get_service_principals(display_name_contains="my-spn")
+    spn = {__key: databricks.get_service_principal(application_id=__value) for __key, __value in std.toset(input=spns.application_ids).result}
+    my_member_spn = []
+    for range in [{"key": k, "value": v} for [k, v] in enumerate(std.toset(input=spns.application_ids).result)]:
+        my_member_spn.append(databricks.GroupMember(f"my_member_spn-{range['key']}",
+            group_id=admins.id,
+            member_id=spn[range["value"]].sp_id))
+    ```
+
+    ## Related Resources
+
+    The following resources are used in the same context:
 
-    :param Sequence[_builtins.str] application_ids: List of `application_ids` of service principals.  Individual service principal can be retrieved using ServicePrincipal data source
+    - End to end workspace management guide.
+    - get_current_user data to retrieve information about User or databricks_service_principal, that is calling Databricks REST API.
+    - 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.
+    - Group data to retrieve information about Group members, entitlements and instance profiles.
+    - GroupInstanceProfile to attach InstanceProfile (AWS) to databricks_group.
+    - GroupMember to attach users and groups as group members.
+    - Permissions to manage [access control](https://docs.databricks.com/security/access-control/index.html) in Databricks workspace.
+    - databricks_service principal to manage service principals
+
+
+    :param Sequence[_builtins.str] application_ids: List of `application_ids` of service principals.  Individual service principal can be retrieved using ServicePrincipal data source or from `service_principals` attribute.
     :param _builtins.str display_name_contains: Only return ServicePrincipal display name that match the given name string
+    :param Sequence[Union['GetServicePrincipalsServicePrincipalArgs', 'GetServicePrincipalsServicePrincipalArgsDict']] service_principals: List of objects describing individual service principals. Each object has the following attributes:
     """
     __args__ = dict()
     __args__['applicationIds'] = application_ids
     __args__['displayNameContains'] = display_name_contains
+    __args__['servicePrincipals'] = service_principals
     opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke('databricks:index/getServicePrincipals:getServicePrincipals', __args__, opts=opts, typ=GetServicePrincipalsResult).value
 
     return AwaitableGetServicePrincipalsResult(
         application_ids=pulumi.get(__ret__, 'application_ids'),
         display_name_contains=pulumi.get(__ret__, 'display_name_contains'),
-        id=pulumi.get(__ret__, 'id'))
+        id=pulumi.get(__ret__, 'id'),
+        service_principals=pulumi.get(__ret__, 'service_principals'))
 def get_service_principals_output(application_ids: Optional[pulumi.Input[Optional[Sequence[_builtins.str]]]] = None,
                                   display_name_contains: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
+                                  service_principals: Optional[pulumi.Input[Optional[Sequence[Union['GetServicePrincipalsServicePrincipalArgs', 'GetServicePrincipalsServicePrincipalArgsDict']]]]] = None,
                                   opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetServicePrincipalsResult]:
     """
     Retrieves `application_ids` of all ServicePrincipal based on their `display_name`
 
     > This data source can be used with an account or workspace-level provider.
 
+    ## Example Usage
+
+    Adding all service principals of which display name contains `my-spn` to admin group
+
+    ```python
+    import pulumi
+    import pulumi_databricks as databricks
+    import pulumi_std as std
+
+    admins = databricks.get_group(display_name="admins")
+    spns = databricks.get_service_principals(display_name_contains="my-spn")
+    spn = {__key: databricks.get_service_principal(application_id=__value) for __key, __value in std.toset(input=spns.application_ids).result}
+    my_member_spn = []
+    for range in [{"key": k, "value": v} for [k, v] in enumerate(std.toset(input=spns.application_ids).result)]:
+        my_member_spn.append(databricks.GroupMember(f"my_member_spn-{range['key']}",
+            group_id=admins.id,
+            member_id=spn[range["value"]].sp_id))
+    ```
+
+    ## Related Resources
+
+    The following resources are used in the same context:
+
+    - End to end workspace management guide.
+    - get_current_user data to retrieve information about User or databricks_service_principal, that is calling Databricks REST API.
+    - 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.
+    - Group data to retrieve information about Group members, entitlements and instance profiles.
+    - GroupInstanceProfile to attach InstanceProfile (AWS) to databricks_group.
+    - GroupMember to attach users and groups as group members.
+    - Permissions to manage [access control](https://docs.databricks.com/security/access-control/index.html) in Databricks workspace.
+    - databricks_service principal to manage service principals
+
 
-    :param Sequence[_builtins.str] application_ids: List of `application_ids` of service principals.  Individual service principal can be retrieved using ServicePrincipal data source
+    :param Sequence[_builtins.str] application_ids: List of `application_ids` of service principals.  Individual service principal can be retrieved using ServicePrincipal data source or from `service_principals` attribute.
     :param _builtins.str display_name_contains: Only return ServicePrincipal display name that match the given name string
+    :param Sequence[Union['GetServicePrincipalsServicePrincipalArgs', 'GetServicePrincipalsServicePrincipalArgsDict']] service_principals: List of objects describing individual service principals. Each object has the following attributes:
     """
     __args__ = dict()
     __args__['applicationIds'] = application_ids
     __args__['displayNameContains'] = display_name_contains
+    __args__['servicePrincipals'] = service_principals
     opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
     __ret__ = pulumi.runtime.invoke_output('databricks:index/getServicePrincipals:getServicePrincipals', __args__, opts=opts, typ=GetServicePrincipalsResult)
     return __ret__.apply(lambda __response__: GetServicePrincipalsResult(
         application_ids=pulumi.get(__response__, 'application_ids'),
         display_name_contains=pulumi.get(__response__, 'display_name_contains'),
-        id=pulumi.get(__response__, 'id')))
+        id=pulumi.get(__response__, 'id'),
+        service_principals=pulumi.get(__response__, 'service_principals')))

pulumi_databricks/get_spark_version.py

@@ -216,7 +216,7 @@ def get_spark_version(beta: Optional[_builtins.bool] = None,
     :param _builtins.bool ml: if we should limit the search only to ML runtimes. Default to `false`.
     :param _builtins.bool photon: if we should limit the search only to Photon runtimes. Default to `false`. *Deprecated with DBR 14.0 release. Specify `runtime_engine=\\"PHOTON\\"` in the cluster configuration instead!*
     :param Union['GetSparkVersionProviderConfigArgs', 'GetSparkVersionProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
-    :param _builtins.str scala: if we should limit the search only to runtimes that are based on specific Scala version. Default to `2.12`.
+    :param _builtins.str scala: if we should limit the search only to runtimes that are based on specific Scala version. Default to `2.1` to select either `2.12` or `2.13` depending on the DBR version (for DBR that has both `2.12` and `2.13` flavors, `2.12` is returned by default).
     :param _builtins.str spark_version: if we should limit the search only to runtimes that are based on specific Spark version. Default to empty string.  It could be specified as `3`, or `3.0`, or full version, like, `3.0.1`.
     """
     __args__ = dict()
@@ -312,7 +312,7 @@ def get_spark_version_output(beta: Optional[pulumi.Input[Optional[_builtins.bool
     :param _builtins.bool ml: if we should limit the search only to ML runtimes. Default to `false`.
     :param _builtins.bool photon: if we should limit the search only to Photon runtimes. Default to `false`. *Deprecated with DBR 14.0 release. Specify `runtime_engine=\\"PHOTON\\"` in the cluster configuration instead!*
     :param Union['GetSparkVersionProviderConfigArgs', 'GetSparkVersionProviderConfigArgsDict'] provider_config: Configure the provider for management through account provider. This block consists of the following fields:
-    :param _builtins.str scala: if we should limit the search only to runtimes that are based on specific Scala version. Default to `2.12`.
+    :param _builtins.str scala: if we should limit the search only to runtimes that are based on specific Scala version. Default to `2.1` to select either `2.12` or `2.13` depending on the DBR version (for DBR that has both `2.12` and `2.13` flavors, `2.12` is returned by default).
     :param _builtins.str spark_version: if we should limit the search only to runtimes that are based on specific Spark version. Default to empty string.  It could be specified as `3`, or `3.0`, or full version, like, `3.0.1`.
     """
     __args__ = dict()

pulumi_databricks/get_tag_policies.py

@@ -73,7 +73,7 @@ def get_tag_policies(page_size: Optional[_builtins.int] = None,
     """
     [![Public Preview](https://img.shields.io/badge/Release_Stage-Public_Preview-yellowgreen)](https://docs.databricks.com/aws/en/release-notes/release-types)
 
-    This data source can be used to list all tag policies in the account.
+    This data source can be used to list all tag policies in the account. Manage tag policy permissions using the `AccessControlRuleSet` resource.
 
     > **Note** This resource can only be used with a workspace-level provider!
 
@@ -108,7 +108,7 @@ def get_tag_policies_output(page_size: Optional[pulumi.Input[Optional[_builtins.
     """
     [![Public Preview](https://img.shields.io/badge/Release_Stage-Public_Preview-yellowgreen)](https://docs.databricks.com/aws/en/release-notes/release-types)
 
-    This data source can be used to list all tag policies in the account.
+    This data source can be used to list all tag policies in the account. Manage tag policy permissions using the `AccessControlRuleSet` resource.
 
     > **Note** This resource can only be used with a workspace-level provider!
 

pulumi_databricks/get_tag_policy.py

@@ -115,7 +115,7 @@ def get_tag_policy(tag_key: Optional[_builtins.str] = None,
     """
     [![Public Preview](https://img.shields.io/badge/Release_Stage-Public_Preview-yellowgreen)](https://docs.databricks.com/aws/en/release-notes/release-types)
 
-    This data source can be used to get a single tag policy by its tag key.
+    This data source can be used to get a single tag policy by its tag key. Manage tag policy permissions using the `AccessControlRuleSet` resource.
 
     > **Note** This resource can only be used with a workspace-level provider!
 
@@ -147,7 +147,7 @@ def get_tag_policy_output(tag_key: Optional[pulumi.Input[_builtins.str]] = None,
     """
     [![Public Preview](https://img.shields.io/badge/Release_Stage-Public_Preview-yellowgreen)](https://docs.databricks.com/aws/en/release-notes/release-types)
 
-    This data source can be used to get a single tag policy by its tag key.
+    This data source can be used to get a single tag policy by its tag key. Manage tag policy permissions using the `AccessControlRuleSet` resource.
 
     > **Note** This resource can only be used with a workspace-level provider!
 

pulumi_databricks/get_users.py

@@ -0,0 +1,194 @@
+# coding=utf-8
+# *** WARNING: this file was generated by pulumi-language-python. ***
+# *** Do not edit by hand unless you're certain you know what you are doing! ***
+
+import builtins as _builtins
+import warnings
+import sys
+import pulumi
+import pulumi.runtime
+from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict, TypeAlias
+else:
+    from typing_extensions import NotRequired, TypedDict, TypeAlias
+from . import _utilities
+from . import outputs
+from ._inputs import *
+
+__all__ = [
+    'GetUsersResult',
+    'AwaitableGetUsersResult',
+    'get_users',
+    'get_users_output',
+]
+
+@pulumi.output_type
+class GetUsersResult:
+    """
+    A collection of values returned by getUsers.
+    """
+    def __init__(__self__, extra_attributes=None, filter=None, id=None, users=None):
+        if extra_attributes and not isinstance(extra_attributes, str):
+            raise TypeError("Expected argument 'extra_attributes' to be a str")
+        pulumi.set(__self__, "extra_attributes", extra_attributes)
+        if filter and not isinstance(filter, str):
+            raise TypeError("Expected argument 'filter' to be a str")
+        pulumi.set(__self__, "filter", filter)
+        if id and not isinstance(id, str):
+            raise TypeError("Expected argument 'id' to be a str")
+        pulumi.set(__self__, "id", id)
+        if users and not isinstance(users, list):
+            raise TypeError("Expected argument 'users' to be a list")
+        pulumi.set(__self__, "users", users)
+
+    @_builtins.property
+    @pulumi.getter(name="extraAttributes")
+    def extra_attributes(self) -> Optional[_builtins.str]:
+        return pulumi.get(self, "extra_attributes")
+
+    @_builtins.property
+    @pulumi.getter
+    def filter(self) -> Optional[_builtins.str]:
+        return pulumi.get(self, "filter")
+
+    @_builtins.property
+    @pulumi.getter
+    def id(self) -> _builtins.str:
+        """
+        The provider-assigned unique ID for this managed resource.
+        """
+        return pulumi.get(self, "id")
+
+    @_builtins.property
+    @pulumi.getter
+    def users(self) -> Sequence['outputs.GetUsersUserResult']:
+        """
+        A list of users matching the specified criteria. Each user has the following attributes:
+        """
+        return pulumi.get(self, "users")
+
+
+class AwaitableGetUsersResult(GetUsersResult):
+    # pylint: disable=using-constant-test
+    def __await__(self):
+        if False:
+            yield self
+        return GetUsersResult(
+            extra_attributes=self.extra_attributes,
+            filter=self.filter,
+            id=self.id,
+            users=self.users)
+
+
+def get_users(extra_attributes: Optional[_builtins.str] = None,
+              filter: Optional[_builtins.str] = None,
+              users: Optional[Sequence[Union['GetUsersUserArgs', 'GetUsersUserArgsDict']]] = None,
+              opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetUsersResult:
+    """
+    Retrieves information about multiple User resources.
+
+    > This data source works with both the account-level and workspace-level provider.
+
+    ## Example Usage
+
+    Adding a subset of users to a group
+
+    ```python
+    import pulumi
+    import pulumi_databricks as databricks
+
+    company_users = databricks.get_users(filter="userName co \\"@domain.org\\"")
+    data_users_group = databricks.Group("data_users_group", display_name="Data Users")
+    add_users_to_group = []
+    for range in [{"key": k, "value": v} for [k, v] in enumerate({user.id: user for user in company_users.users})]:
+        add_users_to_group.append(databricks.GroupMember(f"add_users_to_group-{range['key']}",
+            group_id=data_users_group.id,
+            member_id=%!v(PANIC=Format method: runtime error: index out of range [-1])))
+    ```
+
+    ## Related Resources
+
+    The following resources are used in the same context:
+
+    - **databricks_user**: Resource to manage individual users in Databricks.
+    - **databricks_group**: Resource to manage groups in Databricks.
+    - **databricks_group_member**: Resource to manage group memberships by adding users to groups.
+    - **databricks_permissions**: Resource to manage access control in the Databricks workspace.
+    - **databricks_current_user**: Data source to retrieve information about the user or service principal that is calling the Databricks REST API.
+
+
+    :param _builtins.str extra_attributes: A comma-separated list of additional user attributes to include in the results. By default, the data source returns the following attributes: `id`, `userName`, `displayName`, and `externalId`. Use this argument to request additional attributes as needed. The list of all available attributes can be found in the [API reference](https://docs.databricks.com/api/workspace/users/list).
+    :param _builtins.str filter: Query by which the results have to be filtered. If not specified, all users will be returned. Supported operators are equals (`eq`), contains (`co`), starts with (`sw`), and not equals (`ne`). Additionally, simple expressions can be formed using logical operators `and` and `or`.
+           
+           **Examples:**
+           - User whose `displayName` equals "john":
+    :param Sequence[Union['GetUsersUserArgs', 'GetUsersUserArgsDict']] users: A list of users matching the specified criteria. Each user has the following attributes:
+    """
+    __args__ = dict()
+    __args__['extraAttributes'] = extra_attributes
+    __args__['filter'] = filter
+    __args__['users'] = users
+    opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
+    __ret__ = pulumi.runtime.invoke('databricks:index/getUsers:getUsers', __args__, opts=opts, typ=GetUsersResult).value
+
+    return AwaitableGetUsersResult(
+        extra_attributes=pulumi.get(__ret__, 'extra_attributes'),
+        filter=pulumi.get(__ret__, 'filter'),
+        id=pulumi.get(__ret__, 'id'),
+        users=pulumi.get(__ret__, 'users'))
+def get_users_output(extra_attributes: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
+                     filter: Optional[pulumi.Input[Optional[_builtins.str]]] = None,
+                     users: Optional[pulumi.Input[Optional[Sequence[Union['GetUsersUserArgs', 'GetUsersUserArgsDict']]]]] = None,
+                     opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetUsersResult]:
+    """
+    Retrieves information about multiple User resources.
+
+    > This data source works with both the account-level and workspace-level provider.
+
+    ## Example Usage
+
+    Adding a subset of users to a group
+
+    ```python
+    import pulumi
+    import pulumi_databricks as databricks
+
+    company_users = databricks.get_users(filter="userName co \\"@domain.org\\"")
+    data_users_group = databricks.Group("data_users_group", display_name="Data Users")
+    add_users_to_group = []
+    for range in [{"key": k, "value": v} for [k, v] in enumerate({user.id: user for user in company_users.users})]:
+        add_users_to_group.append(databricks.GroupMember(f"add_users_to_group-{range['key']}",
+            group_id=data_users_group.id,
+            member_id=%!v(PANIC=Format method: runtime error: index out of range [-1])))
+    ```
+
+    ## Related Resources
+
+    The following resources are used in the same context:
+
+    - **databricks_user**: Resource to manage individual users in Databricks.
+    - **databricks_group**: Resource to manage groups in Databricks.
+    - **databricks_group_member**: Resource to manage group memberships by adding users to groups.
+    - **databricks_permissions**: Resource to manage access control in the Databricks workspace.
+    - **databricks_current_user**: Data source to retrieve information about the user or service principal that is calling the Databricks REST API.
+
+
+    :param _builtins.str extra_attributes: A comma-separated list of additional user attributes to include in the results. By default, the data source returns the following attributes: `id`, `userName`, `displayName`, and `externalId`. Use this argument to request additional attributes as needed. The list of all available attributes can be found in the [API reference](https://docs.databricks.com/api/workspace/users/list).
+    :param _builtins.str filter: Query by which the results have to be filtered. If not specified, all users will be returned. Supported operators are equals (`eq`), contains (`co`), starts with (`sw`), and not equals (`ne`). Additionally, simple expressions can be formed using logical operators `and` and `or`.
+           
+           **Examples:**
+           - User whose `displayName` equals "john":
+    :param Sequence[Union['GetUsersUserArgs', 'GetUsersUserArgsDict']] users: A list of users matching the specified criteria. Each user has the following attributes:
+    """
+    __args__ = dict()
+    __args__['extraAttributes'] = extra_attributes
+    __args__['filter'] = filter
+    __args__['users'] = users
+    opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
+    __ret__ = pulumi.runtime.invoke_output('databricks:index/getUsers:getUsers', __args__, opts=opts, typ=GetUsersResult)
+    return __ret__.apply(lambda __response__: GetUsersResult(
+        extra_attributes=pulumi.get(__response__, 'extra_attributes'),
+        filter=pulumi.get(__response__, 'filter'),
+        id=pulumi.get(__response__, 'id'),
+        users=pulumi.get(__response__, 'users')))

pulumi_databricks/get_workspace_entity_tag_assignment.py

@@ -0,0 +1,180 @@
+# coding=utf-8
+# *** WARNING: this file was generated by pulumi-language-python. ***
+# *** Do not edit by hand unless you're certain you know what you are doing! ***
+
+import builtins as _builtins
+import warnings
+import sys
+import pulumi
+import pulumi.runtime
+from typing import Any, Mapping, Optional, Sequence, Union, overload
+if sys.version_info >= (3, 11):
+    from typing import NotRequired, TypedDict, TypeAlias
+else:
+    from typing_extensions import NotRequired, TypedDict, TypeAlias
+from . import _utilities
+
+__all__ = [
+    'GetWorkspaceEntityTagAssignmentResult',
+    'AwaitableGetWorkspaceEntityTagAssignmentResult',
+    'get_workspace_entity_tag_assignment',
+    'get_workspace_entity_tag_assignment_output',
+]
+
+@pulumi.output_type
+class GetWorkspaceEntityTagAssignmentResult:
+    """
+    A collection of values returned by getWorkspaceEntityTagAssignment.
+    """
+    def __init__(__self__, entity_id=None, entity_type=None, id=None, tag_key=None, tag_value=None):
+        if entity_id and not isinstance(entity_id, str):
+            raise TypeError("Expected argument 'entity_id' to be a str")
+        pulumi.set(__self__, "entity_id", entity_id)
+        if entity_type and not isinstance(entity_type, str):
+            raise TypeError("Expected argument 'entity_type' to be a str")
+        pulumi.set(__self__, "entity_type", entity_type)
+        if id and not isinstance(id, str):
+            raise TypeError("Expected argument 'id' to be a str")
+        pulumi.set(__self__, "id", id)
+        if tag_key and not isinstance(tag_key, str):
+            raise TypeError("Expected argument 'tag_key' to be a str")
+        pulumi.set(__self__, "tag_key", tag_key)
+        if tag_value and not isinstance(tag_value, str):
+            raise TypeError("Expected argument 'tag_value' to be a str")
+        pulumi.set(__self__, "tag_value", tag_value)
+
+    @_builtins.property
+    @pulumi.getter(name="entityId")
+    def entity_id(self) -> _builtins.str:
+        """
+        (string) - The identifier of the entity to which the tag is assigned
+        """
+        return pulumi.get(self, "entity_id")
+
+    @_builtins.property
+    @pulumi.getter(name="entityType")
+    def entity_type(self) -> _builtins.str:
+        """
+        (string) - The type of entity to which the tag is assigned. Allowed values are dashboards, geniespaces
+        """
+        return pulumi.get(self, "entity_type")
+
+    @_builtins.property
+    @pulumi.getter
+    def id(self) -> _builtins.str:
+        """
+        The provider-assigned unique ID for this managed resource.
+        """
+        return pulumi.get(self, "id")
+
+    @_builtins.property
+    @pulumi.getter(name="tagKey")
+    def tag_key(self) -> _builtins.str:
+        """
+        (string) - The key of the tag. The characters , . : / - = and leading/trailing spaces are not allowed
+        """
+        return pulumi.get(self, "tag_key")
+
+    @_builtins.property
+    @pulumi.getter(name="tagValue")
+    def tag_value(self) -> _builtins.str:
+        """
+        (string) - The value of the tag
+        """
+        return pulumi.get(self, "tag_value")
+
+
+class AwaitableGetWorkspaceEntityTagAssignmentResult(GetWorkspaceEntityTagAssignmentResult):
+    # pylint: disable=using-constant-test
+    def __await__(self):
+        if False:
+            yield self
+        return GetWorkspaceEntityTagAssignmentResult(
+            entity_id=self.entity_id,
+            entity_type=self.entity_type,
+            id=self.id,
+            tag_key=self.tag_key,
+            tag_value=self.tag_value)
+
+
+def get_workspace_entity_tag_assignment(entity_id: Optional[_builtins.str] = None,
+                                        entity_type: Optional[_builtins.str] = None,
+                                        tag_key: Optional[_builtins.str] = None,
+                                        opts: Optional[pulumi.InvokeOptions] = None) -> AwaitableGetWorkspaceEntityTagAssignmentResult:
+    """
+    [![Public Beta](https://img.shields.io/badge/Release_Stage-Public_Beta-orange)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
+    This data source allows you to get information about a tag assignment for a specific workspace scoped entity using the entity type, entity id, and tag key.
+
+    ## Example Usage
+
+    ```python
+    import pulumi
+    import pulumi_databricks as databricks
+
+    dashboard_tag = databricks.get_workspace_entity_tag_assignment(entity_type="dashboards",
+        entity_id="2807324866692453",
+        tag_key="sensitivity_level")
+    geniespace_tag = databricks.get_workspace_entity_tag_assignment(entity_type="geniespaces",
+        entity_id="2807324866692453",
+        tag_key="sensitivity_level")
+    ```
+
+
+    :param _builtins.str entity_id: The identifier of the entity to which the tag is assigned
+    :param _builtins.str entity_type: The type of entity to which the tag is assigned. Allowed values are dashboards, geniespaces
+    :param _builtins.str tag_key: The key of the tag. The characters , . : / - = and leading/trailing spaces are not allowed
+    """
+    __args__ = dict()
+    __args__['entityId'] = entity_id
+    __args__['entityType'] = entity_type
+    __args__['tagKey'] = tag_key
+    opts = pulumi.InvokeOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
+    __ret__ = pulumi.runtime.invoke('databricks:index/getWorkspaceEntityTagAssignment:getWorkspaceEntityTagAssignment', __args__, opts=opts, typ=GetWorkspaceEntityTagAssignmentResult).value
+
+    return AwaitableGetWorkspaceEntityTagAssignmentResult(
+        entity_id=pulumi.get(__ret__, 'entity_id'),
+        entity_type=pulumi.get(__ret__, 'entity_type'),
+        id=pulumi.get(__ret__, 'id'),
+        tag_key=pulumi.get(__ret__, 'tag_key'),
+        tag_value=pulumi.get(__ret__, 'tag_value'))
+def get_workspace_entity_tag_assignment_output(entity_id: Optional[pulumi.Input[_builtins.str]] = None,
+                                               entity_type: Optional[pulumi.Input[_builtins.str]] = None,
+                                               tag_key: Optional[pulumi.Input[_builtins.str]] = None,
+                                               opts: Optional[Union[pulumi.InvokeOptions, pulumi.InvokeOutputOptions]] = None) -> pulumi.Output[GetWorkspaceEntityTagAssignmentResult]:
+    """
+    [![Public Beta](https://img.shields.io/badge/Release_Stage-Public_Beta-orange)](https://docs.databricks.com/aws/en/release-notes/release-types)
+
+    This data source allows you to get information about a tag assignment for a specific workspace scoped entity using the entity type, entity id, and tag key.
+
+    ## Example Usage
+
+    ```python
+    import pulumi
+    import pulumi_databricks as databricks
+
+    dashboard_tag = databricks.get_workspace_entity_tag_assignment(entity_type="dashboards",
+        entity_id="2807324866692453",
+        tag_key="sensitivity_level")
+    geniespace_tag = databricks.get_workspace_entity_tag_assignment(entity_type="geniespaces",
+        entity_id="2807324866692453",
+        tag_key="sensitivity_level")
+    ```
+
+
+    :param _builtins.str entity_id: The identifier of the entity to which the tag is assigned
+    :param _builtins.str entity_type: The type of entity to which the tag is assigned. Allowed values are dashboards, geniespaces
+    :param _builtins.str tag_key: The key of the tag. The characters , . : / - = and leading/trailing spaces are not allowed
+    """
+    __args__ = dict()
+    __args__['entityId'] = entity_id
+    __args__['entityType'] = entity_type
+    __args__['tagKey'] = tag_key
+    opts = pulumi.InvokeOutputOptions.merge(_utilities.get_invoke_opts_defaults(), opts)
+    __ret__ = pulumi.runtime.invoke_output('databricks:index/getWorkspaceEntityTagAssignment:getWorkspaceEntityTagAssignment', __args__, opts=opts, typ=GetWorkspaceEntityTagAssignmentResult)
+    return __ret__.apply(lambda __response__: GetWorkspaceEntityTagAssignmentResult(
+        entity_id=pulumi.get(__response__, 'entity_id'),
+        entity_type=pulumi.get(__response__, 'entity_type'),
+        id=pulumi.get(__response__, 'id'),
+        tag_key=pulumi.get(__response__, 'tag_key'),
+        tag_value=pulumi.get(__response__, 'tag_value')))