这个问题的来源是在看python中的装饰器的时候,例子里给了这样一段代码:
#-*- coding: UTF-8 -*-
import time
def foo():
print 'in foo()'
# 定义一个计时器,传入一个,并返回另一个附加了计时功能的方法
def timeit(func):
# 定义一个内嵌的包装函数,给传入的函数加上计时功能的包装
def wrapper():
start = time.clock()
func()
end =time.clock()
print 'used:', end - start
# 将包装后的函数返回
return wrapper
foo = timeit(foo)
foo()
上述代码实现了一个装饰器,但没有处理foo()函数带参数以及返回值的情况。为了改进这段代码,首先要理解的是python中参数传递*和**的问题。
如果在函数定义时候,按照如下定义的:
#代码段1
def foo(*a):
pass
或者
#代码段2
def foo(**a):
pass
这种定义方法相当与C语言中的不定参数,在定义的时候,函数将被传入的参数是未知的,运行的时候,代码段1 的代码 参数将会一tuple的形式组织起来,传入函数; 代码段2 的代码,参数将会以dictionary的形式传入
*和**除了在定义函数的时候有用,它的另一个功能就是对tuple和dict进行展开。
于是我们便可以利用这个运算符对最开始的代码进行改进:
#-*- coding: UTF-8 -*-
import time
def foo(a):
print ' in foo() %s'%a
return 100
def foo2():
print ' in foo2()'
return
#定义定时器,传入一个,并返回另一个附加了计时功能的方法
def timeit(func):
#定义一个内嵌的包装函数,给传入的函数加上计时的包装
def wrapper(*args,**args2): ######1#########
start = time.clock()
res=func(*args,**args2) #########2###########
end = time.clock()
print 'used:',end - start
return res
return wrapper
a=1
foo = timeit(foo)
print foo(a)
foo2 = timeit(foo2)
foo2()
相比于原代码,主要更改是标号的两行。 第一行在函数定义的时候,参数里添加了一个不定参数,以便接受原函数的参数,第二行利用了*运算符的第二个功能,将参数展开,传递给原函数。