是否需要为django中的自定义用户模型制作自定义管理器

时间:2015-10-09 08:29:15

标签: django django-models django-managers django-custom-user django-custom-manager

我正在使用AbstractBaseUserPermissionsMixin制作自定义用户模型,方法是遵循这两个教程(tutorial-1tutorial-2)。

到目前为止这个模型:

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField('email address', unique=True, db_index=True)
    username = models.CharField('username', unique=True, db_index=True)
    joined = models.DateField(auto_now_add=True)
    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)
    is_staff = models.BooleanField(default=False)

    USERNAME_FIELD = 'email'

    def __unicode__(self):
        return self.email

现在我感到困惑的是,在tutorial-1中,作者没有为自定义用户模型制作任何自定义管理器。相反,他使用表单来创建用户。

class RegistrationForm(forms.ModelForm):
    email = forms.EmailField(label = 'Email')
    password1 = forms.CharField(widget = forms.PasswordInput(), label = "Password")
    password2 = forms.CharField(widget = forms.PasswordInput(), label = 'Retype password')

    class Meta:
        model = User
        fields = ['email', 'username', 'password1', 'password2']

    def clean(self):
        """
        Verify that the values entered into the password fields match
        """
        cleaned_data = super(RegistrationForm, self).clean()
        if 'password1' in self.cleaned_data and 'password2' in self.cleaned_data:
            if self.cleaned_data['password1'] != self.cleaned_data['password2']:
                raise ValidationError("Password don't match.")
        return self.cleaned_data

    def save(self, commit=True):
        user = super(RegistrationForm, self).save(commit=False)
        user.set_password(self.cleaned_data['password1'])
        if commit:
            user.save()
        return user

但在tutorial-2中,其作者为自定义用户模型制作了自定义管理器。

class UserManager(BaseUserManager):

    def create_user(self, email, password, **kwargs):
        user = self.model(
            email=self.normalize_email(email),
            is_active=True,
            **kwargs
        )
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_superuser(self, email, password, **kwargs):
        user = self.model(
            email=email,
            is_staff=True,
            is_superuser=True,
            is_active=True,
            **kwargs
        )
        user.set_password(password)
        user.save(using=self._db)
        return user

使用Django Docs引用,有一个自定义用户模型的示例,它使用自定义管理器。我的问题是,是否可以不创建任何其他自定义管理器,如果没有创建自定义管理器的用途是什么?

1 个答案:

答案 0 :(得分:1)

我认为这是the docs的相关部分:

  

您还应该为User模型定义自定义管理器。如果您的用户模型定义usernameemailis_staffis_activeis_superuserlast_logindate_joined字段相同作为Django的默认User,你可以安装Django的UserManager;但是,如果您的用户模型定义了不同的字段,则需要定义一个扩展BaseUserManager的自定义管理器,提供[{1}}和create_user()方法。

文档中的示例需要定义自定义管理器,以便它可以设置create_superuser()字段。

教程中的示例似乎需要自定义管理器,因为它使用date_of_birth作为唯一标识符,并且没有单独的用户名字段。