def save(self, commit=True): instance = super().save(commit=False) if self.cleaned_data.get('has_perm_member'): add_group(instance, 'members') else: remove_group(instance, 'members') if commit: instance.save() return instance
def processing(self, user, profile, msg): # Умови при яких зміни не відбудуться: if not has_group(user, 'members'): msg.title = user.username msg.type = msgType.NoChange msg.message = "Акаунт вже позбавлений цього права доступу!" else: # Робимо зміни: remove_group(user, 'members') user.save() msg.title = user.username msg.type = msgType.Change msg.message = "Право доступу вилучено!" e_msg_body = "Ваш акаунт на сайті позбавлений права доступу " \ "члена кооперативу." self.send_e_mail(user, e_msg_body) return user, msg
def test_remove_group_gives_false_if_no_group(self): remove_group(self.user, 'members') self.assertFalse(has_group(self.user, 'members'))
def test_remove_group(self): DummyUser().create_dummy_group(group_name='members') add_group(self.user, 'members') self.assertTrue(has_group(self.user, 'members')) remove_group(self.user, 'members') self.assertFalse(has_group(self.user, 'members'))