from django.db.models import Q
from django.utils.translation import gettext_lazy as _
from six import string_types
from shuup.admin.modules.products.views.list import ProductListView
from shuup.admin.shop_provider import get_shop
from shuup.admin.utils.mass_action import BaseExportCSVMassAction
from shuup.admin.utils.picotable import PicotableMassAction, PicotableRedirectMassAction
from shuup.core.models import ShopProduct, ShopProductVisibility
from shuup.core.utils import context_cache
from shuup.utils.django_compat import reverse
[docs]
class VisibleMassAction(PicotableMassAction):
label = _("Set visible")
identifier = "mass_action_product_visible"
[docs]
def process(self, request, ids):
shop = get_shop(request)
if isinstance(ids, string_types) and ids == "all":
query = Q(shop=shop)
else:
query = Q(pk__in=ids, shop=shop)
ShopProduct.objects.filter(query).update(visibility=ShopProductVisibility.ALWAYS_VISIBLE)
for shop_product in ShopProduct.objects.filter(query).iterator():
context_cache.bump_cache_for_shop_product(shop_product)
[docs]
class InvisibleMassAction(PicotableMassAction):
label = _("Set invisible")
identifier = "mass_action_product_invisible"
[docs]
def process(self, request, ids):
shop = get_shop(request)
if isinstance(ids, string_types) and ids == "all":
query = Q(shop=shop)
else:
query = Q(pk__in=ids, shop=shop)
ShopProduct.objects.filter(query).update(visibility=ShopProductVisibility.NOT_VISIBLE)
for shop_product in ShopProduct.objects.filter(query).iterator():
context_cache.bump_cache_for_shop_product(shop_product)
[docs]
class ExportProductsCSVAction(BaseExportCSVMassAction):
identifier = "mass_action_export_product_csv"
model = ShopProduct
view_class = ProductListView
filename = "products.csv"
[docs]
class EditProductAttributesAction(PicotableRedirectMassAction):
label = _("Edit products")
identifier = "mass_action_edit_product"
redirect_url = reverse("shuup_admin:shop_product.mass_edit")