first try at implementing product viewset "related" action
This commit is contained in:
@@ -23,7 +23,7 @@ from products.models import Product, CategoryTag
|
||||
from products.serializers import ProductSerializer, TagFilterSerializer, SearchResultSerializer
|
||||
from companies.models import Company
|
||||
from back_latienda.permissions import IsCreator, IsSiteAdmin, ReadOnly
|
||||
from .utils import extract_search_filters, find_related_products_v3, find_related_products_v6, product_loader
|
||||
from .utils import extract_search_filters, find_related_products_v6, product_loader, find_related_products_v7
|
||||
from utils.tag_serializers import TaggitSerializer
|
||||
from utils.tag_filters import ProductTagFilter, ProductOrderFilter
|
||||
|
||||
@@ -37,19 +37,24 @@ logging.basicConfig(
|
||||
|
||||
|
||||
class ProductViewSet(viewsets.ModelViewSet):
|
||||
queryset = Product.objects.filter(active=True).order_by('-created')
|
||||
serializer_class = ProductSerializer
|
||||
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
|
||||
filter_backends = [DjangoFilterBackend, OrderingFilter]
|
||||
filterset_class = ProductTagFilter
|
||||
queryset = Product.objects.filter(active=True).order_by('-created')
|
||||
serializer_class = ProductSerializer
|
||||
permission_classes = [IsAuthenticatedOrReadOnly, IsCreator]
|
||||
filter_backends = [DjangoFilterBackend, OrderingFilter]
|
||||
filterset_class = ProductTagFilter
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(creator=self.request.user, company=self.request.user.company)
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(creator=self.request.user, company=self.request.user.company)
|
||||
|
||||
@action(detail=True, methods=['GET',])
|
||||
def related(request):
|
||||
# TODO: find the most similar products
|
||||
return Response(data=[])
|
||||
@action(detail=True, methods=['GET',])
|
||||
def related(self, request, pk=None):
|
||||
"""Find instances similar to the one referenced
|
||||
"""
|
||||
# TODO: find the most similar products
|
||||
product = self.get_object()
|
||||
qs = find_related_products_v7(product.description, product.tags.all(), product.attributes.all(), product.category)
|
||||
serializer = self.serializer_class(qs, many=True)
|
||||
return Response(data=serializer.data)
|
||||
|
||||
|
||||
class MyProductsViewSet(viewsets.ModelViewSet):
|
||||
|
||||
Reference in New Issue
Block a user