Source code for shuup.admin.modules.menu.views.arrange
# -*- coding: utf-8 -*-
# This file is part of Shuup.
#
# Copyright (c) 2012-2021, Shuup Commerce Inc. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
from __future__ import unicode_literals
import json
from django.contrib import messages
from django.http import HttpResponseRedirect
from django.utils.translation import get_language, ugettext_lazy as _
from django.views.generic import RedirectView, TemplateView
from shuup import configuration
from shuup.admin.menu import (
CUSTOM_ADMIN_MENU_STAFF_KEY,
CUSTOM_ADMIN_MENU_SUPERUSER_KEY,
CUSTOM_ADMIN_MENU_SUPPLIER_KEY,
CUSTOM_ADMIN_MENU_USER_PREFIX,
get_menu_entry_categories,
)
from shuup.utils.django_compat import reverse_lazy
[docs]class AdminMenuArrangeView(TemplateView):
"""
Retrieve menus from configuration or display default
"""
template_name = "shuup/admin/menu/arrange.jinja"
success_url = reverse_lazy("shuup_admin:menu.arrange")
reset_url = reverse_lazy("shuup_admin:menu.reset")
[docs] def get_context_data(self, **kwargs):
"""
Populate context with admin_menus
"""
context = super(AdminMenuArrangeView, self).get_context_data(**kwargs)
context["admin_menus"] = get_menu_entry_categories(self.request)
context["reset_url"] = self.reset_url
return context
[docs] def set_configuration(self, request, menus):
config_key = CUSTOM_ADMIN_MENU_USER_PREFIX.format(request.user.pk)
configuration_object = configuration.get(None, config_key, {}) or {}
configuration_object.update({get_language(): menus})
configuration.set(None, config_key, configuration_object)
[docs] def post(self, request):
"""
Save admin menu for current user to the database
"""
self.set_configuration(request, json.loads(request.POST.get("menus")))
messages.add_message(request, messages.SUCCESS, _("Menu saved"))
return HttpResponseRedirect(self.success_url)
[docs]class AdminMenuResetView(RedirectView):
"""
Reset admin menu to default values
"""
permanent = False
url = reverse_lazy("shuup_admin:menu.arrange")
[docs] def reset_configuration(self, request):
configuration.set(None, CUSTOM_ADMIN_MENU_USER_PREFIX.format(request.user.pk), None)
[docs] def get(self, request, *args, **kwargs):
self.reset_configuration(request)
messages.add_message(request, messages.SUCCESS, _("Menu reset"))
return super(AdminMenuResetView, self).get(request, *args, **kwargs)
[docs]class SuperUserMenuArrangeView(AdminMenuArrangeView):
success_url = reverse_lazy("shuup_admin:menu.arrange_superuser")
reset_url = reverse_lazy("shuup_admin:menu.reset_superuser")
[docs] def set_configuration(self, request, menus):
configuration_object = configuration.get(None, CUSTOM_ADMIN_MENU_SUPERUSER_KEY, {}) or {}
configuration_object.update({get_language(): menus})
configuration.set(None, CUSTOM_ADMIN_MENU_SUPERUSER_KEY, configuration_object)
[docs]class SuperUserMenuResetView(AdminMenuResetView):
url = reverse_lazy("shuup_admin:menu.arrange_superuser")
[docs] def reset_configuration(self, request):
configuration.set(None, CUSTOM_ADMIN_MENU_SUPERUSER_KEY, None)
[docs]class StaffMenuArrangeView(AdminMenuArrangeView):
success_url = reverse_lazy("shuup_admin:menu.arrange_staff")
reset_url = reverse_lazy("shuup_admin:menu.reset_staff")
[docs] def set_configuration(self, request, menus):
configuration_object = configuration.get(None, CUSTOM_ADMIN_MENU_STAFF_KEY, {}) or {}
configuration_object.update({get_language(): menus})
configuration.set(None, CUSTOM_ADMIN_MENU_STAFF_KEY, configuration_object)
[docs]class StaffMenuResetView(AdminMenuResetView):
url = reverse_lazy("shuup_admin:menu.arrange_staff")
[docs] def reset_configuration(self, request):
configuration.set(None, CUSTOM_ADMIN_MENU_STAFF_KEY, None)
[docs]class SupplierMenuArrangeView(AdminMenuArrangeView):
success_url = reverse_lazy("shuup_admin:menu.arrange_supplier")
reset_url = reverse_lazy("shuup_admin:menu.reset_supplier")
[docs] def set_configuration(self, request, menus):
configuration_object = configuration.get(None, CUSTOM_ADMIN_MENU_SUPPLIER_KEY, {}) or {}
configuration_object.update({get_language(): menus})
configuration.set(None, CUSTOM_ADMIN_MENU_SUPPLIER_KEY, configuration_object)