get radius ubications
This commit is contained in:
Diego Calvo
@@ -7,6 +7,7 @@ from django.db.models import Q
|
|||||||
from django.core import serializers
|
from django.core import serializers
|
||||||
from django.core.validators import validate_email
|
from django.core.validators import validate_email
|
||||||
from django.contrib.auth import get_user_model
|
from django.contrib.auth import get_user_model
|
||||||
|
from django.contrib.gis.geos import Point
|
||||||
from django.template.loader import render_to_string
|
from django.template.loader import render_to_string
|
||||||
from django.core.mail import EmailMessage
|
from django.core.mail import EmailMessage
|
||||||
from django.db.models import Max, Min
|
from django.db.models import Max, Min
|
||||||
@@ -176,109 +177,122 @@ def product_search(request):
|
|||||||
price_min = request.GET.get('price_min', None)
|
price_min = request.GET.get('price_min', None)
|
||||||
price_max = request.GET.get('price_max', None)
|
price_max = request.GET.get('price_max', None)
|
||||||
order = request.GET.get('order', '')
|
order = request.GET.get('order', '')
|
||||||
|
latitude = request.GET.get('latitude', None)
|
||||||
|
longitude = request.GET.get('longitude', None)
|
||||||
|
|
||||||
try:
|
# try:
|
||||||
# we collect our results here
|
# we collect our results here
|
||||||
result_set = set()
|
result_set = set()
|
||||||
# values for response
|
# values for response
|
||||||
prices = {
|
prices = {
|
||||||
'min': None,
|
'min': None,
|
||||||
'max': None,
|
'max': None,
|
||||||
}
|
}
|
||||||
|
|
||||||
if not q:
|
if not q:
|
||||||
# filter entire queryset
|
# filter entire queryset
|
||||||
products_qs = Product.objects.filter(active=True)
|
products_qs = Product.objects.filter(active=True)
|
||||||
# filter by category
|
# filter by category
|
||||||
if categories is not None:
|
if categories is not None:
|
||||||
descendants = []
|
descendants = []
|
||||||
for entry in categories:
|
for entry in categories:
|
||||||
cat = CategoryTag.objects.filter(label__iexact=entry).first()
|
cat = CategoryTag.objects.filter(label__iexact=entry).first()
|
||||||
# append category tag, and children
|
# append category tag, and children
|
||||||
descendants.append(cat)
|
descendants.append(cat)
|
||||||
descendants.extend(cat.get_descendants())
|
descendants.extend(cat.get_descendants())
|
||||||
|
|
||||||
products_qs = products_qs.filter(category__in=descendants)
|
products_qs = products_qs.filter(category__in=descendants)
|
||||||
# filter by tags
|
# filter by tags
|
||||||
if tags is not None:
|
if tags is not None:
|
||||||
products_qs = products_qs.filter(tags__name__icontains=tags)
|
products_qs = products_qs.filter(tags__name__icontains=tags)
|
||||||
|
|
||||||
# filter by shipping cost
|
# filter by shipping cost
|
||||||
if shipping_cost is True:
|
if shipping_cost is True:
|
||||||
# only instances with shipping costs
|
# only instances with shipping costs
|
||||||
products_qs = products_qs.filter(
|
products_qs = products_qs.filter(
|
||||||
Q(shipping_cost__isnull=False)&
|
Q(shipping_cost__isnull=False)&
|
||||||
Q(shipping_cost__gte=1)
|
Q(shipping_cost__gte=1)
|
||||||
)
|
)
|
||||||
elif shipping_cost is False:
|
elif shipping_cost is False:
|
||||||
# only intances without shpping costs
|
# only intances without shpping costs
|
||||||
products_qs = products_qs.filter(Q(shipping_cost=None)|Q(shipping_cost=0.00))
|
products_qs = products_qs.filter(Q(shipping_cost=None)|Q(shipping_cost=0.00))
|
||||||
|
|
||||||
# filter by discount
|
# filter by discount
|
||||||
if discount is True:
|
if discount is True:
|
||||||
# only instances with shipping costs
|
# only instances with shipping costs
|
||||||
products_qs = products_qs.filter(
|
products_qs = products_qs.filter(
|
||||||
Q(discount__isnull=False)&
|
Q(discount__isnull=False)&
|
||||||
Q(discount__gte=1)
|
Q(discount__gte=1)
|
||||||
)
|
)
|
||||||
elif discount is False:
|
elif discount is False:
|
||||||
# only intances without shpping costs
|
# only intances without shpping costs
|
||||||
products_qs = products_qs.filter(Q(discount=None)|Q(discount=0.00))
|
products_qs = products_qs.filter(Q(discount=None)|Q(discount=0.00))
|
||||||
|
|
||||||
# filter by price
|
if latitude is not None and longitude is not None:
|
||||||
if price_min is not None:
|
coordinates = (float(longitude), float(latitude))
|
||||||
products_qs = products_qs.filter(price__gte=price_min)
|
# (n km / 40,000 km * 360 degrees) = radius length degrees
|
||||||
if price_max is not None:
|
# Radiuses: 10km, 50km, 200km
|
||||||
products_qs = products_qs.filter(price__lte=price_max)
|
radiuses = [0.09, 0.45, 1.8]
|
||||||
|
for radius in reversed(radiuses):
|
||||||
|
products_geo_filtered = products_qs.filter(company__geo__dwithin=(Point(coordinates), radius))
|
||||||
|
if len(products_geo_filtered) >= 10:
|
||||||
|
products_qs = products_geo_filtered
|
||||||
|
|
||||||
# get min_price and max_price
|
|
||||||
prices['min'] = products_qs.aggregate(Min('price'))['price__min']
|
|
||||||
prices['max'] = products_qs.aggregate(Max('price'))['price__max']
|
|
||||||
# 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 = ranked_product_search(chunk, shipping_cost, discount, categories, 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
|
# filter by price
|
||||||
filters = extract_search_filters(result_set)
|
if price_min is not None:
|
||||||
# order the results
|
products_qs = products_qs.filter(price__gte=price_min)
|
||||||
if order == 'newest':
|
if price_max is not None:
|
||||||
# order results by created
|
products_qs = products_qs.filter(price__lte=price_max)
|
||||||
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)
|
|
||||||
|
|
||||||
total_results = len(result_list)
|
# get min_price and max_price
|
||||||
# RESULTS PAGINATION
|
prices['min'] = products_qs.aggregate(Min('price'))['price__min']
|
||||||
if limit is not None and offset is not None:
|
prices['max'] = products_qs.aggregate(Max('price'))['price__max']
|
||||||
limit = int(limit)
|
# serialize and list data
|
||||||
offset = int(offset)
|
serializer = ProductSerializer(products_qs, many=True)
|
||||||
result_list = result_list[offset:(limit+offset)]
|
result_list = [dict(i) for i in serializer.data]
|
||||||
elif limit is not None:
|
else:
|
||||||
limit = int(limit)
|
# split query string into single words
|
||||||
result_list = result_list[:limit]
|
chunks = q.split(' ')
|
||||||
return Response(data={"filters": filters, "count": total_results, "products": result_list, 'prices': prices})
|
for chunk in chunks:
|
||||||
except Exception as e:
|
product_set, min_price, max_price = ranked_product_search(chunk, shipping_cost, discount, categories, tags, price_min, price_max)
|
||||||
return Response({"errors": {"details": str(e)}}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
|
# 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)
|
||||||
|
# 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)
|
||||||
|
|
||||||
|
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)
|
||||||
|
|
||||||
|
|
||||||
class CategoryTagAutocomplete(autocomplete.Select2QuerySetView):
|
class CategoryTagAutocomplete(autocomplete.Select2QuerySetView):
|
||||||
|
|||||||
Reference in New Issue
Block a user