Source code for shuup.admin.modules.suppliers.views.list

from django.conf import settings
from django.db.models import Q
from django.utils.translation import gettext_lazy as _

from shuup.admin.shop_provider import get_shop
from shuup.admin.toolbar import Toolbar
from shuup.admin.utils.picotable import Column, TextFilter
from shuup.admin.utils.views import PicotableListView
from shuup.core.models import Supplier




class SupplierListView(PicotableListView):
    model = Supplier
    default_columns = [
        Column(
            "name",
            _("Name"),
            sort_field="name",
            display="name",
            filter_config=TextFilter(filter_field="name", placeholder=_("Filter by name...")),
        ),
        Column("type", _("Type")),
        Column(
            "supplier_modules",
            _("Modules"),
            display="get_supplier_modules",
            sortable=True,
        ),
    ]
    toolbar_buttons_provider_key = "supplier_list_toolbar_provider"
    mass_actions_provider_key = "supplier_list_mass_actions_provider"



    def get_queryset(self):
        return Supplier.objects.filter(Q(shops=get_shop(self.request)) | Q(shops__isnull=True)).not_deleted()




    def get_supplier_modules(self, instance):
        return (
            ", ".join(instance.supplier_modules.all().values_list("name", flat=True))
            or _("No %s module") % self.model._meta.verbose_name
        )




    def get_toolbar(self):
        if settings.SHUUP_ENABLE_MULTIPLE_SUPPLIERS:
            return super().get_toolbar()
        else:
            return Toolbar.for_view(self)