Django - 管理中的 UserProfile m2m 字段 - 错误

Question

我的模型：

class UserProfile(models.Model):
    TYPES_CHOICES = (
        (0, _(u'teacher')),
        (1, _(u'student')),
    )
    user = models.ForeignKey(User, unique=True)
    type = models.SmallIntegerField(default=0, choices=TYPES_CHOICES, db_index=True)
    cities = models.ManyToManyField(City)
class City(models.Model):
    name = models.CharField(max_length=50)
    slug = models.SlugField(max_length=50)

在 admin.py 中：

admin.site.unregister(User) 
class UserProfileInline(admin.StackedInline):
    model = UserProfile

class UserProfileAdmin(UserAdmin):
    inlines = [UserProfileInline]

admin.site.register(User, UserProfileAdmin)

@receiver(post_save, sender=User)
def create_profile(sender, instance, created, **kwargs):
    """Create a matching profile whenever a user object is created."""
    if created:
        profile, new = UserProfile.objects.get_or_create(user=instance)

但是当我添加新用户并选择一个城市时，我收到该错误： IntegrityError at /admin/auth/user/add/ （1062，“密钥‘user_id’的重复条目‘3’”）

我的代码有什么问题？如果我不选择任何城市 - 用户将被正确添加。以某种方式，用户被多次添加到 UserProfile 中。

Answer 1

我最近也遇到了同样的问题。当你仔细思考时，它实际上是非常有道理的。当您在管理中保存带有内联的表单时，它首先保存主模型，然后继续保存每个内联。当它保存模型时，您的 post_save 信号将被触发，并创建一个 UserProfile 来匹配，但现在是时候保存内联了。 UserProfile 内联被认为是新的，因为它以前不存在（没有 pk 值），因此它尝试另存为全新且不同的 UserProfile，并且您会因违反唯一约束而收到完整性错误。解决方案很简单。只需覆盖 UserProfile.save：

def save(self, *args, **kwargs):
    if not self.pk:
        try:
            p = UserProfile.objects.get(user=self.user)
            self.pk = p.pk
        except UserProfile.DoesNotExist:
            pass

    super(UserProfile, self).save(*args, **kwargs)

本质上，这只是检查相关用户是否存在现有的 UserProfile。如果是这样，它将将此 UserProfile 的 pk 设置为该 UserProfile 的 pk，以便 Django 执行更新而不是创建。