净心诀---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()

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