def make_donor(self):
"""Take away all the user's permissions."""
self.user.is_staff = False
self.user.is_superuser = False
self.user.groups.remove(get_group_by_name(self.ADMIN_GROUP))
self.user.groups.remove(get_group_by_name(self.AMBASSADOR_GROUP))
self.user.save()
def make_donor(self):
"""Take away all the user's permissions."""
self.user.is_staff = False
self.user.is_superuser = False
self.user.groups.remove(get_group_by_name(self.ADMIN_GROUP))
self.user.groups.remove(get_group_by_name(self.AMBASSADOR_GROUP))
self.user.save()
def _assert_group_relationship(self, user, group_name, rel_in):
"""Assert that a user is or is not in a given group.
:user: {User} the user to check
:group_name: {string} the group name to check
:rel_in: {boolean} if true, assert that the user is IN the group.
otherwise, assert that the user is NOT in the group
"""
group = get_group_by_name(group_name)
if rel_in:
self.assertIn(group, user.groups.all())
else:
self.assertNotIn(group, user.groups.all())
def _assert_group_relationship(self, user, group_name, rel_in):
"""Assert that a user is or is not in a given group.
:user: {User} the user to check
:group_name: {string} the group name to check
:rel_in: {boolean} if true, assert that the user is IN the group.
otherwise, assert that the user is NOT in the group
"""
group = get_group_by_name(group_name)
if rel_in:
self.assertIn(group, user.groups.all())
else:
self.assertNotIn(group, user.groups.all())
def is_administrator(self):
return get_group_by_name(self.ADMIN_GROUP) in self.user.groups.all()
def is_ambassador(self):
return get_group_by_name(
self.AMBASSADOR_GROUP
) in self.user.groups.all()
def make_ambassador(self):
self.user.is_staff = False
self.user.is_superuser = False
self.user.groups.remove(get_group_by_name(self.ADMIN_GROUP))
self.user.groups.add(get_group_by_name(self.AMBASSADOR_GROUP))
self.user.save()
def make_administrator(self):
self.user.groups.add(get_group_by_name(self.AMBASSADOR_GROUP))
self.user.groups.add(get_group_by_name(self.ADMIN_GROUP))
self.user.is_staff = True
self.user.is_superuser = True
self.user.save()
def make_ambassador(self):
self.user.is_staff = False
self.user.is_superuser = False
self.user.groups.remove(get_group_by_name(self.ADMIN_GROUP))
self.user.groups.add(get_group_by_name(self.AMBASSADOR_GROUP))
self.user.save()
def make_administrator(self):
self.user.groups.add(get_group_by_name(self.AMBASSADOR_GROUP))
self.user.groups.add(get_group_by_name(self.ADMIN_GROUP))
self.user.is_staff = True
self.user.is_superuser = True
self.user.save()
def is_administrator(self):
return get_group_by_name(self.ADMIN_GROUP) in self.user.groups.all()
def is_ambassador(self):
return get_group_by_name(
self.AMBASSADOR_GROUP) in self.user.groups.all()
def test_correct_groups_exist(self):
get_group_by_name(RevolvUserProfile.AMBASSADOR_GROUP)
get_group_by_name(RevolvUserProfile.ADMIN_GROUP)
def test_correct_groups_exist(self):
get_group_by_name(RevolvUserProfile.AMBASSADOR_GROUP)
get_group_by_name(RevolvUserProfile.ADMIN_GROUP)