Django - 让createview保存两个不同的模型

时间:2015-10-25 18:25:38

标签: django django-models django-forms django-class-based-views

我试图让django自动创建第二个模型的实例,同时通过CreatView创建第一个:

models.py

class MAG_magazyn(models.Model): #1st
  nr_ident=models.CharField(max_length=50, unique=True)
  nazwa=models.CharField(max_length=50)
  typ=models.CharField(max_length=50, blank=True, null=True)
  podzespol=models.BooleanField(default=False)
  kategoria=models.ForeignKey(MAG_kategorie_czesci)

class MAG_magazyn_stan(models.Model): #2nd
  id_czesc=models.OneToOneField(MAG_magazyn)
  regal=models.CharField(max_length=3)
  polka=models.CharField(max_length=3)
  stan_min=models.IntegerField()
  stan=models.IntegerField()
  alert=models.IntegerField(default=0)
  rel_stan=models.DecimalField(...)

现在,随着这些模型的连接,我想添加一些带有一些默认值的2nd实例。我的想法是覆盖form_valid,获取第一个模型的最后一个id,然后创建并保存第二个实例:

  Class NowyMag (CreateView):
'''
Widok pozwalajacy na dodanie nowej czesci magazynowej do listy
'''

model=MAG_magazyn
fields=[
        (...),
        ]
template_name='magazyn/nowy_mag.html'
success_url=reverse_lazy('mag_lista_mag')

  def form_valid(self, form):
     form.save(  )
     nowy_stan=MAG_magazyn_stan(id_czesc=MAG_magazyn.objects.latest('id').id, regal=0,polka=0,stan_min=0,stan=0, alert=1)
     nowy_stan.save()
     return super(NowyMag, self).form_valid(form)

但我得到的只是:

Exception Type: ValueError
Exception Value: Cannot assign "20": "MAG_magazyn_stan.id_czesc" must be a "MAG_magazyn" instance.

我是否正确思考,但是有错误,或者有一些更清洁的方式来获得这样的工作?

一点点更好的解释: 在创建MAG_magazyn的实例时,我想自动创建MAG_magazyn_stan,其id为已创建的MAG_magazyn和一些默认值(此处全部等于0)

1 个答案:

答案 0 :(得分:2)

您的问题不是很明确,但我假设您使用的是与MAG_magazyn相关联的表单,在这种情况下,您希望执行以下操作:

def form_valid(self, form):
    # You need to get a new object based on the form
    self.object = form.save(commit=False)
    ... # in case you want to modify the object before commit
    self.object.save()
    # not you can use it to assign to the second model
    nowy_stan=MAG_magazyn_stan(id_czesc=self.object, ...)
    nowy_stan.save()

    return HttpResponseRedirect(self.get_absolute_url) # or elsewhere