first try at implementing product viewset "related" action
This commit is contained in:
@@ -194,6 +194,21 @@ class ProductViewSetTest(APITestCase):
|
|||||||
# Assert number of instnaces in response
|
# Assert number of instnaces in response
|
||||||
self.assertEquals(len(expected_instance), len(payload))
|
self.assertEquals(len(expected_instance), len(payload))
|
||||||
|
|
||||||
|
def test_anon_can_get_related_products(self):
|
||||||
|
# Create instances
|
||||||
|
instance = self.factory()
|
||||||
|
# make our user the creator
|
||||||
|
instance.creator = self.user
|
||||||
|
instance.save()
|
||||||
|
|
||||||
|
url = f"{self.endpoint}{instance.id}/related/"
|
||||||
|
|
||||||
|
response = self.client.get(url)
|
||||||
|
|
||||||
|
self.assertEquals(response.status_code, 200)
|
||||||
|
payload= response.json()
|
||||||
|
self.assertTrue(len(payload) <= 6)
|
||||||
|
|
||||||
# authenticated user
|
# authenticated user
|
||||||
def test_auth_user_can_paginate_instances(self):
|
def test_auth_user_can_paginate_instances(self):
|
||||||
"""authenticated user can paginate instances
|
"""authenticated user can paginate instances
|
||||||
|
|||||||
@@ -85,6 +85,16 @@ def extract_search_filters(result_set):
|
|||||||
return filter_dict
|
return filter_dict
|
||||||
|
|
||||||
|
|
||||||
|
def find_related_products_v7(description, tags, attributes, category):
|
||||||
|
products_qs = Product.objects.filter(
|
||||||
|
description=description,
|
||||||
|
tags__in=tags,
|
||||||
|
attributes__in=attributes,
|
||||||
|
category=category
|
||||||
|
)[:6]
|
||||||
|
return products_qs
|
||||||
|
|
||||||
|
|
||||||
def find_related_products_v3(keyword):
|
def find_related_products_v3(keyword):
|
||||||
"""
|
"""
|
||||||
Ranked product search
|
Ranked product search
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ from products.models import Product, CategoryTag
|
|||||||
from products.serializers import ProductSerializer, TagFilterSerializer, SearchResultSerializer
|
from products.serializers import ProductSerializer, TagFilterSerializer, SearchResultSerializer
|
||||||
from companies.models import Company
|
from companies.models import Company
|
||||||
from back_latienda.permissions import IsCreator, IsSiteAdmin, ReadOnly
|
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_serializers import TaggitSerializer
|
||||||
from utils.tag_filters import ProductTagFilter, ProductOrderFilter
|
from utils.tag_filters import ProductTagFilter, ProductOrderFilter
|
||||||
|
|
||||||
@@ -47,9 +47,14 @@ class ProductViewSet(viewsets.ModelViewSet):
|
|||||||
serializer.save(creator=self.request.user, company=self.request.user.company)
|
serializer.save(creator=self.request.user, company=self.request.user.company)
|
||||||
|
|
||||||
@action(detail=True, methods=['GET',])
|
@action(detail=True, methods=['GET',])
|
||||||
def related(request):
|
def related(self, request, pk=None):
|
||||||
|
"""Find instances similar to the one referenced
|
||||||
|
"""
|
||||||
# TODO: find the most similar products
|
# TODO: find the most similar products
|
||||||
return Response(data=[])
|
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):
|
class MyProductsViewSet(viewsets.ModelViewSet):
|
||||||
|
|||||||
Reference in New Issue
Block a user