Question

所以我已经阅读了所有类似的问题并复制了他们写的内容，但我仍然遇到问题。所以我想要这样的东西

# Yes, I know django has one but I want to make my own
@rate_limit(seconds=10) 
myview(request, somearg, *args, **kwargs):
    # Return a response
...

def rate_limit(seconds=10):    
    def decorator(view):            
        def wrapper(request, *args, **kwargs):
            # Do some stuff
            return view(request, *args, **kwargs)       
        return wrapper
    return decorator

当我运行它时，我收到错误

decorator() got an unexpected keyword argument 'somearg'

所以我追加装饰器来接受args和kwargs并得到这个错误

# New decorator signature
def decorator(view, *args, **kwargs)

和错误

'function' object has no attribute 'status_code'

修改所以解决方案是使用。谢谢 Martijn Pieters ！

@rate_limit()

而不是

@rate_limit

Answer 1

您的首次尝试工作正常，但您可能忘记致电 rate_limit()装饰工厂了。

换句话说，如果您执行此操作，则会出现第一个错误：

@rate_limit
def myview(request, somearg, *args, **kwargs):

而不是：

@rate_limit(seconds=10)
def myview(request, somearg, *args, **kwargs):

你也真的想对Django中使用的装饰器使用functools.wraps()，特别是如果你想将它与其他Django装饰器混合使用，例如csrf_exempt：

from functools import wraps

def rate_limit(seconds=10):
    def decorator(view):
        @wraps(view)
        def wrapper(request, *args, **kwargs):
            # Do some stuff
            return view(request, *args, **kwargs)
        return wrapper
    return decorator

这可确保将待包装函数上设置的任何属性正确复制到包装器。

制作自定义django视图装饰器时遇到问题（使用args）

1 个答案: