initail elasticsearch

apps/authentication/api/v1/search_view.py

@@ -0,0 +1,53 @@
+from apps.authentication.api.v1.serializers.serializer import UserSerializer
+from rest_framework.pagination import LimitOffsetPagination
+from rest_framework.viewsets import ModelViewSet, ViewSet
+from apps.authentication.documents import UserDocument
+from rest_framework.response import Response
+from django.http.response import HttpResponse
+from apps.authentication.models import User
+from rest_framework.views import APIView
+from elasticsearch_dsl.query import Q
+import abc
+
+
+class PaginatedElasticSearchApiView(APIView, LimitOffsetPagination):
+    """Base ApiView Class for elasticsearch views with pagination
+    Other ApiView classes should inherit from this class"""
+    serializer_class = None
+    document_class = None
+
+    @abc.abstractmethod
+    def generate_q_expression(self, query):
+        """This method should be overridden
+        and return a Q() expression."""
+
+    def get(self, request, query):
+        try:
+            q = self.generate_q_expression(query)
+            search = self.document_class.search().query(q)
+            response = search.execute()
+
+            print(f"Found {response.hits.total.value} hit(s) for query: '{query}'")
+
+            results = self.paginate_queryset(response, request, view=self)  # noqa
+            serializer = self.serializer_class(results, many=True)
+            return self.get_paginated_response(serializer.data)
+        except Exception as e:
+            return HttpResponse(e, status=500)
+
+
+class SearchUsersApiView(PaginatedElasticSearchApiView):  # noqa
+    """Search in Users"""
+
+    serializer_class = UserSerializer
+    document_class = UserDocument
+
+    def generate_q_expression(self, query):
+        return Q(
+            'multi_match',
+            query=query,
+            fields=[
+                'username',
+                'mobile'
+            ], fuzziness='auto'
+        )