装饰器的作用就是用一个新函数封装旧函数(是旧函数代码不变的情况下增加功能)然后会返回一个新函数,新函数就叫做装饰器,一般为了简化装饰器会用语法糖@新函数来简化
例子:
这是一段代码,但功能太少,要对这个进行增强,但又不能改变代码。
1
2
|
def hello():
return "hello world!"
|
现在我们的需求是要增强hello()函数的功能,希望给返回加上HTML标签,比如<i>hello world</i>,但要求我们不得改变hello()函数原来的定义。
1
2
3
4
5
6
7
8
9
10
11
|
def makeitalic(fun): #makitalic传了一个新函数
def wrapped(): #内部函数
return "<i>" + fun() + "</i>" #要加的新功能
return wrapped #返回的是wrapped函数功能
def hello(): #对这个功能进行增强
return "hello world!"
#makeitalic里面传入了hello函数,然后内部函数fun()函数也就相当于hello函数了
hello_2 = makeitalic(hello)
#打印新函数,返回的就是<i>hello world!</i>
print (hello_2())
|
为了增强原函数hello的功能,定义了一个函数,它接收原函数作为参数,并返回一个新的函数,在这个返回的函数中,执行了原函数,并对原函数的功能进行了增强。
事实上,makeitalic就是一个装饰器(decorator),它封装了原函数hello,并返回了一个新函数,用于增强原函数的功能,并将其赋值给hello。
一般情况下,我们使用装饰器提供的@语法糖(Syntactic Sugar),来简化上面的操作。
1
2
3
4
5
6
7
8
9
10
|
####使用@语法糖
def makeitalic(fun):
def wrapped():
return "<i>" + fun() + "</i>"
return wrapped
@makeitalic #使用了装饰器可以直接调用,不需要赋值了
def hello():
return "hello world"
print (hello()) #使用了装饰器可以直接调用,不需要赋值了
|
像上面的情况,可以动态的修改函数(或类的)功能的函数就是装饰器。本质上,它是一个高阶函数,以被装饰的函数(比如上面的hello)为参数,并返回一个包装后的函数(比如上面的wrapped)给被修饰函数(hello)。
当调用hello()函数时,hello函数的执行流程如下分析:
1.把hello函数作为参数传给@符号后面的装饰器函数。
2.然后开始执行装饰器函数,并返回一个包装了的函数,同时,改变原函数的指向,现在原函数指向了这个包装函数。
3.执行原函数,其实此时执行的是包装了的函数,所以说,装饰器增强了一个现有函数的功能,但不会改变现有函数的定义。
普通装饰器的使用形式:
1
2
3
4
5
6
7
8
9
|
@decorator
def fun():
pass
#格式就如同下面的:
def fun():
pass
fun = decorator(fun)#不使用语法糖要进行赋值
|
装饰器可以定义多个,离函数定义最近的装饰器最先被调用,比如:
1
2
3
4
5
6
7
8
9
10
|
@decotator_one
@decorator_two
def fun():
pass
#格式如同下面的:
def fun():
pass
fun = decorator_one(decorator_two(fun))
|
装饰器还可以带参数,比如:
1
2
3
4
5
6
7
8
9
|
@decorator (arg1, arg2)
def fun():
pass
#格式如同下面的:
def fun():
pass
fun = decorator(arg1, arg2)(fun)
|
到此这篇关于python中装饰器的作用的文章就介绍到这了,更多相关python装饰器作用内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!
原文链接:https://www.cnblogs.com/wuzhuangzhuang/p/12673047.html