web-dev-qa-db-ja.com

DjangoのModelFormに外部キーフィールドを追加するにはどうすればよいですか?

私がしたいことは、ユーザーに許可する単一のフォームを表示することです:

  • ドキュメントのタイトルを入力(Documentモデルから)
  • user_defined_codeドロップダウンリストからの選択肢(UserDefinedCodeモデルによって入力)
  • unique_codeCodeモデルに保存)

外部キー関係のフィールドをフォームに表示する方法がわかりません。 (たとえば)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)

私のモデルフォーム

class DocumentForm(ModelForm):
    class Meta:
        model = Document
23
Ben S

フォームでの外部キーフィールドの表示に関しては、 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との多対1の関係を定義しています。つまり、Documentオブジェクトごとに多くのCodeオブジェクトがあり、その逆はあり得ません。 Codeオブジェクトには.document_setが含まれます。ドキュメントオブジェクトから実行できることは、document.codeの使用に関連するCodeへのアクセスです。

edit:これはあなたが探していることを実行すると思います。 (未試験)

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))

実際には、コードオブジェクトを作成するときに、これの代わりにget_or_createを使用することをお勧めします。

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