partial improvements to search
This commit is contained in:
Sam
@@ -90,115 +90,116 @@ def load_coop_products(request):
|
||||
|
||||
@api_view(['GET',]) # include allowed methods
|
||||
def product_search(request):
|
||||
"""
|
||||
Takes a string of data, return relevant products
|
||||
"""
|
||||
Takes a string of data, return relevant products
|
||||
|
||||
Params:
|
||||
- q: used for search [MANDATORY]
|
||||
- limit: max number of returned instances [OPTIONAL]
|
||||
- offset: where to start counting results [OPTIONAL]
|
||||
- shipping_cost: true/false
|
||||
- discount: true/false
|
||||
- category: string
|
||||
- tags: string
|
||||
- order: string (newest/oldest)
|
||||
- price_min: int
|
||||
- price_max: int
|
||||
Params:
|
||||
- q: used for search [MANDATORY]
|
||||
- limit: max number of returned instances [OPTIONAL]
|
||||
- offset: where to start counting results [OPTIONAL]
|
||||
- shipping_cost: true/false
|
||||
- discount: true/false
|
||||
- category: string
|
||||
- tags: string
|
||||
- order: string (newest/oldest)
|
||||
- price_min: int
|
||||
- price_max: int
|
||||
|
||||
In the response:
|
||||
- filters
|
||||
- count
|
||||
- products
|
||||
- price_min
|
||||
- price_max
|
||||
In the response:
|
||||
- filters
|
||||
- count
|
||||
- products
|
||||
- price_min
|
||||
- price_max
|
||||
|
||||
"""
|
||||
# capture query params
|
||||
q = request.GET.get('q', None)
|
||||
limit = request.GET.get('limit', None)
|
||||
offset = request.GET.get('offset', None)
|
||||
shipping_cost = request.GET.get('shipping_cost', None)
|
||||
if shipping_cost is not None:
|
||||
if shipping_cost == 'true':
|
||||
shipping_cost = True
|
||||
elif shipping_cost == 'false':
|
||||
shipping_cost = False
|
||||
else:
|
||||
shipping_cost = None
|
||||
discount = request.GET.get('discount', None)
|
||||
if discount is not None:
|
||||
if discount == 'true':
|
||||
discount = True
|
||||
elif discount == 'false':
|
||||
discount = False
|
||||
else:
|
||||
discount = None
|
||||
category = request.GET.get('category', None)
|
||||
tags = request.GET.get('tags', None)
|
||||
price_min = request.GET.get('price_min', None)
|
||||
price_max = request.GET.get('price_max', None)
|
||||
order = request.GET.get('order', '')
|
||||
"""
|
||||
# capture query params
|
||||
q = request.GET.get('q', None)
|
||||
limit = request.GET.get('limit', None)
|
||||
offset = request.GET.get('offset', None)
|
||||
shipping_cost = request.GET.get('shipping_cost', None)
|
||||
if shipping_cost is not None:
|
||||
if shipping_cost == 'true':
|
||||
shipping_cost = True
|
||||
elif shipping_cost == 'false':
|
||||
shipping_cost = False
|
||||
else:
|
||||
shipping_cost = None
|
||||
discount = request.GET.get('discount', None)
|
||||
if discount is not None:
|
||||
if discount == 'true':
|
||||
discount = True
|
||||
elif discount == 'false':
|
||||
discount = False
|
||||
else:
|
||||
discount = None
|
||||
category = request.GET.get('category', None)
|
||||
tags = request.GET.get('tags', None)
|
||||
price_min = request.GET.get('price_min', None)
|
||||
price_max = request.GET.get('price_max', None)
|
||||
order = request.GET.get('order', '')
|
||||
|
||||
if q is None:
|
||||
return Response({"errors": {"details": "No query string to parse"}})
|
||||
elif q is '':
|
||||
# return everything
|
||||
serializer = ProductSerializer(Product.objects.all(), many=True)
|
||||
products = serializer.data
|
||||
# filters = extract_search_filters(products)
|
||||
return Response(data={"filters": [], "count": len(products), "products": products})
|
||||
try:
|
||||
# we collect our results here
|
||||
result_set = set()
|
||||
# values for response
|
||||
prices = {
|
||||
'min': None,
|
||||
'max': None,
|
||||
}
|
||||
if q is None:
|
||||
return Response({"errors": {"details": "No query string to parse"}})
|
||||
|
||||
try:
|
||||
# we collect our results here
|
||||
result_set = set()
|
||||
# values for response
|
||||
prices = {
|
||||
'min': None,
|
||||
'max': None,
|
||||
}
|
||||
|
||||
if q == '':
|
||||
# filter entire queryset
|
||||
products_qs = Product.objects.all()
|
||||
if tags:
|
||||
products_qs = Product.objects.filter(tags=tags)
|
||||
if category:
|
||||
products_qs = Product.objects.filter(category=category)
|
||||
# serialize and list data
|
||||
serializer = ProductSerializer(products_qs, many=True)
|
||||
result_list = [dict(i) for i in serializer.data]
|
||||
else:
|
||||
# split query string into single words
|
||||
chunks = q.split(' ')
|
||||
for chunk in chunks:
|
||||
product_set, min_price, max_price = find_related_products_v6(chunk, shipping_cost, discount, category, tags, price_min, price_max)
|
||||
# update price values
|
||||
if product_set:
|
||||
# import ipdb; ipdb.set_trace()
|
||||
if prices['min'] is None or min_price['price__min'] < prices['min']:
|
||||
prices['min'] = min_price['price__min']
|
||||
if prices['max'] is None or max_price['price__max'] > prices['max']:
|
||||
prices['max'] = max_price['price__max']
|
||||
# add to result set
|
||||
result_set.update(product_set)
|
||||
# TODO: add search for entire phrase ???
|
||||
product_set, min_price, max_price = find_related_products_v6(chunk, shipping_cost, discount, category, tags, price_min, price_max)
|
||||
# update price values
|
||||
if product_set:
|
||||
if prices['min'] is None or min_price['price__min'] < prices['min']:
|
||||
prices['min'] = min_price['price__min']
|
||||
if prices['max'] is None or max_price['price__max'] > prices['max']:
|
||||
prices['max'] = max_price['price__max']
|
||||
# add to result set
|
||||
result_set.update(product_set)
|
||||
# serialize and list data
|
||||
serializer = SearchResultSerializer(product_set, many=True)
|
||||
result_list = [dict(i) for i in serializer.data]
|
||||
|
||||
# extract filters from result_set
|
||||
filters = extract_search_filters(result_set)
|
||||
# extract filters from result_set
|
||||
filters = extract_search_filters(result_set)
|
||||
# order the results
|
||||
if order == 'newest':
|
||||
# order results by created
|
||||
result_list = sorted(result_list, key= lambda x:x['created'], reverse=True)
|
||||
elif order == 'oldest':
|
||||
# order results by created
|
||||
result_list = sorted(result_list, key= lambda x:x['created'], reverse=False)
|
||||
elif q != '':
|
||||
# order results by RANK
|
||||
result_list = sorted(result_list, key= lambda x:x['rank'], reverse=True)
|
||||
|
||||
result_list = list(result_set)
|
||||
if order == 'newest':
|
||||
# order results by created
|
||||
ordered_products = sorted(result_list, key= lambda x:x.created, reverse=True)
|
||||
elif order == 'oldest':
|
||||
# order results by created
|
||||
ordered_products = sorted(result_list, key= lambda x:x.created, reverse=False)
|
||||
else:
|
||||
# order results by RANK
|
||||
ordered_products = sorted(result_list, key= lambda rank:rank.rank, reverse=True)
|
||||
|
||||
# extract max and min price values
|
||||
serializer = SearchResultSerializer(ordered_products, many=True)
|
||||
product_results = [dict(i) for i in serializer.data]
|
||||
total_results = len(product_results)
|
||||
|
||||
# RESULTS PAGINATION
|
||||
if limit is not None and offset is not None:
|
||||
limit = int(limit)
|
||||
offset = int(offset)
|
||||
product_results = product_results[offset:(limit+offset)]
|
||||
elif limit is not None:
|
||||
limit = int(limit)
|
||||
product_results = product_results[:limit]
|
||||
|
||||
return Response(data={"filters": filters, "count": total_results, "products": product_results, 'prices': prices})
|
||||
except Exception as e:
|
||||
return Response({"errors": {"details": str(e)}}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
|
||||
total_results = len(result_list)
|
||||
# RESULTS PAGINATION
|
||||
if limit is not None and offset is not None:
|
||||
limit = int(limit)
|
||||
offset = int(offset)
|
||||
result_list = result_list[offset:(limit+offset)]
|
||||
elif limit is not None:
|
||||
limit = int(limit)
|
||||
result_list = result_list[:limit]
|
||||
return Response(data={"filters": filters, "count": total_results, "products": result_list, 'prices': prices})
|
||||
except Exception as e:
|
||||
return Response({"errors": {"details": str(e)}}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
|
||||
|
||||
Reference in New Issue
Block a user