configure postgresql search system

Author: RustamovAkrom

apps/blog/filters.py

@@ -0,0 +1,14 @@
+import django_filters
+
+from .models import Post
+
+
+class PostFilter(django_filters.FilterSet):
+    title = django_filters.CharFilter(lookup_expr="icontains")
+    description = django_filters.CharFilter(lookup_expr="icontains")
+    content = django_filters.CharFilter(lookup_expr="icontains")
+    created_at = django_filters.DateFromToRangeFilter()
+
+    class Meta:
+        model = Post
+        fields = ['title', 'description', 'content', 'created_at']

apps/blog/utils.py

@@ -1,23 +1,36 @@
 from django.db.models import QuerySet
 from django.db.models import Q
-from django.core.paginator import Paginator, Page, EmptyPage, PageNotAnInteger
+from django.contrib.postgres.search import SearchQuery, SearchRank, SearchVector
+from django.core.paginator import Paginator, Page, EmptyPage, PageNotAnInteger, InvalidPage
+from django.conf import settings
 
 from .models import PostLike, PostDislike, Post, PostComment
 
 
 def get_search_model_queryset(
-    model_queryset: QuerySet, search_query: str = None
+    model_queryset: QuerySet, query: str = None
 ) -> QuerySet:
-    if not search_query:
+    if not query:
         return model_queryset
 
-    search_query = model_queryset.filter(
-        Q(title__icontains=search_query)
-        | Q(description__icontains=search_query)
-        | Q(content__icontains=search_query)
-    )
+    search_vector = SearchVector("title", "description", "content")
+    search_query = SearchQuery(query)
 
-    return search_query
+    if settings.DATABASES['default']['ENGINE'] == 'django.db.backends.postgresql':
+        # PostgreSQL search
+        queryset = model_queryset.annotate(
+            search=search_vector,
+            rank=SearchRank(search_vector, search_query),
+        ).filter(search=search_query).order_by('-rank')
+    else:
+        # SQLite3 search
+        queryset = model_queryset.filter(
+            Q(title__icontains=query)
+            | Q(description__icontains=query)
+            | Q(content__icontains=query)
+        )
+
+    return queryset
 
 
 def get_pagination_obj(model_queryset: QuerySet, page: int = 1, size: int = 4) -> Page:
@@ -29,6 +42,9 @@ def get_pagination_obj(model_queryset: QuerySet, page: int = 1, size: int = 4) -
     except PageNotAnInteger:
         page_obj = paginator.page(1)
 
+    except InvalidPage:
+        page_obj = paginator.page(1)
+
     except EmptyPage:
         page_obj = paginator.page(paginator.num_pages)
 

apps/users/api_endpoints/users/User/tests.py

@@ -102,3 +102,6 @@ def _create_user():
         _check_user_error_field()
         _check_user_passwords_field()
         _create_user()
+
+    def test_api_user_update(self):
+        pass

apps/users/api_endpoints/users/UserProfile/tests.py

@@ -0,0 +1,22 @@
+from rest_framework.test import APITestCase
+from rest_framework_simplejwt.tokens import RefreshToken
+from apps.users.models import UserProfile, User
+
+
+class UserProfileApiTestCase(APITestCase):
+    def setUp(self) -> None:
+        self.username = "Admin"
+        self.email = "admin@gmail.com"
+        self.password = "password"
+        user = User.objects.create(
+            username=self.username,
+            email=self.email,
+        )
+        user.set_password(self.password)
+        user.save()
+        self.user = user
+        refresh = RefreshToken.for_user(self.user)
+        self.token = str(refresh.access_token)
+        return super().setUp()
+    
+