python函数装饰器(修饰符)

前提知识点：

1. 可以将一个函数赋值给一个变量
2. 可以在函数中定义另外的函数，那么在执行外层函数时候，里面的函数也会执行
3. 从函数中返回函数
4. 将函数作为参数传给另一个函数

具体实例见：https://www.runoob.com/w3cnote/python-func-decorators.html

装饰器的作用

python函数修饰器@的作用是为现有函数增加额外的功能，常用于插入日志、性能测试、事务处理等等。

装饰器的处理逻辑

当解释器读到@装饰符，会先解析@下一行的内容，把下一行的函数或者类作为@后边的函数的参数，执行被装饰的函数。

举例说明

例子1：被修饰函数不带参数

def log(func):
    def wrapper():
        print('log开始 ...')
        func()
        print('log结束 ...')
    return wrapper
    
@log
def test():
    print('test ..')

test()

运行结果：

log开始 ...
test ..
log结束 ...

例子2：被修饰函数带参数

from functools import wraps

def log(func):
    @wraps(func)
    def wrapper(*args,**kwargs):
        print('log开始 ...',func.__name__)
        ret = func(*args,**kwargs)
        print('log结束 ...')
        return ret
    return wrapper
    
@log
def test1(s):
    print('test1 ..', s)
    return s

@log
def test2(s1, s2):
    print('test2 ..', s1, s2)
    return s1 + s2


test1('a')
test2('a','bc')

运行结果：

log开始 ... test1
test1 .. a
log结束 ...
log开始 ... test2
test2 .. a bc
log结束 ...

在函数定义中我们返回函数，为什么是函数名，而不是函数名()。为什么那样？这是因为当你把一对小括号放在后面，这个函数就会执行；然而如果你不放括号在它后面，那它可以被到处传递，并且可以赋值给别的变量而不去执行它。

例子3：修饰符带参数，需要比上面例子多一层包装

from functools import wraps

def log(arg):    
    def _log(func):
        @wraps(func)
        def wrapper(*args,**kwargs):
            print('log开始 ...',func.__name__, arg)            
            ret = func(*args,**kwargs)
            print('log结束 ...')
            return ret
        return wrapper
    return _log
 
@log('module1')
def test1(s):
    print('test1 ..', s)
    return s

@log('module1')
def test2(s1, s2):
    print('test2 ..', s1, s2)
    return s1 + s2


test1('a')
test2('a','bc')

运行结果：

log开始 ... test1 module1
test1 .. a
log结束 ...
log开始 ... test2 module1
test2 .. a bc
log结束 ...