Guardian和自定义用户模型:迁移时的IntegrityError,创建匿名用户

时间:2018-03-21 13:20:45

标签: mysql django django-guardian

我有这个模型文件,其中定义了自定义用户。

class CustomUser(AbstractBaseUser,PermissionsMixin, GuardianUserMixin):
    email = models.EmailField(max_length=50,
            verbose_name='email address', unique=True)
    first_name    = models.CharField('first name', max_length=15,blank=False)
    last_name     = models.CharField('last name', max_length=15,blank=True)
    date_joined   = models.DateTimeField('date joined', auto_now_add=True)
    slug          = models.SlugField('slug', max_length=50, unique=True, null=True) 
    is_active     = models.BooleanField('active',default=True)


    objects = UserManager()

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []


    class Meta:
    app_label = "latiro_app"
    db_table =  "users"


        permissions = (
                ("view_user", "view User"),
                ("edit_profile", "Edit Profile"),
                )


    def get_full_name(self):
        full_name = '{0} {1}'.format(self.first_name, self.last_name)
        return full_name.strip()

    def get_short_name(self):
        return self.first_name


    def generate_user_slug(self):
        max_length = CustomUser._meta.get_field('slug').max_length
        full_name = self.get_full_name()
        slug = original = slugify(full_name)[:max_length]

        for i in itertools.count(1):
            if not CustomUser.objects.filter(slug=slug).exists():
                break

            slug = '{0}-{1}' .format(original[:max_length - len(str(i)) - 1], i)
        return slug


    def save(self, *args, **kwargs):
        if not self.slug:
            self.slug = self.generate_user_slug()
        super().save()         


def get_anonymous_user_instance(CustomUser):
    return CustomUser(first_name='Anonymous')


class Profile (models.Model):
    user            = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete= models.CASCADE,
                                    null=False, verbose_name='list of users')
    phone_regex   = RegexValidator(regex=r'^\+?1?\d{9,15}')
    phone_number  = models.CharField('phone number', validators=[phone_regex], max_length=50, blank=True)
    country         = models.CharField(max_length=50,blank=True)
    province        = models.CharField(max_length=50)
    city            = models.CharField(max_length=50,blank=True)
    profile_picture = models.ImageField(upload_to='user_profile/%y/%m/%d/', blank=True)
    followers       = models.IntegerField(default=0, null=True)
    following       = models.IntegerField(default=0, null=True)

这个信号

def create_user_profile(sender, instance, created, **kwargs):

    if kwargs.get('created', True) and not kwargs.get('raw', False):
        profile = Profile(user=instance)
            profile.save()

post_save.connect(create_user_profile,
              sender = settings.AUTH_USER_MODEL)

首次运行时,Django会完成将我的应用程序迁移到数据库中。但是当我第二次运行迁移时,Django抛出了这个错误:

  

django.db.utils.IntegrityError:(1062,“重复条目”为密钥   '电子邮件'“)。

我正在使用MySQL。这里出了什么问题?

更新:我发现在进行第二次迁移时会抛出错误的代码。

我在我的模型文件中有这个代码:

def get_anonymous_user_instance(CustomUser):
    return CustomUser(first_name='Anonymous')

我将此功能导入到我的设置中,这样当我第一次进行迁移时,将创建匿名用户。和ofcource它按预期进行。但是当我再次运行迁移时,django会尝试创建另一个匿名用户,这就是我收到此错误的原因。

当我在我的设置中注释掉这一行时:

” GUARDIAN_GET_INIT_ANONYMOUS_USER ='latiro_app.models.get_anonymous_user_instance' “

我没有看到这个错误。

1 个答案:

答案 0 :(得分:1)

email是一个唯一字段,因此您不能有两个CustomUser具有相同的e-mail。 Guardian正在从GUARDIAN_GET_INIT_ANONYMOUS_USER设置中的callable返回的对象创建一个新用户,因此它会尝试创建同一个用户并失败。您可以使用get_or_create

解决此问题
def get_anonymous_guardian_user_instance(CustomUser):
    user, _ = CustomUser.objects.get_or_create(
        first_name='Anonymous',
        email='guardian@example.com',
    )
    return user

实际上有一个Guardian issue,但遗憾的是它没有得到解决,也不清楚它是否应该是一个问题。文档cautions the user about syncdb,但在我看来,在section about creating anonymous users(它不涉及unique约束)中,可调用者应该返回的内容并不十分清楚。