python - 向装饰器添加参数

Question

我有这个装饰器，如果 share 参数为 True(由中间件处理)，我不想执行 View 时用于装饰 django View

class no_share(object):
    def __init__(self, view):
        self.view = view

    def __call__(self, request, *args, **kwargs):
        """Don't let them in if it's shared"""

        if kwargs.get('shared', True):
            from django.http import Http404
            raise Http404('not availiable for sharing')

        return self.view(request, *args, **kwargs)

目前它是这样工作的:

@no_share
def prefs(request, [...])

但我想稍微扩展一下功能，这样它就可以像这样工作:

@no_share('prefs')
def prefs(request, [...])

我的问题是如何修改这个装饰器类以便它接受额外的参数？

Answer 1

希望this Bruce Eckel 的文章有所帮助。

更新:根据这篇文章，您的代码将如下所示:

class no_share(object):
    def __init__(self, arg1):
        self.arg1 = arg1

    def __call__(self, f):
        """Don't let them in if it's shared"""

        # Do something with the argument passed to the decorator.
        print 'Decorator arguments:', self.arg1

        def wrapped_f(request, *args, **kwargs):
            if kwargs.get('shared', True):
                from django.http import Http404
                raise Http404('not availiable for sharing')
            f(request, *args, **kwargs)            
        return wrapped_f

根据需要使用:

@no_share('prefs')
def prefs(request, [...])