Python自定义装饰器

文章目录


Python学习笔记—装饰器

装饰器:从生活角度理解,是对一个东西进行装饰,增加它本身的一些功能和内容;
Python中的装饰器,也大致可以这样理解,它本质上就是扩展原本函数功能的一种函数。它是通过函数闭包进行实现的。

1.闭包

闭包是什么?
答: 定义在一个函数中的函数称为闭包函数,前一个函数称为外函数,后一个函数称为内函数,闭包指的就是,内函数总是可以访问其所在的外函数中声明的参数和变量,即使在其外函数被返回之后。

想要实现闭包,需要三个条件:

  • 函数嵌套,即存在内函数和外函数
  • 内函数使用了外函数的变量或者参数
  • 外函数需要return内函数

举例:

def func_out():  # 外函数
    num1 = 10
    def func_in(num2):   # 内函数
        t = num1 + num2
        print("%d + %d = %d"%(num1,num2,t))
    return func_in  # 返回内函数, 不加括号

# 这里实际上是 func_out()函数调用之后返回的内函数地址赋值给f
f = func_out()   
f(1)   # 为了方便记忆,可以直接将内函数与这里的f直接等价,调用的时候,需要以内函数的形参列表为基准传入参数
# 输出: 10 + 1 =11

思考:有没有想过,写在代码中的函数名到底是什么呢?
答: 其实是函数体代码的存储地址。

举例:

def test():
    pass
print(test)    # 输出 :<function test at 0x000001AC070275E0>
  • 这个时候,在回头看一下闭包的例子,外函数返回的是内函数的地址,f接收的是外函数调用的结束后返回的内函数地址,这时想要调用内函数,直接按照内函数的形参列表给f传入参数即可。
  • 在深入一点,一般情况下,在func_out()函数调用结束后,它内部的num1局部变量理应被销毁释放,但是f在调用时候,还是用到了num1的值,这就是闭包的真正含义。

2.不带参数的装饰器

装饰器就是通过闭包实现的,接下来举两个常见例子帮助理解:
  • 日志打印器 – 通知函数执行和结束
def decorate(func):     # 这里的参数是函数
   print("装饰器执行了")      # 可以看出,这里的装饰器就是外函数有了参数的闭包
   def inner():
       print("执行函数了")
       func()
       print("函数执行结束了")

   return inner;


@decorate      # @decorate 值得是装饰器语法糖,它的执行意义是 comment = decorate(comment)  comment()
def comment():
   print("评论开始了")
   
comment()

@ 符号就是装饰器的语法糖。它放在一个函数开始定义的地方,它就像一顶帽子一样戴在这个函数的头上,和这个函数绑定在一起。在我们调用这个函数的时候,第一件事并不是执行这个函数,而是将这个函数做为参数传入它头顶上这顶帽子,这顶帽子我们称之为装饰函数 或 装饰器。

  • 函数时间计时器 --计算一个函数执行的时间
from time import *     # 导入time模块,具体导入模块可以参看我的另一篇文章

def decorate(func):  # 外函数,这里的func参数为@下一行的函数
   def inner():    # 内函数
       begin = time()
       func()            # 这里调用了外函数的局部变量
       end = time()
       result = end - begin   
       print("执行的时间为:"+str(result))

   return inner;

@decorate     # 装饰器语法糖
def work():
   s = 0
   for i in range(1,10001):
       s += i
   print("1到10000的和为%d:"%(s))

work()

以上两个例子就是不带参数的装饰器,在写装饰器的时候要注意装饰器内函数中调用func的形式,有时候func(@下一行 的函数)是有return返回值的,需要注意一下。

3.带参数的装饰器

  • 带参数的装饰器,指的是装饰器内部inner函数是需要有参数,也表明@装饰器修饰的函数也是需要有和inner函数有相同的形参列表。

举例:

def decorate (func):
    def inner(a,b):     # 这里的inner函数有参数,则也需要func也需要有参数
        print("努力计算中!!!")
        func(a,b)
    return inner


@decorate  # 这里本质是 f = decorate(add1), f(a,b) 所以inner函数也需要和add1函数有相同的形参列表
def add1(a,b):    
    print("%d + %d = %d"%(a,b,a+b))
    
add1(2,3)
# 输出: 努力计算中!!!
# 2 + 3 = 5

建议 想要加深理解,一定要参考上面提到的函数名本质是函数本体代码存放的地址,就是一段地址值。将注释后的代码用地址代入走一遍,就更容易理解了

4.不定长参数的装饰器

有了上面的带参数的装饰器,这里的不定长参数的装饰器,就是将inner函数的参数列表改为不定长参数占位符就可以了。具体可以参考我的另外一篇文章《Python函数中的不定长参数》

举例:

def decorate(func):
    def inner(*args,**kwargs):     # 不定长参数列表
        print("不定长装饰器运行啦")
        func(*args,**kwargs)      # 内函数调用外函数变量
        
    return inner     # 返回内函数

@decorate  # 相当于 f = decorate(add_num),  f(*args,**kwargs)
def add_num(*args,**kwargs):
    s = 0
    for i in args:
        s += i
    for i in kwargs.values():
        s += i
    print("和为:", s)

add_num(1,2,3,4)

结果演示:
Python自定义装饰器

5.多重装饰器修饰

  • 即一个函数有多个装饰器修饰。

举例:

def make_div(func):
    print("make_div 装饰器");
    def inne():
        result = '<div>' + func() + '</div>'
        return result
    return inne;


def make_p(func):
    print("make_p 装饰器");
    def inne():
        result = '<p>' + func() + '</p>'
        return result
    return inne;

@make_div    # 当一个函数有多个装饰器时候,谁靠的近先执行谁,
@make_p   # 本质上调用的是 f = make_p(content) f1 = make_div(f) 
def content():
    return "人生苦短,我用python"

print(content())

运行结果:
Python自定义装饰器

  • 主要掌握的是在多重装饰器修饰函数时,装饰器运行的先后顺序:谁距离被修饰的函数就先用谁修饰

6.返回值是装饰器

按照要求,则返回的是装饰器,则return decorate。

举例:



def return_decorate(flag):   # 定义一个函数用来返回装饰器
    def decorate(func):
        def inner(a,b):
            if flag == '+':
                print("执行的是加法运算")
                func(a,b)
            elif flag == '-':
                print("执行的是减法运算")
                func(a,b)
        return inner;
    return decorate

@return_decorate('+')  # 实际上执行的是 f = return_decorate('+') f1 = f(add_num) f1(a,b)
def add_num(a,b):
    print("%d + %d = %d"%(a,b,a+b))

@return_decorate('-')
def sub_num(a,b):
    print("%d - %d = %d"%(a,b,a-b))

add_num(2,4)
sub_num(4,2)

结果演示:
Python自定义装饰器

7.装饰器类

将装饰器函数封装成类,通过__call__函数将类转化为可调用的对象,用对象调用被修饰的函数

举例:

class MyDecorate(object):
    def __init__(self,func):   # 这里相当于将func初始化为类的实例属性,相当于外函数的局部变量
        self.func = func

    def __call__(self, *args, **kwargs):     # 该方法将类变为可调用的对象,可调用的对象能够和函数一样调用使用,这里相当于对inner内函数的相关执行代码进行规范化处理
        print("装饰器类调用了")
        self.func(*args,*kwargs)

@MyDecorate    # f = MyDecorate(show),f(str)
def show(str):
    print(str)

@MyDecorate
def add(a,b):
    print("%d + %d = %d"%(a,b,a+b))
    
show("加油,奥里给")
add(3,5)

结果展示:
Python自定义装饰器

这里的装饰器类,可以大大的增加的代码的通用性,同时需要注意,当类中出现__call__魔法方法时,该类大概率为装饰器类。

上一篇:(奇淫技巧)C# 利用反射特性,生成实际的Func类型的委托,可以用于参与数据集合操作


下一篇:C++函数重载与重载原理:命名倾轧