用学到的新知识,可以写一个记录函数参数的装饰器。为简单起见,仅打印到标准输出:
>>> def logger(func):
... def inner(*args, **kwargs): #1
... print "Arguments were: %s, %s" % (args, kwargs)
... return func(*args, **kwargs) #2
... return inner
*args代表任意数量的参数,**kwargs表示任意数量的键值对和字典
*
符号也可以用在函数调用时,在这里它也有类似的意义。在调用函数时,以 *
开头的变量表示该变量内容需被取出用做位置参数。
>>> def add(x, y):
... return x + y
>>> lst = [1,2]
>>> add(lst[0], lst[1]) # 1
3
>>> add(*lst) # 2
3
注意在 #1
处函数 inner
接收任意数量和任意类型的参数,然后在 #2
处将他们传递给被包装的函数。这样一来我们可以包装或装饰任意函数,而不用管它的签名。
>>> @logger
... def foo1(x, y=1):
... return x * y
>>> @logger
... def foo2():
... return 2
>>> foo1(5, 4)
Arguments were: (5, 4), {}
20
>>> foo1(1)
Arguments were: (1,), {}
1
>>> foo2()
Arguments were: (), {}
2
每一个函数的调用会有一行日志输出和预期的返回值。