0

カスタムユーザーモデルを作成して、django 1.5 をいじっています。私のモデルには、必須フィールド dob = DateTimeField() があります。私は UserManager と AbstractUser をほぼ同じに保ちました。ただし、python manage.py createsuperuser を実行しようとすると、dob を入力せず、エラーが発生します。

IntegrityError: 列 "dob" の null 値が非 null 制約に違反しています

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

1 に答える 1

1

REQUIRED_FIELDSから継承するモデルで上書きする必要がありますAbstractUser。1 つを参照すると、 hasとhasのcontrib/auth/models.py方法に気付くことができます。私の推測では、次のようなものが必要です。AbstractBaseUserREQUIRED_FIELDS = []AbstractUserREQUIRED_FIELDS = ['email']

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

編集：もう推測ではありません。できます。

一般的に言えば、 を見れば何をするかを見ることができcreatesuperuserますcontrib/auth/management/commands/createsuperuser.py。非認証コマンドはすべて にありcore/management/commands/ます。コマンドのソースを見ることで、ほとんどすべてのmanage.py問題を解決できます。

于 2013-03-08T15:09:50.687 に答える