Also disable via admin
This commit is contained in:
parent
1c2b06521e
commit
1744428dd0
@ -1,4 +1,7 @@
|
||||
import types
|
||||
|
||||
from django.contrib.admin.sites import AdminSite
|
||||
from django.contrib.auth.models import User
|
||||
from django.test import TestCase
|
||||
from django.utils import timezone
|
||||
|
||||
@ -6,6 +9,7 @@ from documents import index
|
||||
from documents.admin import DocumentAdmin
|
||||
from documents.models import Document
|
||||
from documents.tests.utils import DirectoriesMixin
|
||||
from paperless.admin import PaperlessUserAdmin
|
||||
|
||||
|
||||
class TestDocumentAdmin(DirectoriesMixin, TestCase):
|
||||
@ -64,3 +68,22 @@ class TestDocumentAdmin(DirectoriesMixin, TestCase):
|
||||
created=timezone.make_aware(timezone.datetime(2020, 4, 12)),
|
||||
)
|
||||
self.assertEqual(self.doc_admin.created_(doc), "2020-04-12")
|
||||
|
||||
|
||||
class TestPaperlessAdmin(DirectoriesMixin, TestCase):
|
||||
def setUp(self) -> None:
|
||||
super().setUp()
|
||||
self.user_admin = PaperlessUserAdmin(model=User, admin_site=AdminSite())
|
||||
|
||||
def test_only_superuser_can_change_superuser(self):
|
||||
non_superuser = User.objects.create(username="requestuser")
|
||||
user = User.objects.create(username="test", is_superuser=False)
|
||||
|
||||
data = {"is_superuser": True}
|
||||
form = self.user_admin.form(data, instance=user)
|
||||
form.request = types.SimpleNamespace(user=non_superuser)
|
||||
self.assertFalse(form.is_valid())
|
||||
self.assertEqual(
|
||||
form.errors.get("__all__"),
|
||||
["Superuser status can only be changed by a superuser"],
|
||||
)
|
||||
|
48
src/paperless/admin.py
Normal file
48
src/paperless/admin.py
Normal file
@ -0,0 +1,48 @@
|
||||
from django import forms
|
||||
from django.contrib import admin
|
||||
from django.contrib.auth.admin import UserAdmin
|
||||
from django.contrib.auth.models import User
|
||||
|
||||
|
||||
class PaperlessUserForm(forms.ModelForm):
|
||||
class Meta:
|
||||
model = User
|
||||
fields = [
|
||||
"username",
|
||||
"first_name",
|
||||
"last_name",
|
||||
"email",
|
||||
"is_staff",
|
||||
"is_active",
|
||||
"is_superuser",
|
||||
"groups",
|
||||
"user_permissions",
|
||||
]
|
||||
|
||||
def clean(self):
|
||||
cleaned_data = super().clean()
|
||||
user_being_edited = self.instance
|
||||
is_superuser = cleaned_data.get("is_superuser")
|
||||
|
||||
if (
|
||||
not self.request.user.is_superuser
|
||||
and is_superuser != user_being_edited.is_superuser
|
||||
):
|
||||
raise forms.ValidationError(
|
||||
"Superuser status can only be changed by a superuser",
|
||||
)
|
||||
|
||||
return cleaned_data
|
||||
|
||||
|
||||
class PaperlessUserAdmin(UserAdmin):
|
||||
form = PaperlessUserForm
|
||||
|
||||
def get_form(self, request, obj=None, **kwargs):
|
||||
form = super().get_form(request, obj, **kwargs)
|
||||
form.request = request
|
||||
return form
|
||||
|
||||
|
||||
admin.site.unregister(User)
|
||||
admin.site.register(User, PaperlessUserAdmin)
|
Loading…
x
Reference in New Issue
Block a user