Operation closing (closes #441)

1 files changed, 11 insertions, 1 deletions

diff --git a/ishtar/furnitures/forms.py b/ishtar/furnitures/forms.py
index e9a9159eb..bb6555547 100644
--- a/ishtar/furnitures/forms.py
+++ b/ishtar/furnitures/forms.py
@@ -1227,6 +1227,16 @@ class FileClosingWizard(DeletionWizard):
                 datas[0][1].append(res[field])
         return datas
 
+    def done(self, request, storage, form_list, **kwargs):
+        obj = self.get_current_object(request, storage)
+        for form in form_list:
+            if form.is_valid():
+                if 'end_date' in form.cleaned_data and hasattr(obj, 'end_date'):
+                    obj.end_date = form.cleaned_data['end_date']
+                    obj.save()
+        return render_to_response('wizard_done.html', {},
+                                  context_instance=RequestContext(request))
+
 class FileDeletionWizard(FileClosingWizard):
     def get_formated_datas(self, forms):
         datas = super(FileDeletionWizard, self).get_formated_datas(forms)
@@ -1790,7 +1800,7 @@ class OperationDateFormSelection(forms.Form):
     end_date = forms.DateField(label=_(u"Closing date"),
                                widget=widgets.JQueryDate)
 
-class OperationClosingWizard(DeletionWizard):
+class OperationClosingWizard(FileClosingWizard):
     model = models.Operation
     fields = ['year', 'operation_code', 'operation_type', 'associated_file',
            'in_charge', 'start_date', 'end_date', 'comment', 'towns', 'remains']