get_or_create用于两个相关的模型和表单集

时间:2011-05-25 17:47:42

标签: django forms get inline

我有两个与ForeignKey和内联formset相关的模型:

class Balanta(models.Model):
    data = models.DateField()

    class Meta:
        ordering=['data']
        verbose_name_plural="Balante"

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

class Conturi(models.Model):
    cont=models.PositiveIntegerField()
    cont_debit=models.DecimalField(default=0, max_digits=30, decimal_places=2)
    cont_credit=models.DecimalField(default=0, max_digits=30, decimal_places=2)
    balanta = models.ForeignKey(Balanta)

    class Meta:
        #oredering=['cont']
        verbose_name_plural="Conturi"

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

我有基于这两个模型的表单集。提交表单后,如果表单的“Balanta部分”存在,表单不应该做任何事情,它不存在它应该保存在数据库中(我设法做到了这一点)。

现在我想对“表单的Conturi部分”做的是查看cont是否在数据库中,表单应该使用输入框中的值更新它,如果它不在数据库中,表单应该创建它和相应的cont_debit或cont_credit。当前状态给我:**之后的get_or_create()参数必须是映射,而不是列表。

我认为这比解决问题的简单/新手更重要。

以下是观点:

* 最后更新 *

from django.http import HttpResponseRedirect
from django.shortcuts import get_object_or_404, render_to_response
from django.template import RequestContext
from sitfin.models import Balanta, Conturi
from sitfin.forms import BalantaForm , ConturiForm
from django.forms.models import inlineformset_factory

def balanta_introducere(request):
    balanta=Balanta()
    ConturiInlineFormSet=inlineformset_factory(Balanta, Conturi, extra=3)
    if  request.method=='POST':
        balanta_form=BalantaForm(request.POST, instance=balanta)
        if balanta_form.is_valid():
            balanta, created=Balanta.objects.get_or_create(**balanta_form.cleaned_data)
            #return HttpResponseRedirect('/sitfin/balantaok')
        formset=ConturiInlineFormSet(request.POST, request.FILES, instance=balanta)
        if formset.is_valid():
            #formset.save()
            for form in formset:
                data={
                        'cont':form.cleaned_data.get('cont'),
                        'cont_debit':form.cleaned_data.get('cont_debit'),
                        'cont_credit':form.cleaned_data.get('cont_credit'),
                        'balanta':form.cleaned_data.get('balanta'),
                }
                try:
                    c=Conturi()#.objects.get(cont=data['cont'])
                except Conturi.DoesNotExist:
                    cont_complete,created=Conturi.objects.get_or_create(**data)
                else:
                    for form in formset:
                        new_data={
                                'cont':form.cleaned_data.get('cont'),
                                'cont_debit':form.cleaned_data.get('cont_debit'),
                                'cont_credit':form.cleaned_data.get('cont_credit'),
                                'balanta':form.cleaned_data.get('balanta'),
                        }
                        cont_complete,created=Conturi.objects.get_or_create(**new_data)
    else:
        balanta_form=BalantaForm()
        formset=ConturiInlineFormSet(instance=balanta)
    return render_to_response('sitfin/balanta_introducere.html',{'balanta_form':balanta_form,'formset':formset}, context_instance=RequestContext(request))

1 个答案:

答案 0 :(得分:1)

formset.cleaned_data是各个表单cleaned_data的列表。您需要迭代formset并为每个创建一个Conturi对象:

if formset.is_valid():
    for form in formset:
        cont, created=Conturi.objects.get_or_create(**form.cleaned_data)

<强>更新

为了保护自己免受因模型属性不对应的其他表单字段而导致的错误,最好明确地将参数提供给get_or_create,而不是使用**form.cleaned_data

if formset.is_valid():
   for form in formset:
       data = {
           'cont': form.cleaned_data.get('cont'),
           'cont_debit': form.cleaned_data.get('cont_debit'),
           'cont_credit': form.cleaned_data.get('cont_credit'),
           'balanta': form.cleaned_data.get('balanta'),
       }
       cont, created = Conturi.objects.get_or_create(**data)

它基本上是相同的过程,但现在您确切地知道将哪些参数传递给get_or_create