From cb0df5aa64808f0575491b3f34f170c9313d562b Mon Sep 17 00:00:00 2001 From: Ryan Ahearn Date: Tue, 9 May 2023 22:10:36 -0400 Subject: [PATCH] Only print user report when run directly as a script --- terraform/ops/cloudgov_user_report.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/terraform/ops/cloudgov_user_report.py b/terraform/ops/cloudgov_user_report.py index e4c6d7984..7a26fe28c 100644 --- a/terraform/ops/cloudgov_user_report.py +++ b/terraform/ops/cloudgov_user_report.py @@ -24,9 +24,15 @@ class RoleCollector: else: self._map[user.guid]["roles"].append(role) - def iterator(self): - for key in self._map.keys(): - yield self._map[key] + def print(self): + for user_roles in self._map.values(): + user = user_roles['user'] + print(f"{user.type}: {user.username} has roles:") + for role in user_roles['roles']: + if role.space: + print(f" {role.type} in {role.space.name}") + else: + print(f" {role.type}") role_collector = RoleCollector() @@ -74,11 +80,5 @@ for role in map(Role, client.v3.roles.list(space_guids=space_guids, include="use role_collector.add(role) -for user_roles in role_collector.iterator(): - user = user_roles['user'] - print(f"{user.type}: {user.username} has roles:") - for role in user_roles['roles']: - if role.space: - print(f" {role.type} in {role.space.name}") - else: - print(f" {role.type}") +if __name__ == '__main__': + role_collector.print()