如何配置Django REST Framework使搜索仅返回第一条匹配记录？

我已经搭建好Django REST Framework，当前可以正常显示记录列表，搜索功能也能正常运行。现在需要修改为：执行搜索时仅返回第一条匹配记录，而非所有匹配结果。以下是原代码和我尝试但未生效的代码，请帮忙指导。

原工作代码 #

class PhoneNumberModelViewSet(viewsets.ModelViewSet):
    queryset = PhoneNumbers.objects.all()
    serializer_class = PhoneNumbersSerializer
    filter_backends = [SearchFilter]
    search_fields = ['area_code']

尝试但未生效的代码 #

class PhoneNumberModelViewSet(viewsets.ModelViewSet):
    queryset = PhoneNumbers.objects.all()
    serializer_class = PhoneNumbersSerializer
    filter_backends = [SearchFilter]
    search_fields = ['area_code']

    def get(self):
        queryset = super().get()
        area_code = self.request.query_params.get('area_code')
        if area_code:
            try:
                queryset = queryset.filter(area_code=area_code)[0:1]
            except IndexError:
                queryset = PhoneNumbers.objects.none()
        return queryset

正确修改方案 #

你问题的核心是没找对DRF ViewSet中处理查询集的正确方法——应该重写get_queryset()，而不是自定义get()方法。另外不需要重复过滤area_code，SearchFilter已经完成了搜索参数的处理，只需在过滤后的结果上做切片即可。

方案一：保留SearchFilter（使用默认search参数） #

from rest_framework import viewsets
from rest_framework.filters import SearchFilter

class PhoneNumberModelViewSet(viewsets.ModelViewSet):
    queryset = PhoneNumbers.objects.all()
    serializer_class = PhoneNumbersSerializer
    filter_backends = [SearchFilter]
    search_fields = ['area_code']

    def get_queryset(self):
        # 获取经过SearchFilter过滤后的原始查询集
        queryset = super().get_queryset()
        # 检查是否存在搜索请求（SearchFilter默认用`search`参数）
        search_query = self.request.query_params.get('search')
        if search_query:
            # 仅保留第一条匹配记录
            queryset = queryset[:1]
        return queryset

方案二：直接用area_code作为查询参数（无需SearchFilter） #

如果不想用DRF默认的search参数，想直接通过?area_code=xxx发起查询，可去掉SearchFilter，直接在get_queryset()中处理：

class PhoneNumberModelViewSet(viewsets.ModelViewSet):
    queryset = PhoneNumbers.objects.all()
    serializer_class = PhoneNumbersSerializer

    def get_queryset(self):
        queryset = super().get_queryset()
        area_code = self.request.query_params.get('area_code')
        if area_code:
            # 过滤并仅取第一条记录
            queryset = queryset.filter(area_code=area_code)[:1]
        return queryset

关键说明 #

1. DRF的ModelViewSet中，查询集的统一入口是get_queryset()，所有过滤逻辑都应该在这里实现，get()是请求处理方法，内部会自动调用get_queryset()。
2. 使用queryset[:1]可以安全获取第一条记录，即使没有匹配结果也会返回空查询集，无需额外捕获IndexError。

内容的提问来源于stack exchange，提问作者Muqeet Mustafa