Question

我在Django 1.11中有一个应用程序，在那里我有一个“帐户”应用程序来管理用户。用户可以分配角色，使用管理器创建usera时会遇到问题。

添加用户名时出现错误：

IntegrityError at /accounts/add_user/

UNIQUE constraint failed: accounts_role.id

这是我的模特：

class User(AbstractBaseUser, PermissionsMixin):
    email = models.EmailField(_('Email address'), unique=True)
    first_name = models.CharField(_('First name'), max_length=60, blank=True)
    last_name = models.CharField(_('Last name'), max_length=60, blank=True)
    roles = models.ManyToManyField(Role, unique=False)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []
    objects = UserManager()

    class Meta:
        verbose_name = _('user')
        verbose_name_plural = _('users')

下面是角色类：

class Role(models.Model):
    ADMIN = 1
    PLAYER = 2
    ROLE_CHOICES = (
        (ADMIN, 'admin'),
        (PLAYER, 'player'),
    )

    id = models.PositiveSmallIntegerField(choices=ROLE_CHOICES, default=2, primary_key=True, unique=False)

    def __str__(self):
        return self.get_id_display()

这里是经理：

class UserManager(BaseUserManager):
    def create_user(self, email, password):
        user = self.model(email=email, password=password)
        user.set_password(password)
        user.is_staff = False
        user.is_superuser = False
        user.save(using=self._db)
        role = Role.objects.create(id=2)
        user.roles.add(role)
        return user

Answer 1

在create_user中，您应该为新用户分配一个或多个现有角色，而不是每次尝试创建相同的新角色（主键始终相同，这显然违反了唯一性约束）。

您的Role模型仅定义两个角色。与用户创建过程分开创建它们，然后分配它们：

def create_user(self, email, password):
    # (...)   
    role = Role.objects.get(id=2)
    user.roles.add(role)

Django项目中的唯一约束失败

1 个答案: