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)

  1. 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())

你可能感兴趣的:(Python装饰器)