diff options
Diffstat (limited to 'archaeological_finds/views.py')
| -rw-r--r-- | archaeological_finds/views.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py index 828f9801d..f2be734e7 100644 --- a/archaeological_finds/views.py +++ b/archaeological_finds/views.py @@ -118,6 +118,10 @@ display_findbasket = display_item(models.FindBasket, show_url='show-find/basket-') +def check_preservation_module(self): + return get_current_profile().preservation + + def check_warehouse_module(self): return get_current_profile().warehouse @@ -129,13 +133,19 @@ def check_not_warehouse_module(self): find_creation_steps = [ ('selecrecord-find_creation', RecordFormSelectionTable), ('find-find_creation', FindForm), + ('preservation-find_creation', PreservationForm), ('dating-find_creation', DatingFormSet), ('final-find_creation', FinalForm) ] +find_creation_condition_dict = { + 'preservation-find_creation': check_preservation_module, +} + find_creation_wizard = FindWizard.as_view( find_creation_steps, label=_(u"New find"), + condition_dict=find_creation_condition_dict, url_name='find_creation',) find_search_condition_dict = { @@ -154,6 +164,7 @@ find_search_wizard = SearchWizard.as_view([ find_modification_condition_dict = { 'selec-find_modification': check_not_warehouse_module, 'selecw-find_modification': check_warehouse_module, + 'preservation-find_modification': check_preservation_module, } find_modification_wizard = FindModificationWizard.as_view([ @@ -161,6 +172,7 @@ find_modification_wizard = FindModificationWizard.as_view([ ('selecw-find_modification', FindFormSelectionWarehouseModule), ('selecrecord-find_modification', RecordFormSelection), ('find-find_modification', FindForm), + ('preservation-find_modification', PreservationForm), ('dating-find_modification', DatingFormSet), ('final-find_modification', FinalForm)], condition_dict=find_modification_condition_dict, |