Django自定义中间件未添加自定义头

2024-10-03 02:42:37 发布

您现在位置:Python中文网/ 问答频道 /正文

因此,我试图为Django应用程序中的每个请求添加一个自定义头,我遵循了this问题,我的设置如下所示:

中间件.py

from django.utils.deprecation import MiddlewareMixin


class ReverseProxyLocalMiddleware(MiddlewareMixin):
    def process_request(self, request):
        request.META['User-Id'] = 1

设置.py

...
MIDDLEWARE = [
    'django.middleware.security.SecurityMiddleware',
    'django.contrib.sessions.middleware.SessionMiddleware',
    'django.middleware.common.CommonMiddleware',
    # 'django.middleware.csrf.CsrfViewMiddleware',
    'django.contrib.auth.middleware.AuthenticationMiddleware',
    'django.contrib.messages.middleware.MessageMiddleware',
    'django.middleware.clickjacking.XFrameOptionsMiddleware',
    'debug_toolbar.middleware.DebugToolbarMiddleware',
    'myapp.middleware.ReverseProxyLocalMiddleware',
]

视图.py

class UserViewSet(
    viewsets.GenericViewSet, mixins.ListModelMixin, mixins.RetrieveModelMixin
):
    queryset = models.User.objects.all()
    serializer_class = serializers.UserSerializer

    def get_queryset(self):
        return models.User.objects.active(dt=timezone.now())

    def list(self, request):
        user_id = request.META['User-Id']
        ...

但是,每当我在测试中尝试访问新头时,我会得到:

    def list(self, request):
>       user_id = request.META['User-Id']
E       KeyError: 'User-Id'

有人知道我做错了什么吗?你知道吗


Tags: djangopyselfidrequestdefcontribmixins
1条回答
网友
1楼 · 发布于 2024-10-03 02:42:37

如果不是HTTP_*_*格式,Django会删除头,因此中间件必须如下所示:

from django.utils.deprecation import MiddlewareMixin


class ReverseProxyLocalMiddleware(MiddlewareMixin):
    def process_request(self, request, user_id=1):
        request.META['HTTP_USER_ID'] = user_id

相关问题 更多 >