制作自定义 django 视图装饰器（带参数）时遇到问题

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

# 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'

edit:所以解决办法就是使用。谢谢马丁·彼得斯!

@rate_limit()

代替

@rate_limit

您的第一次尝试效果很好，但您可能忘记了call the rate_limit()装饰工厂。

换句话说，如果您这样做，就会出现第一个错误：

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

代替：

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

您也确实想使用functools.wraps()关于 Django 中使用的装饰器，特别是如果您想将其与其他 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

这确保了要包装的函数上设置的任何属性都被正确复制到包装器。