将函数传递给Cherrypy中的装饰器

时间:2020-05-09 12:02:51

标签: python cherrypy

如何将index()传递给称为uppercase()的装饰器函数。我希望将从index()传递的值大写。

我收到此错误:

500 Internal Server Error
The server encountered an unexpected condition which prevented it from fulfilling the request.

Traceback (most recent call last):
  File "/home/user/.local/lib/python3.6/site-packages/cherrypy/_cprequest.py", line 638, in respond
    self._do_respond(path_info)
  File "/home/user/.local/lib/python3.6/site-packages/cherrypy/_cprequest.py", line 701, in _do_respond
    self.hooks.run('before_finalize')
  File "/home/user/.local/lib/python3.6/site-packages/cherrypy/_cprequest.py", line 95, in run
    self.run_hooks(iter(sorted(self[point])))
  File "/home/user/.local/lib/python3.6/site-packages/cherrypy/_cprequest.py", line 117, in run_hooks
    hook()
  File "/home/user/.local/lib/python3.6/site-packages/cherrypy/_cprequest.py", line 65, in __call__
    return self.callback(**self.kwargs)
TypeError: uppercase() missing 1 required positional argument: 'func'

import cherrypy
from cherrypy import tools

@cherrypy.tools.register('before_finalize')
def uppercase(func):
    def wrapper():
        original_result = func()
        modified_result = original_result.upper()
        return modified_result
    return wrapper

class HelloWorld(object):
    @cherrypy.expose
    @cherrypy.tools.uppercase()
    def index(self):
        return 'Hello!'


if __name__ == '__main__':
    cherrypy.tools.uppercase = cherrypy.Tool('before_finalize', uppercase)
    cherrypy.quickstart(HelloWorld())

1 个答案:

答案 0 :(得分:2)

您应该像这样使用装饰器:

@cherrypy.tools.uppercase
def index(self):
    return 'Hello!'

请注意,()之后缺少uppercase-装饰器隐式接受第一个参数(装饰的函数),并且不需要括号。

编辑:

根据the documentation,可以使用cherrypy.tools.register装饰器或cherrypy.Tool构造器定义工具。在代码中,您两次定义了uppercase工具。

但是,在您的情况下,uppercase装饰器不需要定义为Tool,因为它不需要每次都作为钩子运行(before_finalize)。

因此,将其用作普通的Python装饰器会更好,就像这样:

from functools import wraps

# just a plain Python decorator
def uppercase(func):
    @wraps(func) #preserve function attributes, such as its name
    def wrapper(*args):
        original_result = func(*args)
        modified_result = original_result.upper()
        return modified_result
    return wrapper

class HelloWorld(object):
    @cherrypy.expose
    @uppercase # decorated once, the exposed function is now uppercase(index) 
    def index(self):
        return 'Hello!'