装饰器是什么
用来修饰别的函数的函数就可以称之为装饰器
这种函数的参数一般就是另外一个函数
也就是说,调用这种函数,需要给这种函数传参,且参数是函数
@语法糖
@语法糖一般用来表示装饰器函数
不用@也可以达到装饰函数的目的,下面会有演示
函数嵌套
在一个函数中定义另外一个函数
def f1(arg="aaa"):
def f2():
return "hello"
def f3():
return "hi"
def f4():
return "haha"
print f2()
print f3()
print f4()
这个f1函数有默认参数,所以可以不传参执行
执行f1()调用之后
执行结果如下:
hello
hi
haha
在函数里返回函数
def hi(arg="aaa"):
def greet():
return "bbb"
def welcome():
return "ccc"
if arg == "aaa":
return greet
else:
return welcome
a = hi()
print(a)
print (a())
这里同样使用了默认参数,则a = hi() 会命中 if arg == "aaa"这个逻辑
返回greet,注意在这里,greet是函数,不是字符串,如果是返回字符串,则要返回的是 return "greet"这种
上面这段代码执行的结果是
为什么是这样的执行结果呢?第一个地方 print(a),打印的是a = hi()的结果
我们可以看到,hi()的返回结果都是在return一个函数,要么是greet函数,要么是welcome函数
函数就是对应一个地址,所以第一处打印的是这个函数的地址
第二处做了a的调用,即a(),则打印返回的函数即greet函数执行的结果,即bbb
将函数作为参数传给另外一个函数
def hi():
return "hi"
def hello(func):
print("before func()")
print(func())
hello(hi)
执行结果是
before func()
hi
这里把hi这个函数作为参数传给hello函数
hello函数先打印一句自身的输出before func()
再执行这个被传入的函数
我们可以看到,通过装饰器,我们可以在一个函数被调用前干一些需要的事情
不用@实现装饰器
def hi(a_func):
def hello():
print("I am doing some boring work before executing a_func()")
a_func()
print("I am doing some boring work after executing a_func()")
return hello
def haha():
print("I am the function which needs some decoration ")
haha = hi(haha)
print haha()
运行结果:
I am doing some boring work before executing a_func()
I am the function which needs some decoration
I am doing some boring work after executing a_func()
None
这个就是装饰器,起到了用hi函数装饰了haha函数的功能
那么怎么用@语法糖实现呢
def hi(a_func):
def hello():
print("I am doing some boring work before executing a_func()")
a_func()
print("I am doing some boring work after executing a_func()")
return hello
@hi
def haha():
print("I am the function which needs some decoration ")
haha()
可以看到,装饰器就是用希望装饰别的函数的函数,比如 ,希望用A装饰B
就在定义B函数的上一行,写上 @A
def A():
pass
@A
def B():
pass
总结:理解就是装饰器其实就是这样一种函数:带参数的,且参数是另外一个函数的函数
使用装饰器的目的一般是为了在运行时改变函数的一些属性/行为,就可以给这个函数加上装饰器,让装饰器去在这个函数被调用前后,干一些你想做的事情
wraps干什么的?
我们修改下上面的代码,多记录一点东西
def hi(a_func):
def hello():
print("I am doing some boring work before executing a_func()")
a_func()
print("I am doing some boring work after executing a_func()")
return hello
@hi
def haha():
print("I am the function which needs some decoration ")
haha()
print(haha.__name__)
我们的意图是打印haha这个函数的函数名,但实际上打印的是hello
为什么?
因为haha这个函数使用装饰器之后,haha的行为部分被装饰器函数改变了
可以看到,用hi函数装饰了haha之后,返回的是hello函数,那么当我们需要拿到被装饰函数的函数名,还有其他属性的时候怎么做呢?
使用functools.wraps方法
from functools import wraps
def hi(a_func):
@wraps(a_func)
def hello():
print("I am doing some boring work before executing a_func()")
a_func()
print("I am doing some boring work after executing a_func()")
return hello
@hi
def haha():
print("I am the function which needs some decoration ")
haha()
print(haha.__name__)
执行结果如下:
I am doing some boring work before executing a_func()
I am the function which needs some decoration
I am doing some boring work after executing a_func()
haha
正是我们想要的结果
@wraps接受一个函数来进行装饰,并加入了复制函数名称、注释文档、参数列表等等的功能。这可以让我们在装饰器里面访问在装饰之前的函数的属性。
举一个记录日志的例子
from functools import wraps
def logit(func):
@wraps(func)
def with_logging(*args, **kwargs):
print(func.__name__ + " was called")
return func(*args, **kwargs)
return with_logging
@logit
def addition_func(x):
"""Do some math."""
return x + x
result = addition_func(4)
# Output: addition_func was called