Preventive file: work on inlines - 2

1 files changed, 6 insertions, 3 deletions

diff --git a/archaeological_files/views.py b/archaeological_files/views.py
index 90d41c68b..805921016 100644
--- a/archaeological_files/views.py
+++ b/archaeological_files/views.py
@@ -389,14 +389,11 @@ class MixFormFormsetUpdateView(UpdateView):
 
     def form_valid(self, form, inline_forms):
         self.object = form.save()
-        print("OK")
         for inline in inline_forms:
             inline.save()
-            pass
         return HttpResponseRedirect(self.get_success_url())
 
     def form_invalid(self, form, inline_forms):
-        print("NOP")
         return self.render_to_response(
             self.get_context_data(form=form, inlines=inline_forms)
         )
@@ -436,6 +433,7 @@ class PreventiveEditView(IshtarMixin, LoginRequiredMixin, MixFormFormsetUpdateVi
             formset.form_label = str(inline_type)
             formset.form_slug = inline_type.txt_idx
             formset.dynamic_add = True
+            formset.type_filter = inline_type.txt_idx
             inlines.append(formset)
         return inlines
 
@@ -468,8 +466,13 @@ class PreventiveEditView(IshtarMixin, LoginRequiredMixin, MixFormFormsetUpdateVi
     def get_context_data(self, **kwargs):
         context = super(PreventiveEditView, self).get_context_data(**kwargs)
         unities = {}
+        flat_rates = set()
         for inline_formset in context["inline_forms"]:
             if inline_formset.forms and hasattr(inline_formset.forms[0], "unities"):
                 unities.update(inline_formset.forms[0].unities)
+            if inline_formset.forms and hasattr(inline_formset.forms[0], "flat_rates"):
+                for form in inline_formset.forms:
+                    flat_rates.update(form.flat_rates)
         context["form_unities"] = unities.items()
+        context["form_flat_rates"] = flat_rates
         return context