python3--装饰器函数

装饰器的本质是一个闭包函数。

装饰器的功能：在不修改原函数及其调用方式的情况下对原函数功能进行扩展。

开放封闭原则：
开放：对扩展是开放的
封闭：对修改是封闭的

1.带参数及返回值的装饰器

def decrator(func):  # 装饰器函数，参数为被装饰函数的函数名
    def inner(*args,**kwargs):  # 任意个参数
        '''装饰前执行的代码'''
        ret = func(*args,**kwargs)  # 被装饰函数
        '''装饰后执行的代码'''
        print(ret)
        return ret
    return inner
@decrator    # @装饰器函数名，作用相当于func1 = decrator(func1)
def func1(a):
    '''func1'''
    return a
@decrator    # 作用相当于func2 = decrator(func2)
def func2(a,b):
    return a,b
func1('hello world')   # hello world
print(func1.__name__)  # inner，此处获得的是装饰器内的函数名，而非调用的函数的函数名
print(func1.__doc__)  # 装饰前执行的代码，此处获得的是装饰器内的函数的注释，而非调用的函数的注释
func2('good','night')  # ('good', 'night')

2.完美的带参数及返回值的装饰器

from functools import wraps
def decrator(func):  # 装饰器函数，参数为被装饰函数的函数名
    @wraps(func)  # 加在最内层函数的正上方
    def inner(*args,**kwargs):  # 任意个参数
        '''装饰前执行的代码'''
        ret = func(*args,**kwargs)  # 被装饰函数
        '''装饰后执行的代码'''
        print(ret)
        return ret
    return inner
@decrator    # @装饰器函数名，作用相当于func1 = decrator(func1)
def func1(a):
    '''func1'''
    return a
@decrator    # 作用相当于func2 = decrator(func2)
def func2(a,b):
    return a,b
func1('hello world')   # hello world
print(func1.__name__)  # func1  调用函数的函数名
print(func1.__doc__)  # func1  调用函数的注释
func2('good','night')  # ('good', 'night')

3.可控制的完美的带参数及返回值的装饰器

from functools import wraps
def decrator_out(flag):
    def decrator(func):  # 装饰器函数，参数为被装饰函数的函数名
        @wraps(func)
        def inner(*args,**kwargs):  # 任意个参数
            if flag:
                '''装饰前执行的代码'''
                print('123')
            ret = func(*args,**kwargs)  # 被装饰函数
            if flag:
                '''装饰后执行的代码'''
                print('456')
            print(ret)
            return ret
        return inner
    return decrator
@decrator_out(True)
def func1(a):
    '''func1'''
    return a
@decrator_out(False)
def func2(a,b):
    return a,b
func1('hello world')
# 123
# 456
# hello world
print(func1.__name__)  # func1
print(func1.__doc__)  # func1
func2('good','night')  # ('good', 'night')  没有执行装饰内容

4.多个装饰器装饰同一个函数（类似俄罗斯套娃）

def wrapper1(f):                     # 1---------3 f=func
    def inner1(*args, **kwargs):     # 7
        print('start inner1')        # 8 start inner1
        ret = f(*args, **kwargs)     # 9 f=func  I'm func
        print('end inner1')          # 10 end inner1
        return ret
    return inner1
def wrapper2(f):                     # 2----------4 f=inner1
    def inner2(*args, **kwargs):
        print('start inner2')        # 5 start inner2
        ret = f(*args, **kwargs)     # 6 f=inner1
        print('end inner2')          # 11 end inner2
        return ret
    return inner2
@wrapper2  # 4.func = wrapper2(inner1)--> inner2
@wrapper1  # 3.func = wrapper1(func) --> inner1
def func():
    print("I'm func")

func()
# start inner2
# start inner1
# I'm func
# end inner1
# end inner2

想知道程序的执行顺序，debug就可以啦。