Source code for shuup.admin.modules.products.views.edit_package

from django.contrib import messages
from django.forms.formsets import formset_factory
from django.http import HttpResponseRedirect
from django.utils.translation import gettext_lazy as _

from shuup.admin.modules.products.forms import PackageChildForm, PackageChildFormSet
from shuup.admin.modules.products.utils import clear_existing_package
from shuup.admin.toolbar import PostActionButton
from shuup.core.models import ProductMode
from shuup.utils.django_compat import reverse
from shuup.utils.excs import Problem

from .edit_parent import ProductChildrenBaseFormPart, ProductParentBaseToolbar, ProductParentBaseView



[docs]
class ProductChildrenFormPart(ProductChildrenBaseFormPart):
    invalid_modes = [
        ProductMode.VARIATION_CHILD,
        ProductMode.VARIABLE_VARIATION_PARENT,
        ProductMode.SIMPLE_VARIATION_PARENT,
    ]
    priority = 1


[docs]
    def get_form_defs(self):
        product = self.object
        if product.mode in self.invalid_modes:
            raise ValueError("Error! Invalid mode.")
        else:
            form = formset_factory(PackageChildForm, PackageChildFormSet, extra=5, can_delete=True)
            template_name = "shuup/admin/products/package/_package_children.jinja"

        form_defs = super().get_form_defs(form, template_name)
        yield from form_defs





[docs]
class ProductPackageViewToolbar(ProductParentBaseToolbar):
    button_text = _("Clear package")
    confirm_text = _("Are you sure? This will remove all products from the package.")


[docs]
    def __init__(self, view):
        super().__init__(view)
        if self.parent_product.get_package_child_to_quantity_map():
            self.append(
                PostActionButton(
                    post_url=self.request.path,
                    name="command",
                    value="clear_package",
                    confirm=self.confirm_text,
                    text=self.button_text,
                    extra_css_class="btn-danger",
                    icon="fa fa-times",
                )
            )





[docs]
class ProductPackageView(ProductParentBaseView):
    template_name = "shuup/admin/products/package/edit.jinja"
    form_part_classes = [ProductChildrenFormPart]
    toolbar_class = ProductPackageViewToolbar


[docs]
    def dispatch(self, request, *args, **kwargs):
        self.object = self.get_object()
        parent = self.object.get_all_package_parents().first()
        if parent:
            # By default, redirect to the first parent
            return HttpResponseRedirect(reverse("shuup_admin:shop_product.edit_package", kwargs={"pk": parent.id}))
        return super().dispatch(request, *args, **kwargs)



[docs]
    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context["title"] = _("Edit Package: %s") % self.object
        context["is_package"] = self.object.is_package_parent()
        return context



[docs]
    def dispatch_command(self, request, command):
        product = self.object
        if command == "clear_package":
            clear_existing_package(product)
            messages.success(self.request, _("Package cleared."))
        else:
            raise Problem(f"Error! Unknown command: `{command}`.")
        return HttpResponseRedirect(self.get_success_url())