diff options
-rw-r--r-- | archaeological_operations/forms.py | 2 | ||||
-rw-r--r-- | ishtar_common/wizards.py | 3 |
2 files changed, 5 insertions, 0 deletions
diff --git a/archaeological_operations/forms.py b/archaeological_operations/forms.py index 6e025c4a4..9f8ab00fe 100644 --- a/archaeological_operations/forms.py +++ b/archaeological_operations/forms.py @@ -113,8 +113,10 @@ class ParcelFormSet(FormSet): def __init__(self, *args, **kwargs): super(FormSet, self).__init__(*args, **kwargs) + self.extra_form = None if self.forms[0].__class__.__name__ == 'ParcelForm': self.selection_form = ParcelSelectionForm() + self.extra_form = self.selection_form def as_table(self): # add dynamic widget diff --git a/ishtar_common/wizards.py b/ishtar_common/wizards.py index f27a2bd55..ec684650f 100644 --- a/ishtar_common/wizards.py +++ b/ishtar_common/wizards.py @@ -513,6 +513,9 @@ class Wizard(NamedUrlWizardView): frm = None if hasattr(form, 'fields') and form.fields.keys(): frm = form + elif hasattr(form, 'extra_form') and hasattr(form.extra_form, 'fields')\ + and form.extra_form.fields.keys(): + frm = form.extra_form elif hasattr(form, 'forms') and form.forms and form.forms[0].fields.keys(): frm = form.forms[0] if frm: |