如何将所有参数传递给装饰器?

时间:2009-03-09 10:51:19

标签: python decorator

我尝试使用装饰器跟踪某些方法的执行情况。这是装饰器代码:

def trace(func):
    def ofunc(*args):
        func_name = func.__name__       
        xargs = args        
        print "entering %s with args %s" % (func_name,xargs)
        ret_val = func(args)
        print "return value %s" % ret_val
        print "exiting %s" % (func_name)
    return ofunc 

问题是,如果我尝试将此装饰器应用于方法,则不会发送self参数。你能告诉我为什么,我该如何解决这个问题?

1 个答案:

答案 0 :(得分:6)

此行不正确:

ret_val = func(args)

当您传递参数列表时,您忘记展开参数列表。它应该是:

ret_val = func(*args)

具有此修改的示例输出:

>>> class Test2:
...     @trace
...     def test3(self, a, b):
...        pass
... 
>>> t = Test2()
>>> t.test3(1,2)
entering test3 with args (<__main__.Test2 instance at 0x7ff2b42c>, 1, 2)
return value None
exiting test3
>>> 

如果您将装饰器扩展为也使用关键字参数,那么在使用**传递它们时,您还需要适当地展开它们。