angstwad

import argparse
import functools
import logging
import os
import signal
import sys
import time

_LOG_FORMAT = "%(levelname)s:%(asctime)s:%(name)s:%(message)s"
logging.basicConfig(stream=sys.stdout, format=_LOG_FORMAT)

angstwad

# Copyright 2020 Paul Durivage <pauldurivage+github@gmail.com>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,

angstwad

roles/axt.admin
roles/billing.admin
roles/billing.creator
roles/compute.xpnAdmin
roles/container.hostServiceAgentUser
roles/datacatalog.categoryFineGrainedReader
roles/datafusion.serviceAgent
roles/iam.serviceAccountTokenCreator
roles/iap.httpsResourceAccessor
roles/orgpolicy.policyAdmin

angstwad

while remaining:
    for perm in remaining:
        # get set of roles which contain permission
        satisfy = perms_to_roles[perm]

        # sort roles by the number of permissions they contain, select role
        # with the most
        sorted_roles = sorted((role, perms_counts[role]) 
                              for role in satisfy)
        selected = sorted_roles[-1][0]

angstwad

remaining = unique_perms - roles_to_perms['roles/owner']

angstwad

selected_roles = {role_with_most_perms,}

angstwad

role_with_most_perms = roles_sorted_by_perms_asc[-1][0]

angstwad

subset_roles = set()
for this_role, _ in roles_sorted_by_perms_asc:
    for other_role, other_perms in roles_to_perms.items():
        if this_role == other_role:
            continue

        this_perms = roles_to_perms[this_role]

        if this_perms.issubset(other_perms):
            subset_roles.add(this_role)

angstwad

roles_sorted_by_perms_asc = sorted(perms_counts.items(),
                                   key=lambda x: x[1])

angstwad

import collections

roles_to_perms = collections.defaultdict(set)
perms_to_roles = collections.defaultdict(set)
perms_counts = collections.defaultdict(int)
unique_perms = set()

for role_name, role_data in raw_role_data.items():
    for perm in role_data.get('includedPermissions', tuple()):
        roles_to_perms[role_name].add(perm)