Archaeological files: add departments to preventive forms - Wizards manage several many to many in one form

1 files changed, 6 insertions, 5 deletions

diff --git a/archaeological_files_pdl/forms.py b/archaeological_files_pdl/forms.py
index a0c5037e7..3b5ed4601 100644
--- a/archaeological_files_pdl/forms.py
+++ b/archaeological_files_pdl/forms.py
@@ -24,7 +24,7 @@ from django.core import validators
 from django.utils.safestring import mark_safe
 from django.utils.translation import ugettext_lazy as _
 
-from ishtar_common.models import Person, PersonType, Town, valid_id
+from ishtar_common.models import Person, PersonType, Town, Department, valid_id
 from archaeological_files import models
 
 from ishtar_common.forms import get_now, reverse_lazy
@@ -74,12 +74,14 @@ class FileFormPreventiveType(forms.Form):
 
 class FileFormPlanning(forms.Form):
     form_label = _(u"Planning")
-    base_model = 'town'
-    associated_models = {'town': Town}
+    base_models = ['town', 'department']
+    associated_models = {'town': Town, 'department': Department}
     name = forms.CharField(label=_(u"Planning name"), required=False,
                            max_length=100)
     town = widgets.MultipleAutocompleteField(
         model=Town, label=_("Towns"), required=False)
+    department = widgets.MultipleAutocompleteField(
+        model=Department, label=_("Departments"), required=False)
     locality = forms.CharField(label=_(u"Locality"), max_length=100,
                                required=False)
     address = forms.CharField(
@@ -266,7 +268,6 @@ class FileFormPlanningService(forms.Form):
 
     def is_valid(self, *args, **kwargs):
         c = super(FileFormPlanningService, self).is_valid(*args, **kwargs)
-        print(self.cleaned_data)
         return c
 
 
@@ -283,7 +284,7 @@ class FileFormInstruction(forms.Form):
             limit={
                 'person_types': [
                     PersonType.objects.get(txt_idx='sra_agent').pk]
-                },
+            },
             associated_model=Person, new=True),
         validators=[valid_id(Person)])
     related_file = forms.IntegerField(