如何让每个Class Method在执行前调用指定的方法?

时间:2010-06-08 15:50:24

标签: python

我想让我的Python类的行为方式是,当调用任何Class方法时,首先执行默认方法,而不在被调用的Class中明确指定它。一个例子可能会有所帮助:)

Class animals:
    def _internalMethod():
        self.respires = True

    def cat():
        self._internalMethod()
        self.name = 'cat'

    def dog():
        self._internalMethod()
        self.name = 'dog'

我希望在从动物实例调用任何方法时自动调用_internalMethod(),而不是在每个方法的def中明确说明它。是否有一种优雅的方式来做到这一点?

干杯,

1 个答案:

答案 0 :(得分:5)

您可以使用元类和getattribute动态修饰所有方法(如果您使用的是Python 2,请务必从object继承!)。

另一种选择就是在课堂上进行修正,例如:

def add_method_call(func, method_name):
    def replacement(self, *args, **kw):
        getattr(self, method_name)()
        return func(self, *args, **kw)
    return replacement

def change_all_attrs(cls, added_method):
    for method_name in dir(cls):
        attr = getattr(cls, method_name)
        if callable(attr):
            setattr(cls, method_name, add_method_call(attr, added_method))

class animals(object):
    ...

change_all_attrs(animals, '_internalMethod')

这有点草率,dir()不会在超类中获取任何方法,并且由于简单的callable(attr)测试,您可能会捕获您不想要的属性和其他对象。但它可能适合你。

如果使用Python 2.7+,您可以在创建类后使用类装饰器而不是调用change_all_attrs,但效果是相同的(除非您必须重写change_all_attrs才能使其成为装饰)。