Files
website-v2/libraries/api.py

50 lines
1.5 KiB
Python

from django.db.models import Q
from rest_framework import permissions
from rest_framework import viewsets
from rest_framework import serializers
from rest_framework import renderers
from rest_framework.response import Response
from .models import Library
class LibrarySearchSerializer(serializers.ModelSerializer):
class Meta:
model = Library
fields = (
"name",
"description",
"slug",
)
class LibrarySearchView(viewsets.ModelViewSet):
model = Library
serializer_class = LibrarySearchSerializer
permission_classes = [permissions.AllowAny]
queryset = Library.objects.all()
renderer_classes = (renderers.TemplateHTMLRenderer,)
def filter_queryset(self, queryset):
"""
This view should return a list of all the libraries that
match the search params limited to 5 results
"""
value = self.request.query_params.get("q")
f = (
Q(name__icontains=value)
| Q(description__icontains=value)
| Q(categories__name__icontains=value)
| Q(authors__display_name__icontains=value)
)
return Library.objects.filter(f).distinct()[:5]
def list(self, request, *args, **kwargs):
queryset = self.filter_queryset(self.get_queryset())
serializer = self.get_serializer(queryset, many=True)
return Response(
{"libraries": serializer.data},
template_name="libraries/includes/search_results.html",
)