基于get_initial()的CBV CreateView限制ForeignKey

Question

我有两个应用程序,family和person.在Person课堂上,我想限制已经创建的对象的可用mother和father选择family.如果我能利用的family_pk设置我get_initial()的person/views.py:PersonCreateview,并用它family_pk来限制的ForeignKey的选择mother和father那似乎是它可以成为一个伟大的方式来组合这些逻辑.这可能吗？

family/models.py

class Family(models.Model):
    lastname=models.CharField(max_length=100)

    def __str__(self):
        return self.lastname

family.views.py

from django.views import generic
from django.core.urlresolvers import reverse

from .models import Family

class FamilyListView(generic.ListView):
    context_object_name='family_list'
    model=Family


class FamilyDetailView(generic.DetailView):
    context_object_name='family'
    model=Family
    pk_url_kwarg='family_pk'


class FamilyCreateView(generic.CreateView):
    model=Family

    def get_success_url(self,**kwargs):
        return reverse('family-detail',kwargs={'family_pk':self.object.pk})

person/models.py

class Person(models.Model):
    fields=('family','firstname','mother','father')
    family=models.ForeignKey(Family)
    firstname=models.CharField(max_length=100)
    mother=models.ForeignKey('self',blank=True,null=True)
    father=models.ForeignKey('self',blank=True,null=True)

    def __str__(self):
        return self.firstname

person/views.py

from django.views import generic
from django.core.urlresolvers import reverse
from django.shortcuts import get_object_or_404

from .models import Person
from family.models import Family

class PersonDetailView(generic.DetailView):
    model=Person
    pk_url_kwarg='person_pk'

class PersonCreateView(generic.CreateView):
    model=Person

    def get_initial(self):
        family=get_object_or_404(Family,pk=self.kwargs.get('family_pk'))
        return {'family':family,}

    def get_success_url(self,**kwargs):
        return reverse('family:family-detail',args=[self.kwargs['family_pk']])

Answer 1

这可能有点旧,但我通过这个答案找到了解决方案:https://stackoverflow.com/a/291968/1971587

我最终使其工作的方式实际上是使用get_form().所以在你的情况下,我相信这样的东西会起作用:

class PersonCreateView(generic.CreateView):
    model = Person

    def get_family(self):
        return get_object_or_404(Family, pk=self.kwargs.get('family_pk'))

    def get_form(self, form_class):
        form = super(generic.CreateView, self).get_form(form_class)
        family = self.get_family()
        form.fields['mother'].queryset = Person.objects.filter(family)
        form.fields['father'].queryset = Person.objects.filter(family)
        return form

    def get_initial(self):
        initial = super(generic.CreateView, self).get_initial()
        initial['family'] = self.get_family()
        return initial

    def get_success_url(self, **kwargs):
        return reverse('family:family-detail', args=[self.kwargs['family_pk']])