Shortcut menu: hide/show information saved in the session

1 files changed, 22 insertions, 4 deletions

diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index 22591164f..c99e78b9c 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -208,6 +208,16 @@ def get_autocomplete_generic(model, extra={'available': True}):
     return func
 
 
+def hide_shortcut_menu(request):
+    request.session['SHORTCUT_SHOW'] = 'off'
+    return HttpResponse('OK', mimetype='text/plain')
+
+
+def show_shortcut_menu(request):
+    request.session['SHORTCUT_SHOW'] = 'on'
+    return HttpResponse('OK', mimetype='text/plain')
+
+
 def activate_all_search(request):
     request.session['SHORTCUT_SEARCH'] = 'all'
     return HttpResponse('OK', mimetype='text/plain')
@@ -249,9 +259,13 @@ def shortcut_menu(request):
         CURRENT_ITEMS.append((_(u"Treatment"), Treatment))
     if hasattr(request.user, 'ishtaruser') and \
             request.user.ishtaruser.advanced_shortcut_menu:
-        dct = {'current_menu': [], 'menu': [],
-               'SHORTCUT_SEARCH': request.session['SHORTCUT_SEARCH']
-               if 'SHORTCUT_SEARCH' in request.session else 'own'}
+        dct = {
+            'current_menu': [], 'menu': [],
+            'SHORTCUT_SEARCH': request.session['SHORTCUT_SEARCH']
+            if 'SHORTCUT_SEARCH' in request.session else 'own',
+            'SHORTCUT_SHOW': request.session['SHORTCUT_SHOW']
+            if 'SHORTCUT_SHOW' in request.session else 'on'
+        }
 
         for lbl, model in CURRENT_ITEMS:
             model_name = model.SLUG
@@ -267,7 +281,11 @@ def shortcut_menu(request):
         return render_to_response(
             'ishtar/blocks/advanced_shortcut_menu.html',
             dct, context_instance=RequestContext(request))
-    dct = {'current_menu': []}
+    dct = {
+        'current_menu': [],
+        'SHORTCUT_SHOW': request.session['SHORTCUT_SHOW']
+        if 'SHORTCUT_SHOW' in request.session else 'off'
+    }
     current_selected_item = {}
     for lbl, model in CURRENT_ITEMS:
         new_selected_item = None