Source code for shuup.admin.views.edit

from django.apps import apps
from django.core.exceptions import PermissionDenied
from django.http.response import Http404, HttpResponseBadRequest, HttpResponseRedirect
from django.utils.translation import gettext_lazy as _
from django.views.generic import View

from shuup.admin.shop_provider import get_shop
from shuup.admin.utils.urls import NoModelUrl, get_model_url
from shuup.utils.django_compat import force_text
from shuup.utils.excs import Problem




class EditObjectView(View):


    def get(self, request):  # noqa (C901)
        model_name = request.GET.get("model")
        object_id = request.GET.get("pk", request.GET.get("id"))

        if not model_name or not object_id:
            return HttpResponseBadRequest(_("Invalid object."))

        url = None

        try:
            model = apps.get_model(model_name)
        except LookupError:
            return HttpResponseBadRequest(_("Invalid model."))

        instance = model.objects.filter(pk=object_id).first()
        if instance:
            try:
                # try edit first
                try:
                    url = get_model_url(
                        instance,
                        kind="edit",
                        user=request.user,
                        shop=get_shop(request),
                        raise_permission_denied=True,
                    )
                except NoModelUrl:
                    # try detail
                    try:
                        url = get_model_url(
                            instance,
                            kind="detail",
                            user=request.user,
                            shop=get_shop(request),
                            raise_permission_denied=True,
                        )
                    except NoModelUrl:
                        pass
            except PermissionDenied as exception:
                raise Problem(force_text(exception)) from exception

            if url:
                # forward the mode param
                if request.GET.get("mode"):
                    url = f"{url}?mode={request.GET['mode']}"

                return HttpResponseRedirect(url)

        raise Http404(_("Object not found."))