如何系统地和透明地覆盖Python方法

时间:2011-01-19 12:42:55

标签: python inheritance override transform

我正在寻找一种“简单”的解决方案来系统地覆盖Python中的一些继承方法(=> 2.6)。根据我之前的问题here,我会找到一个解决方案:

def override(cls, o):
"""Override class method(s)."""
for t in o: # possible problem; nondeterministic order
    for name in o[t]:
        mtbo= getattr(cls, name).im_func
        om= t(mtbo)
        om.__name__= mtbo.__name__
        om.__doc__= mtbo.__doc__
        # What additional magic is needed here to act as 'genuine' method of super class?
        setattr(cls, name, om)

if __name__== '__main__':
    class B(object):
        def f1(self, val):
            """Doc of B.f1"""; print '1: ', val
        def f2(self, val):
            """Doc of B.f2"""; print '2: ', val
    class A(B):
        pass

    def t(f, msg):
        def g(self, *args, **kwargs):
            print msg, ' entering'; result= f(self, *args, **kwargs)
        return g
    t1= lambda f: t(f, 't1'); t2= lambda f: t(f, 't2')
    override(A, {t1: ['f1', 'f2'], t2: ['f2']})

    def tst(c):
        c.f1(1); print c.f1.__name__, c.f1.__doc__
        c.f2(2); print c.f2.__name__, c.f2.__doc__
    tst(B()), tst(A())

这对我(目前)的目的来说似乎运作良好。但是我希望能够覆盖尽可能透明,因此我将保留超类方法名称和doc。现在我的具体问题是:我应该保留其他什么吗?你有什么解决方案?

更新 我想现在这个问题有更广泛的影响:我问我应该(最初)问如何装饰方法,或者功能合理的pythonic方式。

1 个答案:

答案 0 :(得分:2)

您可以调用__name__来确保复制相关属性,而不是手动分配__doc__functools.update_wrapper和其他可能的人。{/ p>