mirror of
https://github.com/datahub-project/datahub.git
synced 2026-02-01 07:42:42 +00:00
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import json
|
|
import pprint
|
|
|
|
with open(
|
|
"./metadata-service/war/src/main/resources/boot/policies.json"
|
|
) as policies_file:
|
|
all_policies = json.loads(policies_file.read())
|
|
|
|
metadata_policies = []
|
|
platform_policies = []
|
|
other_policies = []
|
|
without_info = []
|
|
|
|
metadata_privileges = set()
|
|
platform_privileges = set()
|
|
for policy in all_policies:
|
|
urn = policy["urn"]
|
|
if urn == "urn:li:dataHubPolicy:0":
|
|
root_user_platform_policy_privileges = policy["info"]["privileges"]
|
|
elif urn == "urn:li:dataHubPolicy:editor-platform-policy":
|
|
editor_platform_policy_privileges = policy["info"]["privileges"]
|
|
elif urn == "urn:li:dataHubPolicy:7":
|
|
all_user_platform_policy_privilges = policy["info"]["privileges"]
|
|
try:
|
|
doc_type = policy["info"]["type"]
|
|
privileges = policy["info"]["privileges"]
|
|
if doc_type == "METADATA":
|
|
metadata_policies.append(policy)
|
|
metadata_privileges.update(privileges)
|
|
elif doc_type == "PLATFORM":
|
|
platform_policies.append(policy)
|
|
platform_privileges.update(privileges)
|
|
else:
|
|
other_policies.append(policy)
|
|
except:
|
|
without_info.append(policy)
|
|
pprint.pprint(policy)
|
|
|
|
print(
|
|
f"""
|
|
Number of policies is {len(all_policies)}
|
|
Number of metadata_policies is {len(metadata_policies)}
|
|
Number of platform_policies is {len(platform_policies)}
|
|
Number of other is {len(other_policies)}
|
|
Number without info is {len(without_info)}
|
|
|
|
Number metadata privileges are {len(metadata_privileges)}
|
|
Number platform privileges are {len(platform_privileges)}
|
|
"""
|
|
)
|
|
|
|
diff_policies = set(platform_privileges).difference(
|
|
set(root_user_platform_policy_privileges)
|
|
)
|
|
assert len(diff_policies) == 0, f"Missing privileges for root user are {diff_policies}"
|
|
|
|
diff_policies = set(editor_platform_policy_privileges).difference(
|
|
set(all_user_platform_policy_privilges)
|
|
)
|
|
assert "MANAGE_POLICIES" not in all_user_platform_policy_privilges
|
|
assert (
|
|
len(diff_policies) == 0
|
|
), f"Missing privileges for all user policies are {diff_policies}"
|