python中装饰器的作用

装饰器的作用就是用一个新函数封装旧函数（是旧函数代码不变的情况下增加功能）然后会返回一个新函数，新函数就叫做装饰器，一般为了简化装饰器会用语法糖@新函数来简化

例子：

这是一段代码，但功能太少，要对这个进行增强，但又不能改变代码。

1 def hello():
2     return "hello world!"

现在我们的需求是要增强hello()函数的功能，希望给返回加上HTML标签，比如<i>hello world</i>,但要求我们不得改变hello()函数原来的定义。

 1 def makeitalic(fun):#makitalic传了一个新函数
 2     def wrapped():#内部函数
 3         return "<i>"+fun()+"</i>"#要加的新功能
 4     return wrapped#返回的是wrapped函数功能
 5 
 6 def hello():#对这个功能进行增强
 7     return "hello world!"
 8 #makeitalic里面传入了hello函数，然后内部函数fun()函数也就相当于hello函数了
 9 hello_2=makeitalic(hello)
10 #打印新函数，返回的就是<i>hello world!</i>
11 print(hello_2())

为了增强原函数hello的功能，定义了一个函数，它接收原函数作为参数，并返回一个新的函数，在这个返回的函数中，执行了原函数，并对原函数的功能进行了增强。

事实上，makeitalic就是一个装饰器（decorator），它封装了原函数hello，并返回了一个新函数，用于增强原函数的功能，并将其赋值给hello。

一般情况下，我们使用装饰器提供的@语法糖（Syntactic Sugar），来简化上面的操作。

####使用@语法糖
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.然后开始执行装饰器函数，并返回一个包装了的函数，同时，改变原函数的指向，现在原函数指向了这个包装函数。

扫描二维码关注公众号，回复： 10658593 查看本文章

3.执行原函数，其实此时执行的是包装了的函数，所以说，装饰器增强了一个现有函数的功能，但不会改变现有函数的定义。

普通装饰器的使用形式：

1 @decorator
2 def fun():
3  pass
4 
5 #格式就如同下面的：
6 
7 def fun():
8  pass
9 fun = decorator(fun)#不使用语法糖要进行赋值

装饰器可以定义多个，离函数定义最近的装饰器最先被调用，比如：

 1 @decotator_one
 2 @decorator_two
 3 def fun():
 4  pass
 5 
 6 #格式如同下面的：
 7 
 8 def fun():
 9  pass
10 fun = decorator_one(decorator_two(fun))

装饰器还可以带参数，比如：

1 @decorator(arg1, arg2)
2 def fun():
3  pass
4 
5 #格式如同下面的：
6 
7 def fun():
8  pass
9 fun = decorator(arg1, arg2)(fun)

python中装饰器的作用

猜你喜欢