Проблема создания пользователя пользовательской модели django: не работают разрешения

Question

Я делаю пользовательскую модель пользователя в django, но каким-то образом у всех новых пользователей, созданных с помощью формы создания, отсутствуют свои назначенные права доступа (is_active, is_staff, is_admin), что не позволяет им войти в систему вообще.

Кто-нибудь может показать мне, где я ошибся?Будем благодарны за любые предложения!

Фрагменты кода:

models.py

class MyUserManager(BaseUserManager):
    use_in_migrations = True
    def create_user(self, ID, name, email, privilege, password=None):

        if not ID:
            raise ValueError('Users must have a ID number')

        user = self.model(
            ID=ID,
            name=name,
            email=self.normalize_email(email),
            privilege=privilege,
            )

        if self.privilege == 'Admin':
            self.is_superuser = True
        else:
            user.is_active = True
            user.is_staff = False
            user.is_admin = False

        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, NIP, name, email, privilege, password):
        new_user = self.create_user(
            ID,
            name,
            email,
            privilege,
            password=password,
        )

        new_user.is_superuser = True
        new_user.save(using=self._db)
        return new_user

class employee(AbstractBaseUser, PermissionsMixin):
    ID = models.IntegerField(verbose_name="ID", unique=True)
    name = models.CharField(verbose_name="Name", max_length=50, blank=True, unique=True)
    email = models.EmailField(verbose_name="email", max_length=225, default='this@mail.com')
    Department = models.CharField(max_length=50, choices=DEP_CHOICES, default='')
    Role = models.CharField(max_length=50, choices=ROLE_CHOICES, default='')
    privilege = models.CharField(max_length=10, default='User')
    is_admin = models.BooleanField()
    is_staff = models.BooleanField()
    is_active = models.BooleanField()

forms.py

class CustomUserCreationForm(UserCreationForm):
    ID = forms.CharField(max_length=50)
    password = forms.CharField(widget=forms.PasswordInput)

    class Meta(UserCreationForm):
        model = employee
        fields = "__all__"

class CustomUserChangeForm(UserChangeForm):
    ID = forms.CharField(max_length=50)
    password = forms.CharField(widget=forms.PasswordInput)

    class Meta:
        model = employee
        fields = "__all__"

admin.py

class UserAdmin(BaseUserAdmin):
    # The forms to add and change user instances
    add_form = CustomUserChangeForm
    form = CustomUserChangeForm
    model = employee

    list_display = ('ID', 'name', 'email', 'privilege')
    list_filter = ('is_admin',)
    fieldsets = (
        (None, {'fields': ('ID', 'password')}),
        ('Personal info', {'fields': ('name', 'email', 'privilege')}),
        ('Permissions', {'fields': ('is_admin', 'is_staff', 'is_active')}),
    )

    add_fieldsets = (
        (None, {
            'classes': ('wide',),
            'fields': ('ID', 'name', 'email')}
        ),
    )
    search_fields = ('name',)
    ordering = ('ID',)
    filter_horizontal = ()

views.py

def employee_list(request):
    data_list = Employee.objects.order_by('NIP')
    registered = False

    if request.method == "POST":
        if 'add_user' in request.POST:
            user_form = CustomUserCreationForm(request.POST, prefix='add')

            if user_form.is_valid():
                new_user = user_form.save()
                new_user.set_password(user.password)
                new_user.save()
                registered = True

            return HttpResponseRedirect(reverse('employee'))

    else:
        user_form = CustomUserCreationForm(prefix='add')

    return render(request, 'employee_list.html',
                            {'SignUpForm': user_form,
                            'registered': registered,
                            'Emp_list':data_list})

Answer 1

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

class MyUserManager(BaseUserManager):
use_in_migrations = True

def create_user(self, name, email, privilege='member', is_admin=False, is_staff=False, password=None):

    if not email:
        raise ValueError('Users must have an email address')

    user = self.model(
        name=name,
        email=self.normalize_email(email),
        privilege=privilege,
        )

    user.set_password(password)
    user.is_active = True
    user.is_staff = is_staff
    user.is_admin = is_admin
    user.save(using=self._db)
    return user

def create_superuser(self, name, email, password):
    new_user = self.create_user(
        name,
        email,
        privilege='admin',
        password=password,
        is_staff=True,
        is_admin=True
    )
    return new_user


class Employee(AbstractBaseUser):
    PRIVILEGE_LEVEL = (
        ('Administrator', 'admin'),
        ('Staff', 'staff'),
        ('Member', 'member'),
    )

    # ID = models.IntegerField(verbose_name="ID", unique=True)
    id = models.AutoField(primary_key=True)
    name = models.CharField(verbose_name="Name", max_length=50, blank=True, unique=True)
    email = models.EmailField(verbose_name="email", max_length=225, default='this@mail.com')
    Department = models.CharField(max_length=50, choices=DEP_CHOICES, default='')
    Role = models.CharField(max_length=50, choices=ROLE_CHOICES, default='')
    # privilege = models.CharField(max_length=10, default='User')
    privilege = models.CharField(max_length=120, choices=PRIVILEGE_LEVEL, default='member')
    is_admin = models.BooleanField(default=False)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=False)

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

По умолчанию форма модели django вызывает MyUserManager.create_user .и на основе модификаций, которые я сделал выше, по умолчанию создается пользователь без разрешений.Если is_admin и is_staff неявно установлены на True

Создать суперпользователь из текущей формы, вы можете обновить MyUserManager.create_user примерно так,

def create_user(self, name, email, privilege='member', is_admin=False, is_staff=False, password=None):

    if not email:
        raise ValueError('Users must have an email address')

    user = self.model(
        name=name,
        email=self.normalize_email(email),
        privilege=privilege,
        )

    user.set_password(password)
    user.is_active = True
    user.is_staff = is_staff
    user.is_admin = is_admin
    if privilege == 'admin':
        user.is_superuser = True
    user.save(using=self._db)
    return user

Но я бы посоветовал вам обновить ваш CustomUserCreationForm вместо этого как таковое,

class CustomUserCreationForm(UserCreationForm):
# ID = forms.CharField(max_length=50)
password = forms.CharField(widget=forms.PasswordInput)

class Meta(UserCreationForm):
    model = Employee
    fields = "__all__"

# ADD THIS
def clean(self, value):
    print(self.data)
    print(self.errors)
    cleaned_data = self.cleaned_data
    print(cleaned_data)
    return cleaned_data
# ADD THIS

def save(self, *args, **kwargs):
    data = self.cleaned_data
    # email, password1 = v['email'], v['password1']
    employee = None
    if data['privilege'] == 'admin':
        employee = Employee.objects.create_superuser(
            name=data['name'],
            email=data['email'],
            password=data['password'],
        )
    else:
        employee = Employee.objects.create_user(
            name=data['name'],
            email=data['email'],
            password=data['password'],
        )
    # User is already saved after calling BaseUserManager function.
    return employee

MyUserManager.create_superuser вызывает MyUserManager.create_user и обновляет его на основе переданного параметра

Дайте мне знать, если вам нужны дополнительные разъяснения.Приветствия.