python - Django 模型中的自定义保存方法和登录问题

Question

我为我的模型编写了一个自定义保存方法:

class UserProfile(User):
      sex = models.SmallIntegerField(verbose_name=_(u'Sex'), choices=SEX_TYPES, default=1)
      .
      .


     def save(self, **kwargs):
         if self.pk is not None:
             UserSettings.objects.create(
                 user_id = self.pk
             )
         super(UserProfile, self).save(**kwargs)

和:

class UserSettings(models.Model):
    user = models.ForeignKey('userprofile.UserProfile', verbose_name=_(u"User"), related_name='own_user', null=True, blank=True)

    def __unicode__(self):
        return unicode(self.id)

此自定义方法创建一个 UserSettings 对象(在寄存器中)。但是在login之后，另一个 UserSettings 创建。我怎样才能避免这个问题？

Answer 1

使用 get_or_create :

def save(self, **kwargs):
    settings, created = UserSettings.objects.get_or_create(
        user_id=self.pk
    )
    super(UserProfile, self).save(**kwargs)