如何实现Django API按URL参数过滤DeviceGroup查询集？

我来帮你搞定这个需求！你要实现的是通过/api/v1/device-groups/?customer=<customer_uuid>这个带查询参数的接口，返回对应客户的设备组数据。之前的问题出在对参数获取方式的误解上，我给你一步步捋清楚：

先分析你之前的错误 #

1. 一开始写死过滤条件的方式肯定不灵活，重写get_queryset是正确的方向，但你第一次尝试用self.kwargs['customer_uuid']是错的——kwargs用来获取URL路径里的参数（比如/device-groups/<customer_uuid>/这种路径变量），而你需要的是URL后缀?后的查询参数，所以得从请求的查询参数里取。
2. 你后来改成self.request.GET['customer_uuid']已经接近正确，但直接用索引取值会在用户没传参数时抛出KeyError，最好加容错处理。

正确的实现代码 #

优化后的views.py代码如下，兼顾功能和鲁棒性：

class DeviceGroupViewSet(viewsets.ModelViewSet):
    serializer_class = DeviceGroupSerializer
    queryset = DeviceGroup.objects.all()  # 设置默认查询集

    def get_queryset(self):
        # 用get方法从查询参数里取customer值，避免没传参数时抛KeyError
        customer_uuid = self.request.GET.get('customer')
        if customer_uuid:
            # 有参数时，过滤对应客户的设备组
            return DeviceGroup.objects.filter(customer_uuid=customer_uuid)
        # 无参数时返回所有数据（你也可以改成返回空列表/抛出400错误，按需调整）
        return super().get_queryset()

关键细节说明 #

• 参数获取方式：request.GET.get('customer')对应你URL里的?customer=<uuid>参数名，要注意参数名和代码里的取值名一致（你之前写的是customer_uuid，但URL里用的是customer，这里要统一！）
• 容错处理：用get方法而不是直接索引取值，当用户没传customer参数时会返回None，我们可以做兼容处理（比如返回全量数据），避免接口直接报错。
• 序列化层兼容：你在DeviceGroupSerializer里用source='customer_uuid'关联CustomerSerializer的写法是正确的，序列化后会正常返回对应客户的信息，不需要修改。

现在测试一下：访问/api/v1/device-groups/?customer=你的客户UUID，就能看到该客户下的所有DeviceGroup数据了；如果不带参数，会返回全量设备组（可根据业务需求调整逻辑）。

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