Django - 将工作人员设置为新用户的默认值

Question

大家！

我正在尝试将新用户设置为默认的 staff_member，但我找不到任何解决方案。我真的需要帮助。我的代码如下。

型号

class Participante(models.Model):
    nome = models.CharField(max_length=100)
    cpf = models.CharField(max_length=13)
    email = models.EmailField()
    dt_criacao = models.DateTimeField(auto_now=True)
    is_staff = models.BooleanField(
        ('staff status'),
        default=True,
        help_text=('Designates whether the user can log into this admin site.'),
    )

    def __str__(self):
        return self.nome

表格

class ParticipanteForm(UserCreationForm):

    first_name = forms.CharField(max_length=100, label='Primeiro nome')
    last_name = forms.CharField(max_length=100, label='Último nome')
    class Meta:
        model = User
        fields = ['username', 'first_name', 'last_name', 'email', 'password1', 'password2'] 

观看次数

def cadastro(request):
    form = ParticipanteForm()

    if request.method == 'POST':
        form = ParticipanteForm(request.POST)
        
        if form.is_valid():
            #User(request, username=username, password=password1)
            form.save()
            return redirect('dashboard')

    return render(request, 'cadastro.html', locals())

Answer 1

我认为这里的问题是您的 UserCreationForm 指向的是 User 模型，而不是您的自定义 Participante 模型。因此，用户不会被保存在您期望的表中。

在settings.py 中，将Participante 模型设置为您的用户模型（您的Participante 模型还必须继承AbstractUser 以保留User 模型的方法等。

阅读：https://docs.djangoproject.com/en/3.1/topics/auth/customizing/#django.contrib.auth.models.AbstractUser

# settings.py
AUTH_USER_MODEL = 'your_app.models.Participante'

# your_app.models
from django.contrib.auth.models import AbstractUser

class Participante(AbstractUser):
    nome = models.CharField(max_length=100)
    cpf = models.CharField(max_length=13)
    email = models.EmailField()
    dt_criacao = models.DateTimeField(auto_now=True)
    is_staff = models.BooleanField(
        ('staff status'),
        default=True,
        help_text=('Designates whether the user can log into this admin site.'),
    )

    def __str__(self):
        return self.nome

然后在您的表单中，使用get_user_model() 指向您的AUTH_USER_MODEL

# forms.py
from django.contrib.auth import get_user_model

class ParticipanteForm(UserCreationForm):

    first_name = forms.CharField(max_length=100, label='Primeiro nome')
    last_name = forms.CharField(max_length=100, label='Último nome')
    class Meta:
        model = get_user_model()
        fields = ['username', 'first_name', 'last_name', 'email', 'password1', 'password2']```

Answer 2

你设置.instance的.is_staff属性包裹在form中，所以：

def cadastro(request):
    if request.method == 'POST':
        form = ParticipanteForm(request.POST)
        if form.is_valid():
            form.instance.is_staff = True
            form.save()
            return redirect('dashboard')
    else:
        form = ParticipanteForm()
    return render(request, 'cadastro.html', locals())

您的ParticipanteForm 也适用于User 模型，而不是Participante 模型。如果您将Participante 设置为用户模型，则可以使用get_user_model() function [Django-doc]：

from django.contrib.auth import get_user_model

class ParticipanteForm(UserCreationForm):
    first_name = forms.CharField(max_length=100, label='Primeiro nome')
    last_name = forms.CharField(max_length=100, label='Último nome')
    
    class Meta:
        model = get_user_model()
        fields = ['username', 'first_name', 'last_name', 'email', 'password1', 'password2']

然而，要使Participante 成为用户模型，还需要一些额外的工作。 Django 在substituting a custom user model [Django-doc] 上有一个话题。