diff --git a/extra_views/formsets.py b/extra_views/formsets.py index 08e19b3..1b925ac 100644 --- a/extra_views/formsets.py +++ b/extra_views/formsets.py @@ -129,14 +129,6 @@ class ModelFormSetMixin(FormSetMixin, MultipleObjectMixin): exclude = None fields = None - def get_formset_kwargs(self): - """ - Returns the keyword arguments for instantiating the formset. - """ - kwargs = super().get_formset_kwargs() - kwargs["queryset"] = self.get_queryset() - return kwargs - def get_factory_kwargs(self): """ Returns the keyword arguments for calling the formset factory @@ -155,6 +147,14 @@ def get_formset(self): """ return modelformset_factory(self.model, **self.get_factory_kwargs()) + def get_context_data(self, *args, **kwargs): + context = super().get_context_data(*args, **kwargs) + formset_class = self.get_formset() + formset_kwargs = self.get_formset_kwargs() + formset_kwargs["queryset"] = context['object_list'] + context['formset'] = formset_class(**formset_kwargs) + return context + def formset_valid(self, formset): """ If the formset is valid, save the associated models.