Как изменить queryset для ModelChoiceField на форме?

Question

[Yura Khlyan]

Доброго времени суток.

У меня есть две модели(кроме остальных и упрощено):

class Range(AbstractRange):
    site = models.ManyToManyField(Site)

class ConditionalOffer(AbstractConditionalOffer):
    site = models.ForeignKey(Site, null=True)
    range = models.ForeignKey(Range, null=True)

Еще у меня есть две формы для ConditionalOffer, на каждой из которых есть поле ModelChoiceField. На первой форме выбирается поле Site, а на второй Range. Нужно сделать так, что бы на второй форме можна было выбрать только те Range, в которых поле Site имеет значение поля Site первой формы.

Вот примеры форм:

class MetaDataForm(OscarMetaDataForm):
    site = forms.ModelChoiceField( 
        queryset=Site.objects.all(),
        widget=SITE_WIDGET,
        initial=SITE_INITIAL)

    class Meta:
        model = ConditionalOffer
        fields = ('site', 'name', 'description')


class BenefitForm(OscarBenefitForm):
    range = forms.ModelChoiceField( 
        queryset=Range.objects.filter(site__id=settings.SITE_ID)

    class Meta:
        model = Benefit
        fields = ('range', 'type', 'value', 'max_affected_items')

Тоесть мне нужно вместо settings.SITE_ID в строке

queryset=Range.objects.filter(site__id=settings.SITE_ID)

поставить ID выбраного сайта из первой формы.

Как мне это сделать? Буду очень благодарен за помощь.

Answer 1

[Oscar Django]

Мне кажется проще всего при изменении поля site первой формы отправлять ajax запрос с выбранными site_id. Для BenefitForm сделать что то по типу:

class BenefitForm(OscarBenefitForm):
    range = forms.ModelChoiceField()
    def __init__(self, *args, **kwargs):
        super(BenefitForm, self).__init__(*args, **kwargs)
        if hasattr(kwargs, 'site_ids'):
            self.fields['range'].queryset = Range.objects.filter(site__id__in=kwargs.get('site_ids'))
        else:
            self.fields['range'].queryset = Range.objects.all()

В ответе получать отрендеренную форму и вставлять ее на страницу.

Либо изначально получать все Range, а уже с помощью js отфильтровывать.

Comments

[Yura Khlyan]

Немного сам поискав, немного вашей помщи, сделал вот так:

class BenefitForm(OscarBenefitForm):
    range = forms.ModelChoiceField( 
        queryset=Range.objects.all())

    def __init__(self, site=None, *args, **kwargs):
        super(OscarBenefitForm, self).__init__(*args, **kwargs)
        if site:
            self.fields['range'].queryset = Range.objects.filter(site__id=site.id)

class OfferWizardStepView(OscarOfferWizardStepView):
    ....
    def get_form_kwargs(self, *args, **kwargs):
        form_kwargs = {}
        if self.update:
            form_kwargs['instance'] = self.get_instance()
        if self.step_name != 'metadata':
            form_kwargs['site'] = self._fetch_session_offer().site
        session_kwargs = self._fetch_form_kwargs()
        form_kwargs.update(session_kwargs)
        parent_kwargs = super(OfferWizardStepView, self).get_form_kwargs(
            *args, **kwargs)
        form_kwargs.update(parent_kwargs)
        return form_kwargs