Python装饰器
以前在IMOOC上学习的笔记。今晚整理下发出来。
要理解装饰器,先了解函数作用域和闭包。
函数作用域的查找顺序概括为:L -> E -> G -> B
local 函数内部作用域
enclosing 函数内部与内嵌函数之间
global 全局作用域
build-in 内置作用域
闭包(Closure)就是内部函数中对enclosing作用域的变量进行引用。举例:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24# -*-coding:utf-8 -*-
def foo(func):
def wrapper(x, y):
if isinstance(x, int) and isinstance(y, int):
return func(x, y)
else:
print 'Error'
return None
return wrapper
def mysum(x, y):
return x + y
def mymul(x, y):
return x * y
mysum_ex = foo(mysum)
print mysum_ex(1, 2)
print mysum_ex.__closure__
mymul_ex = foo(mymul)
print mymul_ex(1, 2)
print mymul_ex.__closure__执行完
foo
函数,内部变量被回收,这里是func
变量。但是在执行mysum_ex
函数时,为什么访问得到func
变量呢?因为闭包函数mysum_ex
(实际上就是wrapper
函数)有一个属性__closure__
,存储着需要用到的enclosing 变量,即func
变量。闭包作用:封装和代码复用。怎么体现?
上面的
mysum
用来执行两个整数的加法,mymul
用来执行两个整数的乘法。如果直接执行mysum('1', 2)
或mymul('1', 2)
是会报错:TypeError: cannot concatenate 'str' and 'int' objects
。所以需要添加判断参数是否是int类型的逻辑代码,如mysum
函数1
2
3
4
5
6def mysum(x, y):
if isinstance(x, int) and isinstance(y, int):
return x + y
else:
print 'Error'
return Nonemymul
函数也需要相同的判断逻辑,所以利用闭包把这部分判断逻辑代码封装到wrapper()
中,通过传递函数对象func
给wrapper()
函数内部使用,来决定到底使用加法操作还是乘法操作。这部分判断参数是否是int类型的逻辑代码达到了代码复用。
最后说装饰器。上面的闭包例子用装饰器来写:
1 |
|
执行结果:
1 |
|
对于装饰器的几点说明:
装饰器是对闭包的使用。
传递一个函数对象
func
给装饰器foo
,foo
返回一个新的函数对象。对于闭包,就不一定要传递函数对象,可以传递其它类型的对象。如:1
2
3
4
5
6
7
8
9
10
11
12
13# -*-coding:utf-8 -*-
def set_passline(passline):
def cmp(val):
if val >= passline:
print "pass"
else:
print "failed"
return cmp
f_100 = set_passline(60)
print f_100.__closure__
f_100(89)传递的是整数对象。
在
def mysum(x, y):
上边加@foo
,是Python提供的语法。表示mysum
这个函数对象作为func
参数传递给foo(func)
函数,再把foo
函数返回的新函数对象赋给mysum
。建议给装饰器里的
wrapper
加上@functools.wraps(func)
装饰器。不加的时候mysum.__name__
的值是wrapper
,加上去值才是mysum
。
本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!