initail elasticsearch

apps/authentication/api/v1/api.py

@@ -4,6 +4,7 @@ from rest_framework_simplejwt.views import TokenObtainPairView
 from rest_framework.viewsets import ModelViewSet
 from rest_framework.decorators import action
 from apps.authentication.models import User
+from rest_framework.views import APIView
 from django.db import transaction
 
 
@@ -28,3 +29,7 @@ class Authentication(ModelViewSet):
     @transaction.atomic
     def login(self, request):
         pass
+
+
+class UserViewSet(ModelViewSet):
+    pass

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'
+        )

apps/authentication/api/v1/serializers/serializer.py

@@ -0,0 +1,11 @@
+from rest_framework import serializers
+from apps.authentication.models import User
+
+
+class UserSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = User
+        fields = [
+            'username',
+            'mobile'
+        ]

apps/authentication/api/v1/urls.py

@@ -5,10 +5,14 @@ from rest_framework_simplejwt.views import (
     TokenRefreshView,
     TokenVerifyView
 )
-from .api import CustomizedTokenObtainPairView
+from .api import (
+    CustomizedTokenObtainPairView
+)
+from .search_view import SearchUsersApiView
 
 urlpatterns = [
     path('login/', CustomizedTokenObtainPairView.as_view(), name='token_obtain_pair'),
+    path('search_user/<str:query>', SearchUsersApiView.as_view(), name='search_user'),
     path('token/refresh/', TokenRefreshView.as_view(), name='token_refresh'),
     path('token/verify/', TokenVerifyView.as_view(), name='token_verify'),
 ]

apps/authentication/documents.py

@@ -0,0 +1,24 @@
+from .models import User, Province
+from django_elasticsearch_dsl import Document, fields
+from django_elasticsearch_dsl.registries import registry
+
+
+@registry.register_document
+class UserDocument(Document):
+    """ElasticSearch Document for indexing users"""
+
+    class Index:
+        name = 'users'
+        settings = {
+            'number_of_shards': 1,
+            'number_of_replicas': 0  # number of copies from data in document
+        }
+
+    class Django:
+        model = User
+        fields = [
+            "id",
+            "username",
+            "mobile",
+            "nationality"
+        ]

apps/authentication/views.py

@@ -1,3 +1 @@
-from django.shortcuts import render
-
 # Create your views here.