from django import forms
from django.urls import reverse
from django.utils.translation import get_language
from django.utils.translation import gettext_lazy as _
from enumfields import Enum
from shuup.core.catalog import ProductCatalog, ProductCatalogContext
from shuup.core.models import Product, ProductCrossSell, ProductCrossSellType, ProductMode, ShopProductVisibility
from shuup.front.template_helpers.general import (
get_best_selling_products,
get_newest_products,
get_products_for_categories,
get_random_products,
)
from shuup.front.template_helpers.product import get_product_cross_sells, map_relation_type
from shuup.xtheme import TemplatedPlugin
from shuup.xtheme.plugins.forms import GenericPluginForm, TranslatableField
from shuup.xtheme.plugins.widgets import XThemeSelect2ModelChoiceField, XThemeSelect2ModelMultipleChoiceField
[docs]
class HighlightType(Enum):
NEWEST = "newest"
BEST_SELLING = "best_selling"
RANDOM = "random"
class Labels:
NEWEST = _("Newest")
BEST_SELLING = _("Best Selling")
RANDOM = _("Random")
[docs]
class ProductHighlightPlugin(TemplatedPlugin):
identifier = "async_product_highlight"
name = _("Product Highlights (asynchronous)")
template_name = "shuup/xtheme/plugins/highlight_plugin_async.jinja"
cacheable = True
fields = [
("title", TranslatableField(label=_("Title"), required=False, initial="")),
(
"type",
forms.ChoiceField(
label=_("Type"),
choices=HighlightType.choices(),
initial=HighlightType.NEWEST.value,
),
),
("count", forms.IntegerField(label=_("Count"), min_value=1, initial=5)),
(
"cutoff_days",
forms.IntegerField(label=_("Cutoff days"), min_value=1, initial=30),
),
(
"cache_timeout",
forms.IntegerField(label=_("Cache timeout (seconds)"), min_value=0, initial=120),
),
(
"orderable_only",
forms.BooleanField(
label=_("Only show in-stock and orderable items"),
initial=False,
required=False,
help_text=_(
"Warning: The final number of products can be lower than 'Count' "
"as it will filter out unorderable products from a set of 'Count' products."
),
),
),
]
[docs]
def get_cache_key(self, context, **kwargs) -> str:
title = self.get_translated_value("title")
plugin_type = self.config.get("type")
count = self.config.get("count")
cutoff_days = self.config.get("cutoff_days")
cache_timeout = self.config.get("cache_timeout")
orderable_only = self.config.get("orderable_only", False)
return str(
(
get_language(),
title,
plugin_type,
count,
cutoff_days,
cache_timeout,
orderable_only,
context["request"].is_ajax(),
)
)
[docs]
def get_context_data(self, context):
request = context["request"]
plugin_type = self.config.get("type", HighlightType.NEWEST.value)
count = self.config.get("count", 5)
cutoff_days = self.config.get("cutoff_days", 30)
cache_timeout = self.config.get("cache_timeout", 0)
orderable_only = self.config.get("orderable_only", False)
products = []
if request.is_ajax():
if plugin_type == HighlightType.NEWEST.value:
products = get_newest_products(context, count, orderable_only)
elif plugin_type == HighlightType.BEST_SELLING.value:
products = get_best_selling_products(context, count, cutoff_days, orderable_only)
elif plugin_type == HighlightType.RANDOM.value:
products = get_random_products(context, count, orderable_only)
return {
"request": request,
"title": self.get_translated_value("title"),
"products": products,
"orderable_only": self.config.get("orderable_only", False),
"data_url": reverse(
"shuup:xtheme-product-highlight",
kwargs={
"plugin_type": plugin_type,
"cutoff_days": cutoff_days,
"count": count,
"cache_timeout": cache_timeout,
},
),
}
[docs]
class ProductCrossSellsPlugin(TemplatedPlugin):
identifier = "async_product_cross_sells"
name = _("Product Cross Sells (asynchronous)")
template_name = "shuup/xtheme/plugins/highlight_plugin_async.jinja"
required_context_variables = ["product"]
fields = [
("title", TranslatableField(label=_("Title"), required=False, initial="")),
("type", ProductCrossSell.type.field.formfield()),
("count", forms.IntegerField(label=_("Count"), min_value=1, initial=5)),
(
"use_variation_parents",
forms.BooleanField(
label=_("Show variation parents"),
help_text=_("Render variation parents instead of the children."),
initial=False,
required=False,
),
),
(
"cache_timeout",
forms.IntegerField(label=_("Cache timeout (seconds)"), min_value=0, initial=120),
),
(
"orderable_only",
forms.BooleanField(
label=_("Only show in-stock and orderable items"),
initial=False,
required=False,
help_text=_(
"Warning: The final number of products can be lower than 'Count' "
"as it will filter out unorderable products from a set of 'Count' products."
),
),
),
]
[docs]
def __init__(self, config):
relation_type = config.get("type", None)
if relation_type:
# Map initial config string to enum type
try:
type = map_relation_type(relation_type)
except LookupError:
type = ProductCrossSellType.RELATED
config["type"] = type
super().__init__(config)
[docs]
def get_cache_key(self, context, **kwargs) -> str:
title = self.get_translated_value("title")
relation_type = self.config.get("type")
count = self.config.get("count")
cache_timeout = self.config.get("cache_timeout")
orderable_only = self.config.get("orderable_only", False)
return str(
(
get_language(),
title,
relation_type,
count,
cache_timeout,
orderable_only,
context["request"].is_ajax(),
)
)
[docs]
def get_context_data(self, context):
request = context["request"]
products = []
use_variation_parents = self.config.get("use_variation_parents", False)
count = self.config.get("count", 5)
cache_timeout = self.config.get("cache_timeout", 0)
product = context.get("product", self.config.get("product"))
orderable_only = self.config.get("orderable_only", False)
relation_type = self.config.get("type")
try:
relation_type = map_relation_type(relation_type)
except LookupError:
relation_type = ProductCrossSellType.RELATED
if request.is_ajax() and product:
if not isinstance(product, Product):
product = Product.objects.filter(id=product).first()
if product:
products = get_product_cross_sells(
context,
product,
relation_type,
count=count,
orderable_only=orderable_only,
use_variation_parents=use_variation_parents,
)
return {
"request": context["request"],
"title": self.get_translated_value("title"),
"products": products,
"orderable_only": self.config.get("orderable_only", False),
"data_url": (
reverse(
"shuup:xtheme-product-cross-sells-highlight",
kwargs={
"product_id": product.id if isinstance(product, Product) else product,
"relation_type": relation_type.label,
"use_parents": (1 if use_variation_parents else 0),
"count": count,
"cache_timeout": cache_timeout,
},
)
if product
else "/"
),
}
[docs]
class ProductsFromCategoryPlugin(TemplatedPlugin):
identifier = "async_category_products"
name = _("Category Products Highlight (asynchronous)")
template_name = "shuup/xtheme/plugins/highlight_plugin_async.jinja"
editor_form_class = ProductsFromCategoryForm
fields = [
("title", TranslatableField(label=_("Title"), required=False, initial="")),
("count", forms.IntegerField(label=_("Count"), min_value=1, initial=5)),
(
"cache_timeout",
forms.IntegerField(label=_("Cache timeout (seconds)"), min_value=0, initial=120),
),
(
"orderable_only",
forms.BooleanField(
label=_("Only show in-stock and orderable items"),
initial=False,
required=False,
help_text=_(
"Warning: The final number of products can be lower than 'Count' "
"as it will filter out unorderable products from a set of 'Count' products."
),
),
),
]
[docs]
def get_cache_key(self, context, **kwargs) -> str:
title = self.get_translated_value("title")
count = self.config.get("count")
cache_timeout = self.config.get("cache_timeout")
orderable_only = self.config.get("orderable_only", False)
return str(
(
get_language(),
title,
count,
cache_timeout,
orderable_only,
context["request"].is_ajax(),
)
)
[docs]
def get_context_data(self, context):
request = context["request"]
products = []
category_id = self.config.get("category")
count = self.config.get("count", 5)
cache_timeout = self.config.get("cache_timeout", 0)
orderable_only = self.config.get("orderable_only", False)
if request.is_ajax() and category_id:
products = get_products_for_categories(
context, [category_id], n_products=count, orderable_only=orderable_only
)
return {
"request": request,
"title": self.get_translated_value("title"),
"products": products,
"orderable_only": orderable_only,
"data_url": reverse(
"shuup:xtheme-category-products-highlight",
kwargs={"category_id": category_id, "count": count, "cache_timeout": cache_timeout},
),
}
[docs]
class ProductSelectionPlugin(TemplatedPlugin):
"""
A plugin that renders a selection of products
"""
identifier = "async_product_selection"
name = _("Product Selection (asynchronous)")
template_name = "shuup/xtheme/plugins/highlight_plugin_async.jinja"
editor_form_class = ProductSelectionConfigForm
fields = [
("title", TranslatableField(label=_("Title"), required=False, initial="")),
(
"cache_timeout",
forms.IntegerField(label=_("Cache timeout (seconds)"), min_value=0, initial=120),
),
]
[docs]
def get_cache_key(self, context, **kwargs) -> str:
title = self.get_translated_value("title")
cache_timeout = self.config.get("cache_timeout")
return str((get_language(), title, cache_timeout, context["request"].is_ajax()))
[docs]
def get_context_data(self, context):
request = context["request"]
products = self.config.get("products", [])
cache_timeout = self.config.get("cache_timeout", 0)
products_qs = Product.objects.none()
if request.is_ajax() and products:
catalog = ProductCatalog(
ProductCatalogContext(
shop=request.shop,
user=getattr(request, "user", None),
contact=getattr(request, "customer", None),
purchasable_only=True,
visibility=ShopProductVisibility.LISTED,
)
)
products_qs = catalog.get_products_queryset().filter(
pk__in=products, mode__in=ProductMode.get_parent_modes()
)
return {
"request": request,
"title": self.get_translated_value("title"),
"products": products_qs,
"data_url": reverse(
"shuup:xtheme-product-selections-highlight",
kwargs={
"product_ids": ",".join(
[(str(prod.pk) if hasattr(prod, "pk") else str(prod)) for prod in products]
),
"cache_timeout": cache_timeout,
},
),
}