mirror of
https://github.com/magnum-opus-tender-hack/backend.git
synced 2024-11-25 02:33:45 +03:00
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
from drf_yasg import openapi
|
|
from drf_yasg.utils import swagger_auto_schema
|
|
from rest_framework import status
|
|
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,
|
|
AutoCompleteRequestSerializer,
|
|
AutoCompleteResponseSerializer,
|
|
)
|
|
from search.services.search import process_search
|
|
from search.services.autocomplete_schema import autocomplete_schema
|
|
|
|
from search.services.hints import get_hints
|
|
|
|
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})
|
|
def post(self, request):
|
|
serializer = SearchSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
return Response(
|
|
process_search(serializer.data["body"]), status=status.HTTP_200_OK
|
|
)
|
|
|
|
|
|
class HintApi(APIView):
|
|
@swagger_auto_schema(
|
|
request_body=HintRequestSerializer, responses={200: hint_response}
|
|
)
|
|
def post(self, request):
|
|
serializer = HintRequestSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
return Response(
|
|
{
|
|
"type": get_hints(serializer.data["content"]),
|
|
"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):
|
|
serializer = AutoCompleteRequestSerializer(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
return Response(
|
|
{"nodes": autocomplete_schema(serializer.data["content"])},
|
|
status=status.HTTP_200_OK,
|
|
)
|