def get(self, *args, **kwargs):
self.init_args()
if self.venture:
self.formset = ExtraCostFormSet(
queryset=self.venture.extracost_set.order_by('start',
'type'), )
return super(ExtraCosts, self).get(*args, **kwargs)
def get(self, *args, **kwargs):
self.init_args()
if self.extra_cost_type:
self.formset = ExtraCostFormSet(
queryset=self.extra_cost_type.extracost_set.all(),
)
return super(ExtraCosts, self).get(*args, **kwargs)
def post(self, *args, **kwargs):
self.init_args()
if self.venture:
self.formset = ExtraCostFormSet(
self.request.POST,
queryset=self.venture.extracost_set.order_by('start', 'type'),
)
if self.formset.is_valid():
for form in self.formset.extra_forms:
if form.has_changed():
form.instance.pricing_venture = self.venture
self.formset.save()
messages.success(self.request, "Extra costs updated.")
return HttpResponseRedirect(self.request.path)
else:
messages.error(self.request, "Please fix the errors.")
return super(ExtraCosts, self).get(*args, **kwargs)
