python-装饰器简述

用来修饰别的函数的函数就可以称之为装饰器
这种函数的参数一般就是另外一个函数
也就是说，调用这种函数，需要给这种函数传参，且参数是函数

@语法糖

@语法糖一般用来表示装饰器函数
不用@也可以达到装饰函数的目的，下面会有演示

函数嵌套

在一个函数中定义另外一个函数

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