在 Django Model ViewSet 中实现多对多字段的搜索

from django.db.models import Q
from rest_framework import viewsets
from rest_framework.decorators import action
from .models import Article, Tag
from .serializers import ArticleSerializer
 
class ArticleViewSet(viewsets.ModelViewSet):
    queryset = Article.objects.all()
    serializer_class = ArticleSerializer
    
    @action(detail=False, methods=['get'])
    def search(self, request):
        query = self.request.query_params.get('query', '')
        # 对查询参数进行空格分割，以支持多个标签搜索
        tags = query.split()
        # 使用Q对象构建复杂查询，每个标签作为OR条件
        tag_query = reduce(lambda q, tag: q | Q(tags__name=tag), tags, Q())
        articles = Article.objects.filter(tag_query)
        serializer = self.get_serializer(articles, many=True)
        return Response(serializer.data)

在这个代码实例中，我们首先从Django的Q对象导入，然后定义了一个名为ArticleViewSet的视图集，它使用ModelViewSet并设置了文章的查询集和序列化器。我们还定义了一个名为search的操作，它会根据提供的标签搜索文章。如果用户提供了多个空格分隔的标签，我们将使用这些标签构建一个复杂的Q对象查询，每个标签作为OR条件。最后，我们过滤出符合这些标签的文章，并返回它们的序列化数据。