Как предотвратить повышение разрешений в администраторе Django при предоставлении разрешения «изменение пользователя»? - PullRequest
19 голосов
/ 19 февраля 2010

У меня есть сайт Django с большой клиентской базой. Я хотел бы дать нашему отделу обслуживания клиентов возможность изменять обычные учетные записи пользователей, например, менять пароли, адреса электронной почты и т. Д. Однако, если я предоставлю кому-то встроенное разрешение auth | user | Can change user, они получат возможность установить is_superuser флаг на любом аккаунте, в том числе и своем. (!!!)

Каков наилучший способ удалить эту опцию для сотрудников, не являющихся суперпользователями? Я уверен, что это включает в себя создание подкласса django.contrib.auth.forms.UserChangeForm и подключение его к моему уже настроенному объекту UserAdmin ... каким-то образом. Но я не могу найти какую-либо документацию о том, как это сделать, и я пока недостаточно хорошо разбираюсь во внутренностях.

Ответы [ 5 ]

19 голосов
/ 19 февраля 2010

они получают возможность устанавливать флаг is_superuser для любой учетной записи, включая свою собственную. (!!!)

Мало того, они также получают возможность давать себе любые разрешения один за другим, тот же эффект ...

Я уверен, что это включает в себя создание подкласса django.contrib.auth.forms.UserChangeForm

Ну, не обязательно. Форма, которую вы видите на странице изменений администратора django, динамически создается приложением администратора и основана на UserChangeForm, но этот класс едва добавляет проверку регулярного выражения в поле username.

и подключаем его к моему уже настроенному объекту UserAdmin ...

Обычай UserAdmin - путь сюда. По сути, вы хотите изменить свойство fieldsets на что-то вроде этого:

class MyUserAdmin(UserAdmin):
    fieldsets = (
        (None, {'fields': ('username', 'password')}),
        (_('Personal info'), {'fields': ('first_name', 'last_name', 'email')}),
        # Removing the permission part
        # (_('Permissions'), {'fields': ('is_staff', 'is_active', 'is_superuser', 'user_permissions')}),
        (_('Important dates'), {'fields': ('last_login', 'date_joined')}),
        # Keeping the group parts? Ok, but they shouldn't be able to define
        # their own groups, up to you...
        (_('Groups'), {'fields': ('groups',)}),
    )

Но проблема здесь в том, что это ограничение будет распространяться на всех пользователей. Если это не то, что вы хотите, вы можете, например, переопределить change_view, чтобы вести себя по-разному в зависимости от разрешения пользователей. Фрагмент кода:

class MyUserAdmin(UserAdmin):
    staff_fieldsets = (
        (None, {'fields': ('username', 'password')}),
        (_('Personal info'), {'fields': ('first_name', 'last_name', 'email')}),
        # No permissions
        (_('Important dates'), {'fields': ('last_login', 'date_joined')}),
        (_('Groups'), {'fields': ('groups',)}),
    )

    def change_view(self, request, *args, **kwargs):
        # for non-superuser
        if not request.user.is_superuser:
            try:
                self.fieldsets = self.staff_fieldsets
                response = super(MyUserAdmin, self).change_view(request, *args, **kwargs)
            finally:
                # Reset fieldsets to its original value
                self.fieldsets = UserAdmin.fieldsets
            return response
        else:
            return super(MyUserAdmin, self).change_view(request, *args, **kwargs)
5 голосов
/ 31 марта 2017

В приведенной ниже части принятого ответа есть условие гонки, при котором, если два сотрудника пользователя одновременно пытаются получить доступ к форме администратора, один из них может получить форму суперпользователя.

try:
    self.readonly_fields = self.staff_self_readonly_fields
    response = super(MyUserAdmin, self).change_view(request, object_id, form_url, extra_context, *args, **kwargs)
finally:
    # Reset fieldsets to its original value
    self.fieldsets = UserAdmin.fieldsets

Чтобы избежать этого состояния гонки (и, на мой взгляд, улучшить общее качество решения), мы можем переопределить методы get_fieldsets() и get_readonly_fields() напрямую:

class UserAdmin(BaseUserAdmin):
    staff_fieldsets = (
        (None, {'fields': ('username')}),
        ('Personal info', {'fields': ('first_name', 'last_name', 'email')}),
        # No permissions
        ('Important dates', {'fields': ('last_login', 'date_joined')}),
    )
    staff_readonly_fields = ('username', 'first_name', 'last_name', 'email', 'last_login', 'date_joined')

    def get_fieldsets(self, request, obj=None):
        if not request.user.is_superuser:
            return self.staff_fieldsets
        else:
            return super(UserAdmin, self).get_fieldsets(request, obj)

    def get_readonly_fields(self, request, obj=None):
        if not request.user.is_superuser:
            return self.staff_readonly_fields
        else:
            return super(UserAdmin, self).get_readonly_fields(request, obj)
1 голос
/ 09 марта 2016

Большое спасибо Clément.Когда я делал то же самое для своего сайта, я пришел к выводу, что мне нужно дополнительно сделать все поля доступными только для пользователей, кроме вас.Поэтому, основываясь на ответе Клемента, я добавил поля «только для чтения» и скрытие полей пароля при просмотре, а не себя

class MyUserAdmin(UserAdmin):
    model = User
    staff_self_fieldsets = (
        (None, {'fields': ('username', 'password')}),
        (_('Personal info'), {'fields': ('first_name', 'last_name', 'email')}),
        # No permissions
        (_('Important dates'), {'fields': ('last_login', 'date_joined')}),
    )

    staff_other_fieldsets = (
        (None, {'fields': ('username', )}),
        (_('Personal info'), {'fields': ('first_name', 'last_name', 'email')}),
        # No permissions
        (_('Important dates'), {'fields': ('last_login', 'date_joined')}),
    )

    staff_self_readonly_fields = ('last_login', 'date_joined')

    def change_view(self, request, object_id, form_url='', extra_context=None, *args, **kwargs):
        # for non-superuser
        if not request.user.is_superuser:
            try:
                if int(object_id) != request.user.id:
                    self.readonly_fields = User._meta.get_all_field_names()
                    self.fieldsets = self.staff_other_fieldsets
                else:
                    self.readonly_fields = self.staff_self_readonly_fields
                    self.fieldsets = self.staff_self_fieldsets

                response = super(MyUserAdmin, self).change_view(request, object_id, form_url, extra_context, *args, **kwargs)
            except:
                logger.error('Admin change view error. Returned all readonly fields')

                self.fieldsets = self.staff_other_fieldsets
                self.readonly_fields = ('first_name', 'last_name', 'email', 'username', 'password', 'last_login', 'date_joined')
                response = super(MyUserAdmin, self).change_view(request, object_id, form_url, extra_context, *args, **kwargs)
            finally:
                # Reset fieldsets to its original value
                self.fieldsets = UserAdmin.fieldsets
                self.readonly_fields = UserAdmin.readonly_fields
            return response
        else:
            return super(MyUserAdmin, self).change_view(request, object_id, form_url, extra_context, *args, **kwargs)
0 голосов
/ 21 февраля 2019

Этот подход был составлен из нескольких полезных советов в Интернете. В этом случае мы модифицируем UserAdmin таким образом, чтобы для сотрудников, не являющихся суперпользователями с разрешением на добавление / изменение пользователя, единственными разрешениями и группами, которые они могут предоставить другому пользователю, были те, которые уже есть у сотрудника.

(для Джанго 1.11)

from django.contrib.auth.admin import UserAdmin, User
from django.contrib import admin

class RestrictedUserAdmin(UserAdmin):
    model = User

    def formfield_for_dbfield(self, db_field, **kwargs):
        field = super(RestrictedUserAdmin, self).formfield_for_dbfield(db_field, **kwargs)
        user = kwargs['request'].user
        if not user.is_superuser:
            if db_field.name == 'groups':
                field.queryset = field.queryset.filter(id__in=[i.id for i in user.groups.all()])
            if db_field.name == 'user_permissions':
                field.queryset = field.queryset.filter(id__in=[i.id for i in user.user_permissions.all()])
            if db_field.name == 'is_superuser':
                field.widget.attrs['disabled'] = True
        return field

admin.site.unregister(User)
admin.site.register(User, RestrictedUserAdmin)

Это также должно быть сделано для GroupAdmin, если пользователю дано разрешение на изменение групп.

0 голосов
/ 13 августа 2010

Полный код для django 1.1 (ограничено основной информацией пользователя для сотрудников (не суперпользователей))

from django.contrib.auth.models import User
from django.utils.translation import ugettext_lazy as _


class MyUserAdmin(UserAdmin):
   my_fieldsets = (
       (None, {'fields': ('username', 'password')}),
       (_('Personal info'), {'fields': ('first_name', 'last_name', 'email')}),
   )

   def change_view(self, request, object_id, extra_context=None):
       # for non-superuser
       print 'test'
       if not request.user.is_superuser:
           self.fieldsets = self.my_fieldsets
           response = UserAdmin.change_view(self, request, object_id,
extra_context=None)
           return response
       else:
           return UserAdmin.change_view(self, request, object_id,
extra_context=None)


admin.site.unregister(User)
admin.site.register(User, MyUserAdmin)
...