def assertInitialRoles(self):
for u in self.user_roles:
user_role = UserRole.get(u.get_id)
self.assertEqual(
user_role.permissions,
UserRolePresets.get_permissions(user_role.name)
)
def _assertInitialRoles(self):
for u in self.user_roles:
user_role = UserRole.get(u.get_id)
self.assertEqual(
user_role.permissions,
UserRolePresets.get_permissions(user_role.name)
)
def handle(self, *args, **options):
for domain in Domain.get_all():
UserRole.get_or_create_with_permissions(
domain.name,
UserRolePresets.get_permissions(UserRolePresets.BILLING_ADMIN),
UserRolePresets.BILLING_ADMIN
)
def reset_initial_roles_for_domain(domain):
initial_roles = [
role for role in UserRole.objects.get_by_domain(domain)
if role.name in UserRolePresets.INITIAL_ROLES
]
for role in initial_roles:
role.set_permissions(UserRolePresets.get_permissions(role.name).to_list())
def initialize_domain_with_default_roles(domain):
"""Outside of tests this is only called when creating a new domain"""
for role_name in UserRolePresets.INITIAL_ROLES:
UserRole.create(domain, role_name, permissions=UserRolePresets.get_permissions(role_name))
