今天写篇文章,详解 Python 装饰器。文末有我的微信二维码,打算建个读者群,想交流 Python 的朋友可以加一下。
装饰器(decorator)是一种高级Python语法,也是 Python 函数式编程的一部分。写法是 @xxx,在 Pandas、Django 框架源码里经常能见到,应用非常广泛。
虽然不用装饰器也不影响开发,但用了装饰器可以让你的代码“秀”起来,当然装饰器的作用不仅仅是“秀”,更重要的是它可以让你的代码更简洁,可读性更高;业务逻辑解耦,维护更容易。
1 函数基础
学习装饰器之前,我们先学习下 Python 函数的两个特性,函数体内可以定义函数 和 函数可以被引用。有基础的朋友可以略过,直接看下一小节。
例1. 函数体内定义函数
def func1():
def func2():
return "in func2"
res = func2()
return res
在 func1 函数体内, 定义了一个 func2 函数并调用, 然后返回调用的结果。其实很多静态编程语言,如 C、Java,不支持在函数体内定义函数。但在 Python 中是支持的。
例2. 函数可以被引用
new_func = func1
print(new_func())
func1 是个函数,定义跟 例1 是一样的。可以看到函数名可以像变量那样使用。赋值后,new_func 就引用了 func1,就可以直接通过 new_func() 方式来调用函数。\
既然函数名可以向变量一样赋值,那自然就可以被 return 返回。
def func1():
def func2():
return "in func2"
print(func2)
return func2
new_func = func1()
print(new_func)
print(new_func())
输出:
.func2 at 0x102499b80>
.func2 at 0x102499b80>
in func2
print(func1) 和 print(new_func) 输出结果一样,说明他们是同一个函数。\
2 装饰器
有了上面的基础,再学习装饰器就很容易了。下面通过一个 “炒土豆丝” 的例子来学习装饰器。
例3. “炒土豆丝”基础版
def tu_dou_si():
"""
"清炒土豆丝" 流程
"""
print('清洗土豆')
print('去皮,切丝')
print('清炒')
print('出锅')
这是最基础的版本。为了跟国际接轨, 例子中使用中文拼音定义函数名或变量名。
下面我们改造 例3,将其变成装饰器版本。
例4. “炒土豆丝”装饰器版
def my_decorator(func):
def wrapper():
func()
print('清炒')
print('出锅')
return wrapper
def tu_dou_si():
print('清洗土豆')
print('去皮,切丝')
有没有发现它跟 例2 特别像。
my_decorator 函数就是一个装饰器,它参数 func 表示可以传入一个函数,它里面定义了一个函数 wrapper,然后将 wrapper 函数返回。
wrapper 的中文含义是包裹、包装的意思,所以在这里表示对 func() 进行了包装,包装的效果就是多了两个 print 输出。
tu_dou_si 函数,只保留跟土豆丝相关的 print 语句。
定义好了装饰器,就可以按照下面的方式,实现 例3 中“清炒土豆丝”的流程
qing_chao_tu_dou_si = my_decorator(tu_dou_si)
qing_chao_tu_dou_si()
输出:
清洗土豆
去皮,切丝
清炒
出锅
对于上面的代码 Python 有一种更简洁的方式来实现\
@my_decorator
def tu_dou_si():
print('清洗土豆')
print('去皮,切丝')tu_dou_si()
就是 Python 的装饰器。因为 tu_dou_si 函数已经被 my_decorator 装饰过了, 所以,直接调用 tu_dou_si 函数即可完成 “清炒土豆丝”的流程。
3 装饰器的优势
虽然我们已经实现了一个装饰器,但你心里可能会有疑问,装饰器写法不简单,代码量比 例3 也不少,它的优势究竟在哪。
从“清炒土豆丝”这个单点确实看不出装饰器优势,如果从线再到面角度来看,就能看出装饰器的优势了。
“清炒土豆丝”流程可以拆分为两个业务,一个是准备食材业务,即: print('清洗土豆') 和 print('去皮,切丝') ;另一个是制作食材的业务,即: print('清炒') 和 print('出锅')。之所以这样拆,是因为这两部分是相互独立的子业务,并且能方便看到装饰器的优势。问题1. 要再实现一个“清炒山药”的流程,怎么做
@my_decorator
def shan_yao():
print('清洗山药')
print('去皮,切丝')
可以看到, 用装饰器来实现可以复用 my_decorator 中制作食材的业务过程,增加代码复用性,使整体代码更简洁。
问题2. 想把“清炒土豆丝”改成“凉拌土豆丝”,怎么办
def my_decorator(func):
def wrapper():
func()
print('凉拌')
print('出锅')
return wrapper
直接修改 my_decorator 业务就可以了,不会对其他业务产生副作用。
理解了上面两个问题,我们也就知道了装饰器的作用。如果不使用装饰器(如:例3 )要么会产生代码冗余,要么就业务耦合在一起改一个业务影响到另外的业务。
4 装饰器的进阶
接下来,来看看装饰器高级一些的用法。
例5. 装饰器的参数
def cai_factory(cai_type='清炒'):
def my_decorator(func):
def wrapper():
func()
print(f'{cai_type}')
print('出锅')
return wrapper
return my_decorator
@cai_factory('凉拌')
def tu_dou_si():
print('清洗土豆')
print('去皮,切丝')
定义了一个 cai_factory 函数(即:菜工厂),它里面包含了之前定义的 my_decorator 装饰器。cai_factory 函数接收 cai_type 参数用来指定如果制作食材。
定义 tu_dou_si 函数时,使用新的装饰器,并且传入参数即可。
例6. 被装饰函数的参数
def cai_factory(cai_type='清炒'):
def my_decorator(func):
def wrapper(cai_name):
func(cai_name)
print(f'{cai_type}')
print('出锅')
return wrapper
return my_decorator
@cai_factory()
def cai(cai_name='土豆'):
print(f'清洗{cai_name}')
print('去皮,切丝')
定义一个用 cai_factory 修饰的函数 cai,接收 cai_name 参数来指定制作的食材。同时,在 cai_factory 装饰器中的 wrapper 函数中增加 cai_name 参数,并在调用 func 时,传入该参数。
例7. 被装饰函数的不定参数
def cai_factory(cai_type='清炒'):
def my_decorator(func):
def wrapper(*args, **kwargs):
func(*args, **kwargs)
print(f'{cai_type}')
print('出锅')
return wrapper
return my_decorator
@cai_factory()
def cai(cai_name, cai_cnt):
print(f'清洗{cai_name} {cai_cnt}')
print('去皮,切丝')
cai('山药', '两个')
例6 只能支持1个参数,本例中的 wrapper 函数的参数改为 args 和 *kwargs,就可以支持任意参数,把装饰器变得更通用。
5 装饰类
类装饰器的用法跟函数是一样的。这次我们不做菜了,开始玩游戏了。游戏里面皮肤和英雄的关系就完美地表现了装饰类与装饰类的关系。下面我们就定义一个英雄类,然后定义一个皮肤类装饰它。
def pi_fu(cls):
class PiFuClass:
def __init__(self, name, pi_fu_name):
self.wrapped = cls(name, pi_fu_name)
self.pi_fu_name = pi_fu_name
def display(self):
self.wrapped.display()
print(f'展示皮肤{self.pi_fu_name}')
return PiFuClass
@pi_fu
class YingXiong:
def __init__(self, name, pi_fu_name):
self.name = name
self.pi_fu_name = pi_fu_name
def display(self):
print(f'展示英雄{self.name}')
ya_se = YingXiong('亚瑟', '死亡骑士')
ya_se.display()
写法和用法跟函数装饰器类似,这里就不再赘述了。
6 内置装饰器
Python 内置了一些装饰器,这里列几个我们经常会用到的一些:
- @property:修饰类方法,使得我们可以像访问属性一样来获取一个函数的返回值
- @staticmethod:修饰类方法,表示该方法是一个静态方法,可以直接被调用无需实例化
- @wraps:函数来进行装饰,可以让我们在装饰器里面访问在装饰之前的函数的属性
简单总结一下,Python 装饰器支持修饰方法和类,可能增加被修饰方法和方法的功能。同时又能起到业务之间解耦、增加代码复用性和简化代码的作用。
以上就是本次分享的所有内容,如果你觉得文章还不错,欢迎关注公众号:Python编程学习圈,每日干货分享,发送“J”还可领取大量学习资料。或是前往编程学习网,了解更多编程技术知识。