Manage multiple condition for cascading shortcut menu - Manage shortcut menu for treatment and treatment files (refs #3384)

1 files changed, 15 insertions, 3 deletions

diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index 783c6badb..cf3f968b1 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -264,7 +264,7 @@ def shortcut_menu(request):
             'ishtar/blocks/advanced_shortcut_menu.html',
             dct, context_instance=RequestContext(request))
     dct = {'current_menu': []}
-    current_selected_item = None
+    current_selected_item = {}
     for lbl, model in CURRENT_ITEMS:
         new_selected_item = None
         model_name = model.SLUG
@@ -294,7 +294,8 @@ def shortcut_menu(request):
                 pass
         if items:
             dct['current_menu'].append((lbl, model_name, cls, items))
-        current_selected_item = new_selected_item
+        if new_selected_item:
+            current_selected_item[model_name] = new_selected_item
     return render_to_response('ishtar/blocks/shortcut_menu.html', dct,
                               context_instance=RequestContext(request))
 
@@ -317,6 +318,16 @@ def get_current_items(request):
 
 
 def unpin(request, item_type):
+    if item_type not in ('find', 'contextrecord', 'operation', 'file',
+                          'treatment', 'treatmentfile'):
+        logger.warning("unpin unknow type: {}".format(item_type))
+        return HttpResponse('nok')
+    request.session['treatment'] = ''
+    if item_type == 'treatment':
+        return HttpResponse('ok')
+    request.session['treatmentfile'] = ''
+    if item_type == 'treatmentfile':
+        return HttpResponse('ok')
     request.session['find'] = ''
     if item_type == 'find':
         return HttpResponse('ok')
@@ -327,7 +338,8 @@ def unpin(request, item_type):
     if item_type == 'operation':
         return HttpResponse('ok')
     request.session['file'] = ''
-    return HttpResponse('ok')
+    if item_type == 'file':
+        return HttpResponse('ok')
 
 
 def update_current_item(request, item_type=None, pk=None):