mirror of
https://github.com/magnum-opus-tender-hack/backend.git
synced 2024-11-10 19:46:35 +03:00
add autocomplete
This commit is contained in:
parent
7b3b3391d6
commit
a06eb94215
|
@ -1,8 +1,9 @@
|
|||
from django.urls import path
|
||||
|
||||
from search.api.views import SearchApi, HintApi
|
||||
from search.api.views import SearchApi, HintApi, AutoCompleteApi
|
||||
|
||||
urlpatterns = [
|
||||
path("search", SearchApi.as_view(), name="search_api"),
|
||||
path("hint", HintApi.as_view(), name="hint api")
|
||||
path("hint", HintApi.as_view(), name="hint api"),
|
||||
path('autocomplete_schema', AutoCompleteApi.as_view(), name='autocomplete api')
|
||||
]
|
||||
|
|
|
@ -1,4 +1,6 @@
|
|||
from rest_framework import serializers
|
||||
from django.core.validators import MinLengthValidator, MinValueValidator
|
||||
|
||||
|
||||
|
||||
class SearchSerializer(serializers.Serializer):
|
||||
|
@ -31,3 +33,16 @@ class HintRequestSerializer(serializers.Serializer):
|
|||
class HintResponseSerializer(serializers.Serializer):
|
||||
type = serializers.CharField()
|
||||
content = serializers.CharField()
|
||||
|
||||
|
||||
class AutoCompleteRequestSerializer(serializers.Serializer):
|
||||
content = serializers.CharField(validators=[MinLengthValidator(3)])
|
||||
|
||||
|
||||
class AutoCompleteSerializerNode(serializers.Serializer):
|
||||
coordinate = serializers.IntegerField(validators=[MinValueValidator(0)])
|
||||
value = HintResponseSerializer()
|
||||
|
||||
|
||||
class AutoCompleteResponseSerializer(serializers.Serializer):
|
||||
nodes = serializers.ListField(child=AutoCompleteSerializerNode())
|
||||
|
|
|
@ -5,11 +5,14 @@ from rest_framework.response import Response
|
|||
from rest_framework.views import APIView
|
||||
from search.api.serializers import HintRequestSerializer
|
||||
|
||||
from search.api.serializers import SearchSerializer, ResponseSerializer, HintResponseSerializer
|
||||
from search.api.serializers import SearchSerializer, ResponseSerializer, HintResponseSerializer, AutoCompleteRequestSerializer, AutoCompleteResponseSerializer
|
||||
from search.services.search import process_string
|
||||
from search.services.autocomplete_schema import autocomplete_schema
|
||||
|
||||
user_response = openapi.Response("search results", ResponseSerializer)
|
||||
hint_response = openapi.Response("hints", HintResponseSerializer)
|
||||
autocomplete_response = openapi.Response("autocomplete schema", AutoCompleteResponseSerializer)
|
||||
|
||||
|
||||
class SearchApi(APIView):
|
||||
@swagger_auto_schema(request_body=SearchSerializer, responses={200: user_response})
|
||||
|
@ -32,4 +35,15 @@ class HintApi(APIView):
|
|||
'value': serializer.data['content']
|
||||
},
|
||||
status=status.HTTP_200_OK
|
||||
)
|
||||
)
|
||||
|
||||
class AutoCompleteApi(APIView):
|
||||
@swagger_auto_schema(request_body=AutoCompleteRequestSerializer, responses={200: autocomplete_response})
|
||||
def post(self, request, format=None):
|
||||
serializer = AutoCompleteRequestSerializer(data=request.data)
|
||||
serializer.is_valid(raise_exception=True)
|
||||
return Response(
|
||||
{
|
||||
'nodes': autocomplete_schema(serializer.data['content'])
|
||||
}, status=status.HTTP_200_OK
|
||||
)
|
||||
|
|
37
app/search/services/autocomplete_schema.py
Normal file
37
app/search/services/autocomplete_schema.py
Normal file
|
@ -0,0 +1,37 @@
|
|||
from search.models import Product, Category, Characteristic
|
||||
|
||||
def autocomplete_schema(val: str):
|
||||
schema = []
|
||||
schema.extend(
|
||||
[
|
||||
{
|
||||
'coordinate': product['name'].index(val),
|
||||
'value': {
|
||||
'type': 'Name',
|
||||
'value': product['name'],
|
||||
}
|
||||
} for product in Product.objects.filter(name__contains=val).values('name')]
|
||||
)
|
||||
schema.extend(
|
||||
[
|
||||
{
|
||||
'coordinate': cat['name'].index(val),
|
||||
'value': {
|
||||
'type': 'Category',
|
||||
'value': cat['name']
|
||||
}
|
||||
} for cat in Category.objects.filter(name__contains=val).values('name')
|
||||
]
|
||||
)
|
||||
schema.extend(
|
||||
[
|
||||
{
|
||||
'coordinate': char.name.index(val),
|
||||
'value': {
|
||||
'type': char.name,
|
||||
'value': char.value
|
||||
}
|
||||
} for char in Characteristic.objects.filter(name__contains=val).values('name', 'value')
|
||||
]
|
||||
)
|
||||
return schema
|
Loading…
Reference in New Issue
Block a user