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))