如何在Django中向ModelForm添加外键字段?

时间:2011-04-18 20:42:20

标签: django django-models django-forms foreign-keys

我想要做的是显示一个允许用户使用的表单:

  • 输入文档标题(来自Document型号)
  • 从下拉列表中选择其中一个user_defined_code选项(由UserDefinedCode模型填充)
  • 输入unique_code(存储在Code模型中)

我不确定如何在表单中显示外键关系的字段。我知道在视图中你可以使用document.code_set(例如)来访问当前document对象的相关对象,但我不确定如何将它应用于ModelForm。

我的模特:

class UserDefinedCode(models.Model):
    name = models.CharField(max_length=8)
    owner = models.ForeignKey(User)

class Code(models.Model):
    user_defined_code = models.ForeignKey(UserDefinedCode)
    unique_code = models.CharField(max_length=15)

class Document(models.Model):
    title = models.CharField(blank=True, null=True, max_length=200)
    code = models.ForeignKey(Code)
    active = models.BooleanField(default=True)

我的ModelForm

class DocumentForm(ModelForm):
    class Meta:
        model = Document

1 个答案:

答案 0 :(得分:30)

关于在表单中显示外键字段,您可以使用forms.ModelChoiceField并将其传递给查询集。

所以,forms.py:

class DocumentForm(forms.ModelForm):
    class Meta:
        model = Document

    def __init__(self, *args, **kwargs):
        user = kwargs.pop('user','')
        super(DocumentForm, self).__init__(*args, **kwargs)
        self.fields['user_defined_code']=forms.ModelChoiceField(queryset=UserDefinedCode.objects.filter(owner=user))

views.py:

def someview(request):
    if request.method=='post':
        form=DocumentForm(request.POST, user=request.user)
        if form.is_valid():
            selected_user_defined_code = form.cleaned_data.get('user_defined_code')
            #do stuff here
    else:
        form=DocumentForm(user=request.user)

    context = { 'form':form, }

    return render_to_response('sometemplate.html', context, 
        context_instance=RequestContext(request))

来自你的问题:

  

我知道你可以使用的视图   document.code_set(例如)到   访问相关对象   当前的文档对象,但我不是   确定如何将其应用于ModelForm。

实际上,由于FK关系是在文档模型中定义的,因此Document个对象不会有.code_set。它定义了与Code的多对一关系,这意味着每个Document对象可以有很多Code个对象,而不是相反。您的Code个对象会有.document_set。您可以从文档对象中执行的操作是Codedocument.code相关的访问。

编辑:我认为这可以满足您的需求。 (另)

forms.py:

class DocumentForm(forms.ModelForm):
    class Meta:
        model = Document
        exclude = ('code',)

    def __init__(self, *args, **kwargs):
        user = kwargs.pop('user','')
        super(DocumentForm, self).__init__(*args, **kwargs)
        self.fields['user_defined_code']=forms.ModelChoiceField(queryset=UserDefinedCode.objects.filter(owner=user))
        self.fields['unique_code']=forms.CharField(max_length=15)

views.py:

def someview(request):
    if request.method=='post':
        form=DocumentForm(request.POST, user=request.user)
        if form.is_valid():
            uniquecode = form.cleaned_data.get('unique_code')
            user_defined_code = form.cleaned_data.get('user_defined_code')
            doc_code = Code(user_defined_code=user_defined_code, code=uniquecode)
            doc_code.save()
            doc = form.save(commit=False)
            doc.code = doc_code
            doc.save()
            return HttpResponse('success')
    else:
        form=DocumentForm(user=request.user)

    context = { 'form':form, }

    return render_to_response('sometemplate.html', context, 
        context_instance=RequestContext(request))

实际上你可能想在创建Code对象时使用get_or_create而不是这个。

doc_code = Code(user_defined_code=user_defined_code, code=uniquecode)