python 装饰器

此文为个人总结的廖雪峰学习笔记，详细知识见廖雪峰。

---

装饰器用途：
要增强now()函数的功能，又不希望修改now()函数的定义，
这种在代码运行期间动态增加功能的方式，称之为“装饰器”（Decorator）

本质上，装饰器是一个接受函数作为参数的返回函数。

装饰器的常见定义方式（两种）：
①

def decorator(func):
    def wrapper(*arg,**kw):
        print('call %s():' % func.__name__)
        return func(*arg,**kw)
    return wrapper

@log()
def f():
    print(2)

这种定义方式实现的功能较为简单，不需额外自定义log文本，可以在不改变原函数定义前提下，给原函数增加执行前输出日志的功能。
调用方式是，直接输入原函数f()即可。

②

def log(text):
    def decorator(func):
        def wrapper(*arg,**kw):
            print(text)
            return func()
        return wrapper
    return decorator


@log('abc')
def f():
    print(2)

这种定义方式可以自定义log中的text文本，这时装饰器需要再套一层，见上面代码。调用也是直接敲f()即可。

**上两种都可以看成原函数变成了wrapper(*arg,**kw)。

---

有人有疑问，下面也可实现自定义log中的text文本功能，为啥还需嵌套一层?

def log(text):
    def wrapper(fun):
        print('%s %s():' % (text, fun.__name__))
        return (fun)
    return wrapper


@log('excute')
def now():
    print('2020-02-04')


now()

虽然也可以输出：

excute now():
2020-02-04

但是，这样定义的now()不能带参数，因为参数传不进wrapper里，所以若要带参数，还得按上面两种常见定义。