Python装饰器

在写模块时，由于可以把很多细节封装起来，很喜欢用到”装饰器（decorator）“这个玩意儿，但关于它的用法都是从网上抄的，每次用时，特别是写带参装饰器，都要去参考别人的”模板“。
今天干脆来探个究竟。

Hello

这里写了一个装饰器”upper“，将返回值转成大写

def upper(f):
    def func():
        resp = f()
        return resp.upper()
    return func

@upper
def hello():
    return 'hello, world'

if __name__ == '__main__':
    print(hello())

输出：

HELLO, WORLD

实际上，装饰器就是一个语法糖，它的真面目是：

def upper(f):
    def func():
        resp = f()
        return resp.upper()
    return func

def hello():
    return 'hello, world'

hello = upper(hello) # 装饰器的真面目

if __name__ == '__main__':
    print(hello())

多个装饰器

这里又写了一个装饰器”tail“，用于给函数的返回值添加一个小尾巴

def upper(f):
    def func():
        resp = f()
        return resp.upper()
    return func

def tail(f):
    def func():
        resp = f()
        return resp + ',hahaha~'
    return func

@tail
@upper
def hello():
    return 'hello, world'

if __name__ == '__main__':
    print(hello())

输出：

HELLO, WORLD,hahaha~

将两个装饰器的位置交换后，输出结果是不一样的，上述代码中，先进行了”upper“操作，再进行的”tail“操作，所以小尾巴并没有转成大写。
按照上一节中的示例，不难看出多个装饰器的原理：

hello = tail(upper(hello))

带参装饰器

如果想为函数添加自定义的小尾巴，就得需要带参装饰器了：

def tail(s):
    def warpper(f):
        def func():
            resp = f()
            return resp + s
        return func
    return warpper

@tail(', wow~')
def hello():
    return 'hello, world'

if __name__ == '__main__':
    print(hello())

按照装饰器的原理，将语法糖展开，它的真面目是这样的：

hello = tail(', wow~')(hello)

一步一步展开：

1. tail(', wow~') 执行后，返回warpper函数

hello = warpper(hello)

2. warpper(hello)执行后，返回了它的真面目，func，并赋值给了hello

hello = func

装饰带参函数

上述例子中，均是”hello world“，那如果想”hello，baby“呢，这就得hello函数带参数了
又把上面的代码改了改：

def tail(s):
    def warpper(f):
        def func(p):
            resp = f(p)
            return resp + s
        return func
    return warpper

@tail(', wow~')
def hello(p):
    return 'hello, {}'.format(p)

if __name__ == '__main__':
    print(hello('baby'))

输出：

hello, baby, wow~

上例中，只是简单地将func的声明改成了和hello的声明一致，
但这也只能用于明确知道被装饰函数的声明，不具有通用性。
例如，若将上例中的”tail“用于装饰下面不带参的goodnight函数，这就报错啦：

@tail(', wow~')
def goodnight():
    return 'goodnight'

输出：

TypeError: func() missing 1 required positional argument: 'p'

为了解决这个问题，又修改下代码，改用”*args, **kwargs“来传递参数：

def tail(s):
    def warpper(f):
        def func(*args, **kwargs):
            resp = f(*args, **kwargs)
            return resp + s
        return func
    return warpper

@tail(', wow~')
def goodnight():
    return 'goodnight'

if __name__ == '__main__':
    print(goodnight())