扩展Django用户模型 - 表单填充错误

时间:2017-05-15 18:10:34

标签: python django django-forms modelform

我正在使用Player类扩展Django的(v1.9)内置User模型,以添加一些额外的属性。

class Player(models.Model):
    TIMEZONES=()
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    ... (player-specific properties here)
    time_zone = models.CharField(max_length=255, choices=PRETTY_TIMEZONE_CHOICES, blank=True, null=True,)

从Django管理面板创建用户时,我并不总是需要创建播放器,因此有时只会创建用户。因此,播放器和用户ID不完全匹配。事实证明,当填充链接到Player的模型的ModelForms时,这会导致问题,如下所示:

class City(models.Model):
    player = models.ForeignKey(Player, on_delete=models.CASCADE)
    name = models.CharField(max_length = 100)
    x_coord = models.SmallIntegerField()
    y_coord = models.SmallIntegerField()
    region = models.CharField(max_length = 100)
    def __unicode__(self):
        return str(self.player) + "-" + str(self.name)
    class Meta:
        db_table = 'cities'

class CityForm(ModelForm):
    class Meta:
        model = City
        fields = (
            'name',
            'player',
            'x_coord',
            'y_coord',
            'region')

创建新城市时使用此modelForm。当用户ID和玩家ID匹配时,没有问题,玩家ID将填入表单并成功创建城市。当用户ID和玩家ID不同时,表单中不会填充玩家ID,表单无法验证,城市创建失败。

我从request.user获取玩家ID没有问题,我可以在获取POST数据后验证之前修复玩家ID。我还添加了一个后保存挂钩,以便始终创建播放器,因此ID将始终匹配。但似乎表单首先填充玩家ID,因为用户数据是可访问的,并且它是一对一的关系。

我在这里缺少什么?

1 个答案:

答案 0 :(得分:0)

您缺少的是当您实例化ModelForm以创建与某个现有对象相关的新行时,Django无法知道相关对象的ID。你需要以某种方式告诉它。

这样做的一种方法是,当您显示表单以响应GET时,请使用表单构造函数的initial参数:

myform = MyModelFormClass(None, initial={ 'myfkfield': myrelatedobject.pk })

现在,表单类知道在呈现表单时要预填的值,当表单发布时,该字段将随之发布。

另一种方法是从表单中省略关系字段,然后在保存之前填写它,方法是使用表单保存方法的commit参数:

myform = MyModelFormClass(request.POST)
# this causes form values to be filled into the instance without actually 
# writing to the database yet.
myinstance = myform.save(commit=False)
myinstance.myfkfield = myrelatedobject
# now really write to database
myinstance.save()

请注意,这将是一个插入。对于更新,您需要将现有对象提供给modelform构造函数,如下所示:

myinstance = MyModel.objects.get(pk=self.kwargs.pk)
myform = MyModelFormClass(request.POST, instance=myinstance)

没有实例,ModelForm不知道它在数据库中更新了哪一行。您会认为这一切都存在于HTML中,所以它不应该是必要的......但这不是Django的工作方式。您需要获取数据库中存在的对象,并将其与request.POST数据一起传递给ModelForm构造函数。然后,当您调用myform.save()时,它将验证表单,将其数据与现有对象合并,然后保存对象。使用commit=False会导致延迟执行这三个步骤中的最后一个步骤,以便您可以在更新的实例实际保存之前对其进行任何调整或检查。