Django Model ViewSet 多对多字段搜索进阶指南

2023-02-07 10:21:55

利用 Django Model ViewSet 精通多对多字段搜索

解锁后端开发的强大工具

在当今数字化的世界中，后端开发人员必不可少的一项技能是熟练运用后端框架。在这方面，Django Model ViewSet 脱颖而出，为开发者提供了一套强大的工具和简洁的架构，以构建可扩展且高效的后端应用程序。对于多对多字段，充分理解如何在 ViewSets 中实现这一功能至关重要。

多对多字段的意义

多对多字段在数据库中表示两个表之间的多对多关系。例如，在一个学生管理系统中，一个学生可能参加多个课程，而一个课程也可能有多个学生参加。在这种情况下，学生表和课程表之间就存在多对多的关系。

通过过滤器 API 实现搜索

Django 提供了过滤器 API，允许开发者过滤查询集，仅返回满足特定条件的对象。对于多对多字段，我们可以利用 in 过滤器或 contains 过滤器。

in 过滤器的应用

in 过滤器用于检查查询集中的对象是否包含在指定值列表中。例如，若要过滤参加课程 ID 为 1 的学生，可以使用以下代码：

from django.db.models import Q
from rest_framework import viewsets

class StudentViewSet(viewsets.ModelViewSet):
    queryset = Student.objects.all()

    def get_queryset(self):
        queryset = super().get_queryset()
        course_id = self.request.query_params.get('course_id', None)
        if course_id:
            queryset = queryset.filter(Q(courses__in=[course_id]))
        return queryset

contains 过滤器的应用

contains 过滤器用于检查查询集中的对象是否包含指定值。例如，若要过滤姓名包含 "John" 的学生，可以使用以下代码：

from django.db.models import Q
from rest_framework import viewsets

class StudentViewSet(viewsets.ModelViewSet):
    queryset = Student.objects.all()

    def get_queryset(self):
        queryset = super().get_queryset()
        name = self.request.query_params.get('name', None)
        if name:
            queryset = queryset.filter(Q(name__contains=name))
        return queryset