写点什么

二十一、深入 Python 强大的装饰器

用户头像
刘润森
关注
发布于: 2020 年 10 月 22 日
二十一、深入Python强大的装饰器

@Author: Runsen


最近有同学在问关于 Python 中装饰器的问题,说不太理解装饰器的装饰过程。


那么在下面 Runsen 来给大家深入讲解一下装饰器的整个实现过程的。


闭包


想要理解 Python 中的装饰器,不得不先理解闭包(closure)这一概念。


闭包就应该想起了嵌套函数,也可以将闭包理解为一种特殊的函数,这种函数由两个函数的嵌套组成,外函数和内函数。


def 外层函数(参数):    def 内层函数():        print("内层函数执行", 参数)
return 内层函数
内层函数的引用 = 外层函数("传入参数")内层函数的引用()
复制代码


在一个外函数中定义了一个内函数,内函数里运用了外函数的临时变量,并且外函数的返回值是内函数的引用。这样就构成了一个闭包。


下面举一个具体的闭包函数的实例,代码如下。


# outer是外部函数def outer(a):    # inner是内函数    def inner( b ):        #在内函数中 用到了外函数的临时变量        print(a+b)    # 外函数的返回值是内函数的引用    return innerret = outer(5) #ret = innerret(10) #15 ret 存了外函数的返回值,也就是inner函数的引用,这里相当于执行inner函数
复制代码


装饰器


装饰器,顾名思义,就是用来“装饰”的。比如@Runsen就是一个装饰器,其中"Runsen"是你的装饰器的名字。它能装饰的东西有:函数、类。装饰器一般在函数、类的上面用@符号定义。


装饰器本质上是一个 Python 函数(一定有参数),如果严格来说,装饰器只是语法糖,也可以将装饰器叫做一种特殊的闭包。


装饰器是可调用的对象,可以像常规的可调用对象那样调用,特殊的地方是装饰器的参数是一个函数名。


下面就是最简单的装饰器,代码来自 Python3 官方文档。


def warp(obj):    return obj        @warp    # 等价于 foo = warp(foo)def foo():        print('hello decorator!') foo()    # => hello decorator!  
复制代码

上面使用了装饰器的代码,其实我们可以通过其它方式达到相同的效果,具体见下。


def foo():    print('hello decorator!') foo = warp(foo)foo()    # => hello decorator!
复制代码


嵌套函数的装饰器


在上面代码中装饰器都只是一个普通的函数,如果该函数是嵌套函数,那么函数传入的参数,在内部函数依然可以使用。


先看两段代码,在这里my_decorator就是一个装饰器。

def my_decorator(func):    def wrapper():        print('wrapper of decorator')        func()    return wrapper
def greet(): print('hello world')
greet = my_decorator(greet)greet()
# 输出wrapper of decoratorhello world
复制代码


my_decorator函数传入greet函数名方法,中间有一个wrapper内函数方法, 而return wrapper说明要执行wrapper内函数,wrapper内函数,于是执行greet函数名方法。


其实,greet = my_decorator(greet)这个代码可以用装饰器来替代,在greet上面加一个@my_decorator,然后直接执行 greet(),最终输出一样。


def my_decorator(func):    def wrapper():        print('wrapper of decorator')        func()    return wrapper
@my_decoratordef greet(): print('hello world')
greet()
wrapper of decoratorhello world
复制代码


装饰器就是继承了 my_decorator函数,因此先调用my_decorator中的wrapper打印出

wrapper of decorator,然后func()被调用,传入的参数是greet,因此指的就是 greet(),所以在打印出hello world


带参数嵌套函数的装饰器


有时候嵌套函数需要传入参数到内部函数,这时候用*args, **kwargs接受就可以了。*args接收元组,**kwargs接受字典。


下面,我们来看一个例子。


# repeat重复输出,num指重复输出次数def repeat(num):    def my_decorator(func):        def wrapper(*args, **kwargs):            for i in range(num):                print('wrapper of decorator')                func(*args, **kwargs)        return wrapper    return my_decorator

@repeat(4)def greet(message): print(message)
greet('hello world')
# 输出:wrapper of decoratorhello worldwrapper of decoratorhello worldwrapper of decoratorhello worldwrapper of decoratorhello world
复制代码


上面代码的意思:@repeat(4)将会执行greet(4),由于存在return my_decorator,所以下一步执行my_decorator(greet).由于又存在 return wrapper。所以下一步将会执行wrapper(*args, **kwargs)。这里的*args, **kwargs指的是hello world字符串。因此最终打印四次wrapper of decoratorhello world


但是自定义参数的装饰器将改变函数本身的元信息,即函数不再是本身的函数


greet.__name__## 输出'wrapper'
复制代码


这时,需要使用内置模块functools.wrap会保留原函数的元信息。


import functools
def repeat(num): def my_decorator(func): @functools.wraps(func) def wrapper(*args, **kwargs): for i in range(num): print('wrapper of decorator') func(*args, **kwargs) return wrapper return my_decorator
@repeat(4)def greet(message): print(message)
greet.__name__
# 输出 不是wrapper'greet'
复制代码


类装饰器


类装饰器主要依赖函数__call__ ,因此我们主要重写__call__即可。


每当调用一个类的实例,函数__call__就会执行一次。


下面,我们来看一个例子。


class Count:    def __init__(self, func):        self.func = func        self.num_calls = 0
def __call__(self, *args, **kwargs): self.num_calls += 1 print('num of calls is: {}'.format(self.num_calls)) return self.func(*args, **kwargs)
@Countdef example(): print("hello world")
example()
# 输出num of calls is: 1hello world
example()
# 输出num of calls is: 2hello world
复制代码


嵌套装饰器


我们可以把多个装饰器叠加在同一个函数上,这个就叫做嵌套装饰。


嵌套装饰器的顺序是从下到上,


@f2@f1def greet(name):    print(f"Hello {name}")
复制代码

所以上面的例子就是等价于:`greet = f2(f1(greet))

`


那么相当于,从里到外。


import functools
def my_decorator1(func): @functools.wraps(func) def wrapper(*args, **kwargs): print('execute decorator1') func(*args, **kwargs) return wrapper

def my_decorator2(func): @functools.wraps(func) def wrapper(*args, **kwargs): print('execute decorator2') func(*args, **kwargs) return wrapper

@my_decorator1@my_decorator2def greet(message): print(message)
greet('hello world') #相当于my_decorator1(my_decorator2(greet('hello world')))
# 输出execute decorator1execute decorator2hello world
复制代码


看完这篇文章还不理解装饰器,只有一种可能,说明我写的还不够清晰,那点赞鼓励鼓励我吧。


今天也学到了很多东西呢,明天有什么新知识呢?真期待鸭~如果喜欢文章可以关注我哦~


本文已收录 GitHub,传送门~ ,里面更有大厂面试完整考点,欢迎 Star。


<br>

<br>


发布于: 2020 年 10 月 22 日阅读数: 36
用户头像

刘润森

关注

刘润森 2018.09.17 加入

17年就读于东莞XX学院化学工程与工艺专业,GitChat作者。Runsen的微信公众号是"Python之王",关注后回复「小白」即可免费获取原创的Python学习资料;喜欢的微信搜索:「Python之王」。个人微信号:RunsenLiu

评论

发布
暂无评论
二十一、深入Python强大的装饰器