净心诀---python3装饰器

python3装饰器

装饰器作用

简单理解：可以为已有函数添加额外功能

例：

已有2个函数如下

def MyFunc1():
    print("This is a print function1")


def MyFunc2():
    print("This is a print function2")

需求：现想要在函数执行前打印类似（开始使用XXX功能）

不用装饰器方式：

def MyFunc1():
    print("开始使用MyFunc1功能")
    print("This is a print function1")


def MyFunc2():
    print("开始使用MyFunc2功能")
    print("This is a print function2")

调用函数时输出：

思考：如果有很多很多的函数，那么每个函数中都要修改(而且这需要在已有函数内部去修改，部分情况下是不允许修改已有函数的），这个方式显然不可靠

采取装饰器的方式做

装饰器模板（写法）

def ShowFunction(func):
    def inner(*args, **kwargs):
        # 这里写函数执行前的功能
        ret = func(*args, **kwargs)
        return ret        
    return inner

利用上述装饰器模板，来写一个上述需求的装饰器

# 这是一个装饰器
def ShowFunction(func):
    def inner(*args, **kwargs):
        name = func.__name__
        print("开始使用%s功能" %name)
        ret = func(*args, **kwargs)
        return ret        
    return inner


@ShowFunction    # @加函数名  这种形式就是使用某个装饰器的写法
def MyFunc1():
    print("This is a print function1")

@ShowFunction
def MyFunc2():
    print("This is a print function2")


MyFunc1()
MyFunc2()

效果如下：

 现在基本上就能满足上述需求

思考：有时候不是所有的装饰器都是永久装饰函数的，在某些情况下我们会希望装饰器不生效，此时需要用到带有参数的装饰器

此时我们一般会在装饰器外部在套一个函数，来接受一个Flag，在内部用这个Flag来判断是否生效装饰器

# 现需要在调用每个函数前说明这是哪个函数，用到带装饰装饰器(可控制是否使装饰器生效)
from functools import wraps

# 这是一个装饰器
Flag = 1
def FuncWrap(Flag):
    def ShowFunction(func):
        @wraps(func)    #使外部函数调用函数属性时，不被装饰器影响
        def inner(*args, **kwargs):
            if Flag:
                name = func.__name__
                print("开始使用%s功能" %(name))
                ret = func(*args, **kwargs)
            else:
                ret = func(*args, **kwargs)    
            return ret        
        return inner        
    return ShowFunction



@FuncWrap(Flag)    # 等于写法-->MyFunc1 = ShowFunction(MyFunc1)
def MyFunc1():
    print("This is a print function1")

@FuncWrap(Flag)
def MyFunc2():
    print("This is a print function2")

# print(MyFunc1.__name__)
MyFunc1()
MyFunc2()

我们可以修改Flag的值1(生效)或0(不生效)

扩展：多个装饰器共同装饰一个函数

# 这是一个装饰器1
def Wrap1(func):
    def inner(*args, **kwargs):
        print("start装饰器1")
        ret = func(*args, **kwargs)
        print("end装饰器1")
        return ret        
    return inner

# 这是一个装饰器2
def Wrap2(func):
    def inner(*args, **kwargs):
        print("start装饰器2")
        ret = func(*args, **kwargs)
        print("end装饰器2")
        return ret        
    return inner

@Wrap1
@Wrap2
def MyFunc1():
    print("This is a print function1")

MyFunc1()

关于多装饰器现象(记住即可)：