Manage find basket modification wizard

1 files changed, 21 insertions, 1 deletions

diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 7523df145..c340639c5 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -118,11 +118,31 @@ get_find_basket = get_item(
 )
 
 basket_search_wizard = FindBasketSearch.as_view(
-    [('general-basket_search', FindBasketFormSelection)],
+    [('selec-find_basket_search', FindBasketFormSelection)],
     label=_(u"Basket search"),
     url_name='find_basket_search',
 )
 
+basket_modify_wizard = FindBasketEditWizard.as_view(
+    [
+        ('selec-find_basket_modification', FindBasketFormSelection),
+        ('basket-find_basket_modification', FindBasketForm),
+        ('final-find_basket_modification', FinalForm)
+     ],
+    label=_(u"Basket modify"),
+    url_name='find_basket_modification',
+)
+
+
+def find_basket_modify(request, pk):
+    basket_modify_wizard(request)
+    key = 'selec-find_basket_modification'
+    FindBasketEditWizard.session_set_value(
+        request, key, 'pk', pk, reset=True)
+    return redirect(
+        reverse('find_basket_modification',
+                kwargs={'step': 'basket-find_basket_modification'}))
+
 
 def check_preservation_module(self):
     return get_current_profile().preservation