这是我们面对装饰器时遇到的第一个例子。但我无法意识到我到底想要什么。
一个名为 LOG 的简单装饰器。它应该像这样工作:
@LOG
def f(a, b=2, *c, **d):
pass
结果应该是这样的:
f(1, pippo=4, paperino='luca')
===== Enter f =====
a = 1
b = 2
pippo = 4
paperino = luca
===== Exit f =====
作为参数传递给函数的每个参数都显示其值。
我发现这个问题比我想象的要难,主要是因为您可以通过多种不同的方式将参数传递给函数(想想带有 *c 的元组或带有 **d 的字典)。
我尝试了一个解决方案,但我不确定它是否正确。有人认为是这样的:
def LOG(fn):
import inspect
varList, _, _, default = inspect.getargspec(fn)
d = {}
if default is not None:
d = dict((varList[-len(default):][i], v) for i, v in enumerate(default))
def f(*argt, **argd):
print ('Enter %s' % fn).center(100, '=')
d.update(dict((varList[i], v) for i, v in enumerate(argt)))
d.update(argd)
for c in d.iteritems():
print '%s = %s' % c
ret = fn(*argt, **argd)
print 'return: %s' % ret
print ('Exit %s' % fn).center(100, '=')
return ret
return f
我认为这并不像我预期的那么容易,但奇怪的是我没有在 Google 上找到我想要的东西。
你能告诉我我的解决方案是否可以吗?或者你能提出一个更好的解决方案来解决我提出的问题吗?
谢谢大家。