django - 装饰器中的 Django 视图参数

Question

我有一个函数，我不想用一个参数把它变成一个装饰器，请任何人帮忙

def get_permission(request,permission_level):
    if has_permission(request,permission_level):#this is another function
        #my work will be done here
    else:
        raise Http404

我只是想在我看来像这样使用它

@get_permission(permission_level)
def a_view(request)

有人可以帮忙吗？谢谢

Answer 1

看看django/contrib/auth/decorators.py，类似

from functools import wraps
from django.utils.decorators import available_attrs

def get_permission(permission_level):
    def decorator(func):
        @wraps(func, assigned=available_attrs(func))
        def _wrapped(request, *args, **kwargs):
            if has_permission(request, permission_level): 
                'special logic goes here...'
                return func(request, *args, **kwargs)
            else:
                raise Http404

检查如何制作功能装饰器链？和PEP 318用于编写装饰器。

或者你可以直接在django/contrib/auth/decorators.pyuser_passes_test中使用or ，如果装饰器只是检查权限来确定返回的结果。permission_required