def change_view(self, request, *args, **kwargs):
# for non-superuser
if not request.user.is_superuser:
try:
self.fieldsets = self.staff_fieldsets
response = UserAdmin.change_view(self, request, *args, **kwargs)
finally:
# Reset fieldsets to its original value
self.fieldsets = UserAdmin.fieldsets
return response
else:
return UserAdmin.change_view(self, request, *args, **kwargs)
def change_view(self, request, *args, **kwargs):
# for non-superuser
if not request.user.is_superuser:
try:
self.fieldsets = self.staff_fieldsets
response = UserAdmin.change_view(self, request, *args, **kwargs)
finally:
# Reset fieldsets to its original value
self.fieldsets = UserAdmin.fieldsets
return response
else:
return UserAdmin.change_view(self, request, *args, **kwargs)
def change_view(self, request, *args, **kwargs):
# for superuser
try:
if request.user.is_superuser:
self.fieldsets = self.standard_fieldsets + self.superuser_fieldsets
else:
self.fieldsets = self.standard_fieldsets + self.staff_fieldsets
response = UserAdmin.change_view(self, request, *args, **kwargs)
finally:
# Reset fieldsets to its original value
self.fieldsets = self.standard_fieldsets
return response
def test_submit_row(self):
"""
submit_row template tag should pass whole context.
"""
request = self.request_factory.get(reverse('admin:auth_user_change', args=[self.superuser.pk]))
request.user = self.superuser
admin = UserAdmin(User, site)
extra_context = {'extra': True}
response = admin.change_view(request, str(self.superuser.pk), extra_context=extra_context)
template_context = submit_row(response.context_data)
self.assertIs(template_context['extra'], True)
self.assertIs(template_context['show_save'], True)
def test_submit_row(self):
"""
submit_row template tag should pass whole context.
"""
request = self.request_factory.get(reverse('admin:auth_user_change', args=[self.superuser.pk]))
request.user = self.superuser
admin = UserAdmin(User, site)
extra_context = {'extra': True}
response = admin.change_view(request, str(self.superuser.pk), extra_context=extra_context)
template_context = submit_row(response.context_data)
self.assertIs(template_context['extra'], True)
self.assertIs(template_context['show_save'], True)
def change_view(self, request, *args, **kwargs):
# for superuser
try:
if request.user.is_superuser:
self.fieldsets = self.standard_fieldsets + self.superuser_fieldsets
else:
self.fieldsets = self.standard_fieldsets + self.staff_fieldsets
response = UserAdmin.change_view(self, request, *args, **kwargs)
finally:
# Reset fieldsets to its original value
self.fieldsets = self.standard_fieldsets
return response
def change_view(self, request, *args, **kwargs):
try:
group, created = Group.objects.get_or_create(name = 'Admissions')
user_perm_list = UserPermissionList.objects.get(
user=request.user
)
except UserPermissionList.DoesNotExist:
groups = list()
self.form = MyUserAdminForm
# for non-superuser
if not request.user.is_superuser and request.user.has_perm('auth.change_users') and group.id in user_perm_list.group_fk_list:
self.form = AdmissionsAdminForm
try:
self.fieldsets = self.staff_fieldsets
response = UserAdmin.change_view(self, request, *args, **kwargs)
finally:
# Reset fieldsets to its original value
self.fieldsets = UserAdmin.fieldsets
return response
else:
return UserAdmin.change_view(self, request, *args, **kwargs)
def change_view(self, request, *args, **kwargs):
try:
group, created = Group.objects.get_or_create(name='Admissions')
user_perm_list = UserPermissionList.objects.get(user=request.user)
except UserPermissionList.DoesNotExist:
groups = list()
self.form = MyUserAdminForm
# for non-superuser
if not request.user.is_superuser and request.user.has_perm(
'auth.change_users'
) and group.id in user_perm_list.group_fk_list:
self.form = AdmissionsAdminForm
try:
self.fieldsets = self.staff_fieldsets
response = UserAdmin.change_view(self, request, *args,
**kwargs)
finally:
# Reset fieldsets to its original value
self.fieldsets = UserAdmin.fieldsets
return response
else:
return UserAdmin.change_view(self, request, *args, **kwargs)
def test_override_show_save_and_add_another(self):
request = self.request_factory.get(
reverse('admin:auth_user_change', args=[self.superuser.pk]),
)
request.user = self.superuser
admin = UserAdmin(User, site)
for extra_context, expected_flag in (
({}, True), # Default.
({'show_save_and_add_another': False}, False),
):
with self.subTest(show_save_and_add_another=expected_flag):
response = admin.change_view(
request,
str(self.superuser.pk),
extra_context=extra_context,
)
template_context = submit_row(response.context_data)
self.assertIs(template_context['show_save_and_add_another'], expected_flag)
