今天想用python的装饰器做一个运算，代码如下
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
>>> def mu(x):
def _mu(*args,**kwargs):
return x*x
return _mu
>>> @mu
def test(x,y):
print '%s,%s' %(x,y)
>>> test(3,5)
Traceback (most recent call last):
File "<pyshell#111>", line 1, in <module>
test(3,5)
File "<pyshell#106>", line 3, in _mu
return x*x
TypeError: unsupported operand type(s) for *: 'function' and 'function'
|
原来是不能这样弄的 函数与函数是不能运算的啊！
怎么办呢？
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
In [1]: from functools import wraps
In [2]: def mu(x):
...: @wraps(x)
...: def _mu(*args,**kwargs):
...: x,y=args
...: return x*x
...: return _mu
...:
In [3]: @mu
...: def test(x,y):
...: print '%s,%s' %(x,y)
...:
In [4]: test(3,4)
Out[4]: 9
|
Python装饰器（decorator）在实现的时候，有一些细节需要被注意。例如，被装饰后的函数其实已经是另外一个函数了（函数名等函数属性会发生改变）
Python的functools包中提供了一个叫wraps的decorator来消除这样的副作用。写一个decorator的时候，最好在实现之前加上functools的wrap，它能保留原有函数的名称和docstring。
以上所述就是本文的 全部内容了，希望大家能够喜欢。