从 URL 参数中过滤 Django ModelViewSet 列表

Question

您好，我有一个类似的模型：

class Appointment(models.Model):
    hospital = models.ForeignKey(Hospital, on_delete=models.CASCADE)
    patient = models.ForeignKey(Patient, on_delete=models.CASCADE)

我的视图如下：

class AppointmentViewSet(viewsets.ModelViewSet):
    queryset = Appointment.objects.all()
    serializer_class = AppointmentSerializer

在我的网址中：

router.register(r'appointments', AppointmentViewSet)

现在我想按一些患者 ID 过滤预约列表。此 id 应由请求者通过 url 提供。我正在考虑使用 kwargs 来捕捉它。但我不知道该怎么做。我知道我必须重写 list 方法。

def list(self, request, *args, **kwargs):
    # what do I write here? so that the queryset would be filtered by patient id sent through the url? 

如何自定义 url 和/或 view 以适应患者 ID 参数？我只想修改列表请求，所有其他操作（创建、详细信息、销毁）应由模型视图集的默认行为处理。

谢谢。

Answer 1

这是我最终的做法：

我添加了一个这样的 url 条目：

url(r'appointments/ofpatient/(?P<patient>\d+)', AppointmentViewSet.as_view({'get': 'list'})),

我可以从浏览器调用：

http://localhost:8000/appointments/ofpatient/6

在视野中：

def list(self, request, patient=None):
    if patient:
        patient = Patient.active.filter(id=patient)
        appts = Appointment.active.order_by('appt_time').filter(patient=patient)
        serializer = self.get_serializer(appts, many=True)
        return Response(serializer.data)
    else:
        appts = Appointment.active.order_by('appt_time')
        serializer = self.get_serializer(appts, many=True)
        return Response(serializer.data)

这样，/appointments url 也被保留了下来。

Answer 2

您可以定义一个自定义方法来处理带有患者 ID 的 url 的路由：

from rest_framework.decorators import list_route
rom rest_framework.response import Response

class AppointmentViewSet(viewsets.ModelViewSet):
    ...

    @list_route()
    def patient_appointments(self, request, id=None):
        serializer = self.get_serializer(queryset.filter(patient_id=id), many=True)
        return Response(serializer.data)

list_route 装饰器将您的方法标记为需要路由。

更新：

您可以手动将网址注册为：

url(r'^(?P<id>[0-9]+)/appointments/$', AppointmentViewSet.as_view({'get': 'patient_appointments'}), name='patient-appointments')

Answer 3

试试

if(self.request.get('pid')):
    pk = self.request.get('pid')
    all_appointment = Appointment.objects.filter(patient__pk=pk)

对于患者的 url 将是 appoinment/?pid=9

预约详情appoinment/9