自定义 Django 用户模型时使用新的管理表单

Question

我正在扩展用户模型，但似乎在使用新的管理表单时遇到问题。 我在 models.py 中有以下代码：

    class Preference(models.Model):
        choice = models.TextField(choices = (('grid', 'grid'), ('list','list'))) 

        def __unicode__(self):
            return self.choice

    class UserProfile2(models.Model):
        preference = models.ForeignKey(Preference, default = Preference.objects.get(id=2).id)
        user = models.OneToOneField(User, unique=True)

    def create_user_profile(sender, instance, created, **kwargs):
        if created:
            print 'creating user profile2'
            u = UserProfile2.objects.create(user=instance)   

    post_save.connect(create_user_profile, sender=User)

以下代码在 admin.py 中：

    class UserProfileInline(admin.TabularInline):
        model = UserProfile2
        fk_name = 'user'

    class CustomUserAdmin(UserAdmin):
        inlines = [UserProfileInline,]

    admin.site.unregister(User)
    admin.site.register(User, CustomUserAdmin)     

以下代码在 settings.py 中：

    AUTH_PROFILE_MODULE = 'userextension.UserProfile2' 

当用户不尝试控制管理中首选项对象的值并创建一个新用户使用默认值。但是，如果用户尝试放弃“列表”的默认值并改用“网格”，我会收到“密钥 user_id 的重复条目”错误。

我是否需要显式从管理表单中获取额外字段的值并保存 userprofile2 对象和用户对象？如果是这样，这与我收到的错误有什么关系？我还没有找到太多关于如何做到这一点的文档，并且非常感谢任何指导。

更新： 这似乎也很重要：当我删除 UserProfile2 模型中首选项的默认值时，我得到的错误是“列'preference_id'不能为空”

感谢您查看我的问题。

Answer 1

您的信号根据“created”的值在 User 对象的 post_save 上创建一个新的 UserProfile2 对象。我敢打赌，每次调用此信号时，“created”都是 True，并且每次您在 admin 中保存有关 User 对象的任何信息时，它都会尝试创建一个新的 UserProfile2 对象。

UserProfile2 的用户外键设置为唯一，这就是您收到重复键错误的地方。

尝试将信号代码更改为：

def create_user_profile(sender, instance, created, **kwargs):
        if created:
            u = UserProfile2.objects.get_or_create(user=instance)

应该防止重复键错误。希望对您有所帮助。