mirror of https://github.com/databricks/cli.git
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
|
from dataclasses import dataclass
|
||
|
|
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, VariableOrOptional
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
from typing_extensions import Self
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass(kw_only=True)
|
||
|
|
class Permission:
|
||
|
|
""""""
|
||
|
|
|
||
|
|
level: VariableOr[str]
|
||
|
|
"""
|
||
|
|
The allowed permission for user, group, service principal defined for this permission.
|
||
|
|
"""
|
||
|
|
|
||
|
|
group_name: VariableOrOptional[str] = None
|
||
|
|
"""
|
||
|
|
The name of the group that has the permission set in level.
|
||
|
|
"""
|
||
|
|
|
||
|
|
service_principal_name: VariableOrOptional[str] = None
|
||
|
|
"""
|
||
|
|
The name of the service principal that has the permission set in level.
|
||
|
|
"""
|
||
|
|
|
||
|
|
user_name: VariableOrOptional[str] = None
|
||
|
|
"""
|
||
|
|
The name of the user that has the permission set in level.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __post_init__(self):
|
||
|
|
union_fields = [
|
||
|
|
self.user_name,
|
||
|
|
self.service_principal_name,
|
||
|
|
self.group_name,
|
||
|
|
]
|
||
|
|
|
||
|
|
if sum(f is not None for f in union_fields) != 1:
|
||
|
|
raise ValueError(
|
||
|
|
"Permission must specify exactly one of 'user_name', 'service_principal_name', 'group_name'"
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def from_dict(cls, value: "PermissionDict") -> "Self":
|
||
|
|
return _transform(cls, value)
|
||
|
|
|
||
|
|
def as_dict(self) -> "PermissionDict":
|
||
|
|
return _transform_to_json_value(self) # type:ignore
|
||
|
|
|
||
|
|
|
||
|
|
class PermissionDict(TypedDict, total=False):
|
||
|
|
""""""
|
||
|
|
|
||
|
|
level: VariableOr[str]
|
||
|
|
"""
|
||
|
|
The allowed permission for user, group, service principal defined for this permission.
|
||
|
|
"""
|
||
|
|
|
||
|
|
group_name: VariableOrOptional[str]
|
||
|
|
"""
|
||
|
|
The name of the group that has the permission set in level.
|
||
|
|
"""
|
||
|
|
|
||
|
|
service_principal_name: VariableOrOptional[str]
|
||
|
|
"""
|
||
|
|
The name of the service principal that has the permission set in level.
|
||
|
|
"""
|
||
|
|
|
||
|
|
user_name: VariableOrOptional[str]
|
||
|
|
"""
|
||
|
|
The name of the user that has the permission set in level.
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
PermissionParam = PermissionDict | Permission
|