前段时间,在学习函数装饰器的过程中,一直想实现python跟踪某个函数调用次数的功能。但自己技艺不精,虽然实现了,但总是感觉不满意。
最近,重新翻看《python学习手册》,发现有此实现,特意附上供大家一起学习:
>>> class tracer:
def __init__(self, func):
self.calls = 0
self.func = func
def __call__(self, *args, **keys):
self.calls += 1
print('call %s to %s' % (self.calls, self.func.__name__))
self.func(*args, **keys)
# python在定义此函数时,便会将spam作为参数传入__init__
# 类似于spam = tracer(spam)
# 此时的spam已经指向tracer(spam)对象了
>>> @tracer
def spam(a, b, c):
print(a, b, c)
# 此时调用spam函数,便相当于执行__call__(spam, *(1, 2, 3))
# 这里有运算符重载在起作用。此时的spam对象已经有calls属性了
>>> spam(1, 2, 3)
call 1 to spam
1 2 3
>>> spam('a', 'b', 'c')
call 2 to spam
a b c
# 可以通过calls属性查看其调用次数
>>> spam.calls
2
哈哈,以上就是python小工具关于python如何实现对函数调用次数的跟踪的实现,此出自《python学习手册》第31章,欢迎大家关注:python小工具。