我有一个课程如下:
class MyClass(object):
def __init__(self):
self.foo = "foo"
self.bar = "bar"
self.methodCalls = 0 #tracks number of times any function in the instance is run
def get_foo(self):
addMethodCall()
return self.foo
def get_bar(self):
addMethodCall()
return self.bar
def addMethodCall(self):
self.methodCalls += 1
是否有调用方法而不是经常运行addMethodCall()
时调用的内置函数?
答案 0 :(得分:6)
不,类上没有挂钩来执行此操作。方法是属性 too ,虽然有些特殊,因为它们是在访问实例上的函数对象时产生的;函数是descriptors。
对方法对象的调用是生成方法对象的单独步骤:
>>> class Foo(object):
... def bar(self):
... return 'bar method on Foo'
...
>>> f = Foo()
>>> f.bar
<bound method Foo.bar of <__main__.Foo object at 0x100777bd0>>
>>> f.bar is f.bar
False
>>> stored = f.bar
>>> stored()
'bar method on Foo'
调用描述符协议是object.__getattribute__()
method的任务,所以你可以用它来查看生成方法的时间,但是你仍然需要 wrap 生成方法对象以检测调用。例如,您可以返回一个__call__
method代替实际方法的对象。
然而,使用装饰器装饰每个方法会更容易,装饰器每次调用时都会递增计数器。考虑装饰器在绑定之前应用于函数,因此您必须通过self
:
from functools import wraps
def method_counter(func):
@wraps(func)
def wrapper(self, *args, **kwargs):
self.methodCalls += 1
return func(self, *args, **kwargs)
return wrapper
您仍然需要将此应用于班级中的所有功能。您可以手动将其应用于您想要计算的所有方法:
class MyClass(object):
def __init__(self):
self.foo = "foo"
self.bar = "bar"
self.methodCalls = 0 #tracks number of times any function method is run
@method_counter
def get_foo(self):
return self.foo
@method_counter
def get_bar(self):
return self.bar
或者您可以使用元类:
import types
class MethodCounterMeta(type):
def __new__(mcls, name, bases, body):
# create new class object
for name, obj in body.items():
if name[:2] == name[-2:] == '__':
# skip special method names like __init__
continue
if isinstance(obj, types.FunctionType):
# decorate all functions
body[name] = method_counter(obj)
return super(MethodCounterMeta, mcls).__new__(mcls, name, bases, body)
def __call__(cls, *args, **kwargs):
# create a new instance for this class
# add in `methodCalls` attribute
instance = super(MethodCounterMeta, cls).__call__(*args, **kwargs)
instance.methodCalls = 0
return instance
这会照顾装饰者需要的所有内容,为您设置methodCalls
属性,因此您的课程不必:
class MyClass(object):
__metaclass__ = MethodCounterMeta
def __init__(self):
self.foo = "foo"
self.bar = "bar"
def get_foo(self):
return self.foo
def get_bar(self):
return self.bar
后一种方法的演示:
>>> class MyClass(object):
... __metaclass__ = MethodCounterMeta
... def __init__(self):
... self.foo = "foo"
... self.bar = "bar"
... def get_foo(self):
... return self.foo
... def get_bar(self):
... return self.bar
...
>>> instance = MyClass()
>>> instance.get_foo()
'foo'
>>> instance.get_bar()
'bar'
>>> instance.methodCalls
2
上面的元类只考虑函数对象(所以def
语句的结果和lambda
表达式)是类体的一部分用于装饰。它忽略任何其他可调用对象(有更多类型具有__call__
方法,例如functools.partial
个对象),以及稍后添加到类的函数。