配置Django UserManager为超级用户输入出生日期?

2024-04-26 13:23:16 发布

您现在位置:Python中文网/ 问答频道 /正文

通过创建我的自定义用户模型,一直在捣乱django1.5。在我的模型中,我有一个必填字段dob=DateTimeField()。我基本上保持了UserManager和AbstractUser不变。但是,当我尝试python时管理.pycreatesuperuser,我没有输入dob,我得到错误:

IntegrityError:“dob”列中的null值违反了not null约束

如何编辑UserManager以允许我在终端中输入出生日期,以及如何将输入配置为在数据库中存储为datetime字段?在

编辑我的答案:

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']
...

Tags: selfnonetruefieldsisemailcreateusername
1条回答
网友
1楼 · 发布于 2024-04-26 13:23:16

您需要重写继承自AbstractUser的模型中的REQUIRED_FIELDS。浏览contrib/auth/models.py可以注意到AbstractBaseUser有{},而{}有{}。我猜你应该有这样的东西:

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

编辑:不再是猜测。它起作用了。在

一般来说,您可以通过查看createsuperuser来了解createsuperuser的作用。任何非授权命令都可以位于core/management/commands/。几乎任何manage.py问题都可以通过查看命令的源代码来解决。在

相关问题 更多 >