flask中如何自定义装饰器和函数的参数

Question

现在装饰器是这么写的

def test(param=None):
    def deco(func):
        def wrapper(*args,**kwargs):
            return func(*args,**kwargs)
        return wrapper
    return deco

这么调用：

@app.route('/<name>')
@test
def show_name(name):
    return name

报错：

Traceback (most recent call last):
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/gunicorn/workers/sync.py", line 135, in handle
    self.handle_request(listener, req, client, addr)
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/gunicorn/workers/sync.py", line 176, in handle_request
    respiter = self.wsgi(environ, resp.start_response)
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 2309, in __call__
    return self.wsgi_app(environ, start_response)
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 2295, in wsgi_app
    response = self.handle_exception(e)
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 1741, in handle_exception
    reraise(exc_type, exc_value, tb)
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/_compat.py", line 35, in reraise
    raise value
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 2292, in wsgi_app
    response = self.full_dispatch_request()
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 1815, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 1718, in handle_user_exception
    reraise(exc_type, exc_value, tb)
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/_compat.py", line 35, in reraise
    raise value
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 1813, in full_dispatch_request
    rv = self.dispatch_request()
  File "/Users/cevin/Documents/projects/bit/venv/lib/python3.7/site-packages/flask/app.py", line 1799, in dispatch_request
    return self.view_functions[rule.endpoint](**req.view_args)
TypeError: deco() got an unexpected keyword argument 'name'

Answer 1

改成这样解决：

def test(func=None, param=None):
    def deco(func):
        def wrapper(*args,**kwargs):
            return func(*args,**kwargs)
        wrapper.__name__ = func.__name__
        return wrapper
    return deco if not func else deco(func)

有其他方案的欢迎补充

Answer 2

把视图中的装饰器位置调换一下位置试试。
写成这样：

@test
@app.route('/<name>')
def show_name(name):
    return name

Answer 3

def test(param=None):
    def deco(func):
        def wrapper(*args,**kwargs):
            return func(*args,**kwargs)
        return wrapper
    return deco
    

@app.route('/<name>')
@test(param='xxx')
def show_name(name):
    return name