如何在 django 中实现自定义用户管理器？答案

【问题标题】：How can I implement custom user manager in django?如何在 django 中实现自定义用户管理器？
【发布时间】：2016-12-29 23:31:13
【问题描述】：

我在Django 中创建了继承自AbstractUser 的自定义User 模型：

class ChachaUser(AbstractUser):
    birth = models.DateField()
    gender = models.CharField(max_length=1, choices=GENDER_CHOICES)

还有我的CustomUserCreationForm：

GENDER_CHOICES = (
    ('M', '남'),
    ('F', '여'),
)


class MyUserCreationForm(UserCreationForm):

    birth = forms.DateField(
        widget=forms.SelectDateWidget(
            years=range(1970, 2015)
        ),
        required=True,
    )
    gender = forms.ChoiceField(choices=GENDER_CHOICES, initial='M')

    class Meta(UserCreationForm.Meta):
        model = ChachaUser
        fields = UserCreationForm.Meta.fields + ('birth', 'gender')

但是我想使用python manage.py createsuperuser创建一个超级用户，我也必须实现CustomUserManager。

有什么想法或例子吗？

【问题讨论】：

首先在任何地方导入ChachaUser 使用get_user_model()。如果您需要将字段gender, birth 添加到createsuperuser 命令使用REQUIRED_FIELDS = ['email', 'birth', 'gender']
@AnnaVracheva：效果很好！谢谢:)

标签： django django-users

【解决方案1】：

您应该实现一个继承自BaseUserManager 的类。您必须实现方法.create_user 和.create_superuser。我强烈建议您将此类放在与您的自定义 ChachaUser 模型相同的应用程序中的文件 managers.py 中。

from django.contrib.auth.base_user import BaseUserManager


class UserManager(BaseUserManager):

    def create_user(self, email, password, **extra_fields):
        if not email:
            raise ValueError('Email for user must be set.')
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save()
        return user

    def create_superuser(self, email, password, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)

        if extra_fields.get('is_staff') is not True:
            raise ValueError('Superuser must have is_staff=True.')
        if extra_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must have is_superuser=True.')
        return self.create_user(email, password, **extra_fields)

有关使用AbstractUser 或AbstractBaseUser 创建自定义用户模型的完整指南，请参阅this tutorial。

【讨论】：

【解决方案2】：

或者这样：

class CustomUser(User):
    objects = MyManager()
    class Meta(object):
        proxy=True

【讨论】：

【解决方案3】：

您对UserManager (code) 感兴趣。

例子：

from django.contrib.auth.models import AbstractUser, UserManager

class CustomUserManager(UserManager):
    # your methods

class CustomUser(AbstractUser):
    # fields

    objects = CustomUserManager()

【讨论】：