基于先前表单的选择，在formtools向导中限制Django表单中ManyToManyField查询集的方法

Question

我正在使用来自django-formtools的SessionWizardView构建一个包含两个表单的向导。我面临的挑战是，我需要引用第一个表单中的输入以限制第二个表单上可用的查询集。

让事情更有趣的是，我正在使用crispy表单进行布局，而查询集需要通过相关项目上的方法进行限制。

这是我所处的情况（大大简化）的要点：

模型

class Product(models.Model):
    # pk, name, etc....
    catalogitem = ForeignKey("myapp.CatalogItem")
    colors = ManyToManyField("myapp.Colors")

class Colors(models.Model):
    # pk, name, etc....

class CatalogItem(models.Model):
    # Colors are stored within CatalogVariants, which I've left
    # as a blackbox in this example, since they are retrieved as
    # a queryset on this model with this method:

    # pk, name, etc....

    def get_colors(self):
        # Returns a queryset of color objects.

Views

ProductFormWizard(SessionWizardView):
    form_list = [
        productFormWizard_Step1,
        productFormWizard_Step2,
    ]

    def get_context_data(self, **kwargs):
       # ...
       pass

    def get_form_initial(self, step):
        initial = {}
        # ...
        return self.initial_dict.get(step, initial)

    def process_step(self, form):
        if self.steps.step1 == 1:
            pass
        return self.get_form_step_data(form)

    def done(self, form_list, **kwargs):
        return render(self.request, 'done.html', {
            'form_data': [form.cleaned_data for form in form_list],
        })

Forms

productFormWizard_Step1(forms.ModelForm):
    # Defines a form where the user selects a CatalogProduct.
    model = Product

productFormWizard_Step2(forms.ModelForm):
    """
    Defines a form where the user chooses colors based on 
    the CatalogProduct they selected in the previous step.
    """
    model = Product

根据通过谷歌搜索和一些SO问题的研究（没有直接相关的），我假设我需要在colors字段上设置.queryset属性，但我不确定在哪里设置。有两个想法：

• 我猜它以某种方式进入.get_form_initial()，但我不知道实现的最佳方法。
• 或者，适当的代码可能以某种方式进入productFormWizard.get_context_data()方法。

在.get_form_initial()中，我可以这样做：

if step == '1':
    itemID = self.storage.get_step_data('0').data.get('0-pfProduct', "")
    if itemID:
        obj = CatalogItem.objects.get(id=itemID)
        initial['colors'] = obj.get_get_colors()

然而，这只是选择可用的相关项目...它没有限制列表。

附加信息

Python == 3.5.3
Django == 1.10.6
django-crispy-forms == 1.6.1
django-formtools == 2.0

- Andrew Marconi

Answer 1

解决方案是在视图上重写.get_form()方法：

def get_form(self, step=None, data=None, files=None):

  form = super(bzProductFormWizard, self).get_form(step, data, files)

    if step == '1':
        past_data = self.get_cleaned_data_for_step('0')
        product = past_data['product']
        form.fields['colors'].queryset = ... #CUSTOM QUERYSET

    return form

- Andrew Marconi