django - DRF 指定所需的标题

Question

有没有办法为特定的 View 类指定所需的自定义 header ？在我正在使用的 API 上，一些信息是使用 header 传递的，如果缺少 header ，我需要返回 BAD_REQUEST 或类似信息。

我可以想到使用 mixins 实现它的方法，但不确定是否已经有合适的方法。

Answer 1

您可以通过重写 dispatch() 方法来实现，

<b>from django.core.exceptions import PermissionDenied</b>
from rest_framework import viewsets


class SampleViewset(viewsets.ModelViewSet):
    serializer_class = SampleSerializer
    queryset = SampleModel.objects.all()

    <b>def dispatch(self, request, *args, **kwargs):
        if "HTTP_CUSTOM_HEADER" not in request.META:
            raise PermissionDenied('not found')
        return super().dispatch(request, *args, **kwargs)</b>

使用 Python 的请求库的自定义 header 的用法示例

In [10]: wrong_header = {"custom-header-wrong":"soem value"}

In [11]: orginal_header = {"custom-header":"soem value"}

In [12]: requests.get(url=url,headers=wrong_header).status_code
Out[12]: 403

In [13]: requests.get(url=url,headers=orginal_header).status_code
Out[13]: 200