鍍金池/ 問(wèn)答/Python/ flask中如何自定義裝飾器和函數(shù)的參數(shù)

flask中如何自定義裝飾器和函數(shù)的參數(shù)

現(xiàn)在裝飾器是這么寫的

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

這么調(diào)用:

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

報(bào)錯(cuò):

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'

回答
編輯回答
櫻花霓

把視圖中的裝飾器位置調(diào)換一下位置試試。
寫成這樣:

@test
@app.route('/<name>')
def show_name(name):
    return name
2017年12月22日 12:48
編輯回答
只愛你

改成這樣解決:

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)

有其他方案的歡迎補(bǔ)充

2017年1月2日 05:21
編輯回答
凝雅
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
2017年12月28日 07:28