在基于Django类的视图中检查视图方法参数名称

Question

我在我的Django项目中创建了一个装饰器,用于将参数值注入到装饰方法的参数中.

我这样做是inspect.getargspec为了检查方法中存在哪些参数并将它们放入kwargs.否则,由于方法中的参数数量不正确,我会收到错误.

虽然这在单个视图方法中正常工作,但在Django基于类的视图方面却失败了.

我相信这可能是因为装饰器@method_decorator在类级别应用于dispatch方法而不是个人get和post方法.

我是一个蟒蛇新手,可能会忽略一些明显的东西.

有没有更好的方法来做我正在做的事情？是否可以在基于类的视图中获取方法参数名称？

我正在使用Python 2.7和Django 1.11

装饰者

def need_jwt_verification(decorated_function):
    @wraps(decorated_function)
    def decorator(*args, **kwargs):
        request = args[0]
        if not isinstance(request, HttpRequest):
            raise RuntimeError(
                "This decorator can only work with django view methods accepting a HTTPRequest as the first parameter")

        if AUTHORIZATION_HEADER_NAME not in request.META:
            return HttpResponse("Missing authentication header", status=401)

        jwt_token = request.META[AUTHORIZATION_HEADER_NAME].replace(BEARER_METHOD_TEXT, "")

        try:
            decoded_payload = jwt_service.verify_token(jwt_token)

            parameter_names = inspect.getargspec(decorated_function).args

            if "phone_number" in parameter_names or "phone_number" in parameter_names:
                kwargs["phone_number"] = decoded_payload["phone"]
            if "user_id" in parameter_names:
                kwargs["user_id"] = decoded_payload["user_id"]
            if "email" in parameter_names:
                kwargs["email"] = decoded_payload["email"]

            return decorated_function(*args, **kwargs)
        except JWTError as e:
            return HttpResponse("Incorrect or expired authentication header", status=401)

    return decorator

基于类的视图

@method_decorator([csrf_exempt, need_jwt_verification], name="dispatch")
class EMController(View):
    def get(self, request, phone_number, event_id):
        data = get_data()

        return JsonResponse(data, safe=False)

    def post(self, request, phone_number, event_id):


        return JsonResponse("Operation successful", safe=False)

编辑:

在方法级别应用装饰器的明显解决方案不适用于Django基于类的视图.您需要在url配置中应用装饰器或将装饰器应用于dispatch方法.

编辑:我发布了与我正在探索的解决方法相关的代码,将参数名称作为参数传递给装饰器.

Answer 1

我发现这篇文章:函数装饰器在Django中基于类的视图上有参数

这可能会为您的问题提供答案:

如果要传递带参数的装饰器,您只需:

• 评估decorator-creator函数中的参数.

• 将评估值传递给@method_decorator.

上面提到的和在考虑中的链接答案中提供的代码,你应该:

injectables=[inject_1, inject_2, ..., inject_n]
decorators = [csrf_exempt, need_jwt_verification(injectables)]

@method_decorator(decorators, name="dispatch")
class EMController(View):
    ...

由于遗留原因,我在此留下我以前的错误答案,不要在家里(或任何地方,在django,尝试!!)尝试这个!

如果我们观察"装饰类"文档,我们可以看到以下内容:

或者,更简洁地说,您可以改为装饰类,并将要装饰的方法的名称作为关键字参数名称传递:

所以你必须改变你的name参数@method_decorator以匹配将适用于的方法:

decorators = [csrf_exempt, need_jwt_verification(injectables=[])]

@method_decorator(decorators, name='get')
@method_decorator(decorators, name='post')
class EMController(View):

就个人而言,我更喜欢将装饰器置于他们将应用的特定方法之上:

class EMController(View):
    @method_decorator(decorators)
    def get(self, request, phone_number, event_id):
        ...

    @method_decorator(decorators)    
    def post(self, request, phone_number, event_id):
        ...

Answer 2

在图书馆目前的状况下，我想要的似乎是不可能的。这就是我最终选择的。

            parameter_names = inspect.getargspec(decorated_function).args

            if "phone_number" in parameter_names or "phone_number" in injectables:
                kwargs["phone_number"] = decoded_payload["phone"]
            if "user_id" in parameter_names:
                kwargs["user_id"] = decoded_payload["user_id"]
            if "email" in parameter_names:
                kwargs["email"] = decoded_payload["email"]

            request.__setattr__("JWT", {})
            request.JWT["phone_number"] = decoded_payload["phone"]
            request.JWT["user_id"] = decoded_payload["user_id"]
            request.JWT["email"] = decoded_payload["email"]

该装饰器将按预期自动填充基于方法的视图中的参数。

但它还会JWT向请求对象注入一个属性，供基于类的视图使用。喜欢request.GET和request.POST。