调用另一个方法时运行Python类方法

时间:2016-06-15 10:03:35

标签: python python-2.7 class

我有一个课程如下:

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()时调用的内置函数?

1 个答案:

答案 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个对象),以及稍后添加到的函数。