我正在使用来自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