向装饰器添加参数

Question

我有这个装饰器，如果share 参数是True（由中间件处理），我不想执行视图时用于装饰 django 视图

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

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

Answer 2

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

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

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, [...])

Answer 3

由于您似乎在某处弄错了，这里有一个更完整的示例，可以帮助您了解自己做错了什么。将其用作插件应该可以。

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

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

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

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

Answer 4

Li0liQ 提到的 Bruce Eckel article 应该有助于解决这个问题。带参数和不带参数的装饰器的行为略有不同。最大的区别在于，当您传递参数时，__call__ 方法在 __init__ 上被调用一次，并且它应该返回一个函数，该函数将在调用装饰函数时被调用。当没有参数时，每次调用装饰函数时都会调用 __call__ 方法。

这对您意味着什么？为 @no_arg_decorator 调用 __init__ 和 __call__ 的方式与为 @decorator('with','args') 调用的方式不同。

这里有两个装饰器可能会为您解决问题。只要您始终将其与括号一起使用，您就可以只使用 @no_share_on(...) 装饰器。

def sharing_check(view, attr_name, request, *args, **kwargs):
    if kwargs.get(attr_name, True):
        from django.http import Http404
        raise Http404('not availiable for sharing')

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

class no_share(object):
    """A decorator w/o arguments.  Usage:
    @no_share
    def f(request):
        ...
    """
    def __init__(self, view):
        self.view = view

    def __call__(self, request, *args, **kwargs):
        return sharing_check(self.view, 'sharing', request, *args, **kwargs)

class no_share_on(object):
    """A decorator w/ arguments.  Usage:
    @no_share_on('something')
    def f(request):
        ...
    --OR--
    @no_share_on()
    def g(request):
        ...
    """
    def __init__(self, attr_name='sharing'):
        self.attr_name = attr_name

    def  __call__(self, view):
        def wrapper_f(request, *args, **kwargs):
            return sharing_check(view, self.attr_name, request, *args, **kwargs)

Answer 5

我认为关闭在这里可能有效。

def no_share(attr):
    def _no_share(decorated):
        def func(self, request, *args, **kwargs):
            """Don't let them in if it's shared"""

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

            return decorated(request, *args, **kwargs)
        return func
    return _no_share

Answer 6

我知道这有点晚了.. 但我没有看到任何关于这种做事方式的提及（可能是因为在提出问题时它不存在），但是，为了完整起见 我发现看看 Django 自己是如何实现这样的东西很有用。看看：

django.views.decorators.http.require_http_methods https://github.com/django/django/blob/master/django/views/decorators/http.py

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

def require_http_methods(request_method_list):
    """
    Decorator to make a view only accept particular request methods. Usage::

    @require_http_methods(["GET", "POST"])
    def my_view(request):
    # I can assume now that only GET or POST requests make it this far
    # ...

    Note that request methods should be in uppercase.
    """
    def decorator(func):
        @wraps(func, assigned=available_attrs(func))
        def inner(request, *args, **kwargs):
            # .. do stuff here

            return func(request, *args, **kwargs)
        return inner
    return decorator