5

Mam dwie aplikacje, family i person. W klasie Person chciałbym ograniczyć dostępne opcje wyboru mother i father do obiektów już utworzonych z tego samego family. Jeśli mógłbym skorzystać z family_pk ustawiłem w get_initial() w person/views.py:PersonCreateview i używam tego family_pk do ograniczenia wyborów ForeignKey z mother i father, co wydaje się być świetnym sposobem na grupowanie logiki. czy to możliwe?CBV CreateView limit ForeignKey na podstawie get_initial()

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

Odpowiedz

10

To może nieco stary, ale znalazłem rozwiązanie za pośrednictwem tej odpowiedzi: https://stackoverflow.com/a/291968/1971587

Sposób skończyło się co praca była faktycznie z get_form(). W twoim przypadku wierzę, że coś takiego zadziała:

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']]) 
+1

Nie wierzę, że get_initial() będzie nadal potrzebny tutaj, ale może być konieczne w twoim przypadku dla innych elementów. –