From 6fe43294e91aea724d7edce14905cc598eb9e140 Mon Sep 17 00:00:00 2001 From: ajurna Date: Wed, 8 Jul 2015 16:18:08 +0100 Subject: [PATCH] changed menu to include user so that we can hide items from non superusers. --- comic/util.py | 5 +++-- comic/views.py | 8 ++++---- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/comic/util.py b/comic/util.py index fd3f330..5e4e1d8 100644 --- a/comic/util.py +++ b/comic/util.py @@ -6,7 +6,7 @@ import os class Menu: - def __init__(self, page=''): + def __init__(self, user, page=''): """ :type page: str @@ -14,7 +14,8 @@ class Menu: self.menu_items = OrderedDict() self.menu_items['Browse'] = '/comic/' self.menu_items['Account'] = '/comic/account/' - self.menu_items['Settings'] = '/comic/settings/' + if user.is_superuser: + self.menu_items['Settings'] = '/comic/settings/' self.menu_items['Logout'] = '/logout/' self.current_page = page diff --git a/comic/views.py b/comic/views.py index 1e1a56d..de9d8dd 100644 --- a/comic/views.py +++ b/comic/views.py @@ -27,7 +27,7 @@ def comic_list(request, comic_path=''): context = RequestContext(request, { 'file_list': files, 'breadcrumbs': breadcrumbs, - 'menu': Menu('Browse'), + 'menu': Menu(request.user, 'Browse'), }) return render(request, 'comic/comic_list.html', context) @@ -62,7 +62,7 @@ def account_page(request): }) context = RequestContext(request, { 'form': form, - 'menu': Menu('Account'), + 'menu': Menu(request.user, 'Account'), 'error_message': '
'.join(error_message), 'success_message': '
'.join(success_message), }) @@ -97,7 +97,7 @@ def settings_page(request): context = RequestContext(request, { 'error_message': error_message, 'form': form, - 'menu': Menu('Settings') + 'menu': Menu(request.user, 'Settings') }) return render(request, 'comic/settings_page.html', context) @@ -122,7 +122,7 @@ def read_comic(request, comic_path, page): 'orig_file_name': book.page_name(page), 'nav': book.nav(comic_path, page), 'breadcrumbs': breadcrumbs, - 'menu': Menu() + 'menu': Menu(request.user) }) return render(request, 'comic/read_comic.html', context)