如何配置Django的UserManager以输入超用户的出生日期？

Question

我在玩 Django 1.5，尝试创建自己的用户模型。在我的模型里，有一个必填字段 dob = DateTimeField()。我基本上把 UserManager 和 AbstractUser 保持得差不多。不过，当我运行 python manage.py createsuperuser 时，我没有输入 dob，就出现了这个错误：

IntegrityError: null value in column "dob" violates not-null constraint

我该怎么修改 UserManager，让我可以在终端输入出生日期，并且怎么配置输入，让它能存储为数据库里的 DateTimeField 呢？

编辑我的回答：

class UserManager(BaseUserManager):

    def create_user(self, username, email=None, password=None, dob=None, **extra_fields):
        """
        Creates and saves a User with the given username, email and password.
        """
        now = timezone.now()
        if not username:
            raise ValueError('The given username must be set')
        if not email:
            raise ValueError('Email must be given')
        email = UserManager.normalize_email(email)
        user = self.model(username=username, email=email,
                          is_staff=False, is_active=True, is_superuser=False,
                          last_login=now, date_joined=now, dob=dob, **extra_fields)

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

    def create_superuser(self, username, email, password, dob, **extra_fields):
        u = self.create_user(username, email, password, dob, **extra_fields)
        u.is_staff = True
        u.is_active = True
        u.is_superuser = True
        u.save(using=self._db)
        return u


# ... in the user model:
...
    REQUIRED_FIELDS = ['email', 'dob']
...

Answer 1

你需要在从 AbstractUser 继承的模型中重写 REQUIRED_FIELDS。在浏览 contrib/auth/models.py 文件时，可以看到 AbstractBaseUser 里有 REQUIRED_FIELDS = []，而 AbstractUser 里有 REQUIRED_FIELDS = ['email']。我猜你应该写成这样：

class MyCoolUserModel(AbstractUser):
    REQUIRED_FIELDS = ['email', 'dob']
    ...

编辑：不再是猜测了，这确实有效。

一般来说，你可以通过查看 createsuperuser 的实现，来了解它是怎么工作的，具体可以看 contrib/auth/management/commands/createsuperuser.py。任何非认证的命令都可以在 core/management/commands/ 找到。基本上，任何 manage.py 相关的问题都可以通过查看命令的源代码来解决。