feat: add filters to endpoints

Author: Juan Benitez

api_crud/settings.py

@@ -30,7 +30,10 @@
 REST_FRAMEWORK = {
     'DEFAULT_AUTHENTICATION_CLASSES': (
         'rest_framework_simplejwt.authentication.JWTAuthentication',
-    )
+    ),
+    'DEFAULT_FILTER_BACKENDS': (
+        'django_filters.rest_framework.DjangoFilterBackend',
+    ),
 }
 
 
@@ -44,6 +47,7 @@
     'django.contrib.messages',
     'django.contrib.staticfiles',
     'rest_framework',
+    'django_filters',
     'authentication',
     'movies',
 ]

movies/filters.py

@@ -0,0 +1,17 @@
+from django_filters import rest_framework as filters
+from .models import Movie
+
+
+# We create filters for each field we want to be able to filter on
+class MovieFilter(filters.FilterSet):
+    title = filters.CharFilter(lookup_expr='icontains')
+    genre = filters.CharFilter(lookup_expr='icontains')
+    year = filters.NumberFilter()
+    year__gt = filters.NumberFilter(field_name='year', lookup_expr='gt')
+    year__lt = filters.NumberFilter(field_name='year', lookup_expr='lt')
+    creator__username = filters.CharFilter(lookup_expr='icontains')
+
+    class Meta:
+        model = Movie
+        fields = ['title', 'genre', 'year', 'year__gt', 'year__lt', 'creator__username']
+

movies/models.py

@@ -9,7 +9,7 @@ class Movie(models.Model):
     updated_at = models.DateTimeField(auto_now=True)
     creator = models.ForeignKey('auth.User', related_name='movies', on_delete=models.CASCADE)
 
-
-
+    class Meta:
+        ordering = ['-id']
 
 

movies/views.py

@@ -1,17 +1,20 @@
 from rest_framework.generics import RetrieveUpdateDestroyAPIView, ListCreateAPIView
 from rest_framework.permissions import IsAuthenticated
+from django_filters import rest_framework as filters
 from .models import Movie
 from .permissions import IsOwnerOrReadOnly
 from .serializers import MovieSerializer
 from .pagination import CustomPagination
+from .filters import MovieFilter
 
 
 class ListCreateMovieAPIView(ListCreateAPIView):
-
     serializer_class = MovieSerializer
     queryset = Movie.objects.all()
     permission_classes = [IsAuthenticated]
     pagination_class = CustomPagination
+    filter_backends = (filters.DjangoFilterBackend,)
+    filterset_class = MovieFilter
 
     def perform_create(self, serializer):
         # Assign the user who created the movie