How do I add a Foreign Key Field to a ModelForm in Django?

by

In regards to displaying a foreign key field in a form you can use the forms.ModelChoiceField and pass it a queryset.

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

from your question:

I know in a view you can use
document.code_set (for example) to
access the related objects for the
current document object, but I’m not
sure how to apply this to a ModelForm.

Actually, your Document objects wouldn’t have a .code_set since the FK relationship is defined in your documents model. It is defining a many to one relationship to Code, which means there can be many Document objects per Code object, not the other way around. Your Code objects would have a .document_set. What you can do from the document object is access which Code it is related to using document.code.

edit: I think this will do what you are looking for. (untested)

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

actually you probably want to use get_or_create when creating your Code object instead of this.

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

More Related Contents:

  1. Setting the selected value on a Django forms.ChoiceField
  2. Django: Set foreign key using integer?
  3. Django ModelForm: What is save(commit=False) used for?
  4. How can I restrict Django’s GenericForeignKey to a list of models?
  5. Dropdown in Django Model
  6. Django – Overriding the Model.create() method?
  7. Django unique_together with nullable ForeignKey
  8. How to add Check Constraints for Django Model fields?
  9. Extending the User model with custom fields in Django
  10. How to define two fields “unique” as couple
  11. Django – Circular model import issue
  12. How to filter empty or NULL names in a QuerySet?
  13. How to view corresponding SQL query of the Django ORM’s queryset?
  14. How to update an object from edit form in Django?
  15. In a django model custom save() method, how should you identify a new object?
  16. Unique fields that allow nulls in Django
  17. Tying in to Django Admin’s Model History
  18. Django migration error :you cannot alter to or from M2M fields, or add or remove through= on M2M fields
  19. Django: Validate file type of uploaded file
  20. Django data migration when changing a field to ManyToMany
  21. Django accessing ManyToMany fields from post_save signal
  22. How to use django models with foreign keys in different DBs?
  23. how to show datepicker calendar on datefield
  24. Django comparing model instances for equality
  25. Setting default value for Foreign Key attribute
  26. django most efficient way to count same field values in a query
  27. Sharing models between Django apps
  28. ModelForm with OneToOneField in Django
  29. Django: dynamic database file
  30. Get type of Django form widget from within template

Leave a Comment