python:如何将实例的方法调用转发给它的属性

时间:2012-07-24 10:50:37

标签: python python-2.7

class Number(object):
    def __init__(self):
        super(Number, self).__init__()
        self.data = 10

    def __getattr__(self, name):
        def _missing(*args, **kwargs):
            method = getattr(self.data, name)
            return method(args[0])

        return _missing

a = Number()
b = Number()

print a.__add__(10)   # this is ok!
print a + 10          # TypeError: "unsupported operand type(s) for +: 'Number' and 'int'"
print a + b           # TypeError: "unsupported operand type(s) for +: 'Number' and 'Number'"

问题:     “a .__ add __(10)”和“a + 10”之间的区别是什么,如何挂钩运算符“+”?

2 个答案:

答案 0 :(得分:4)

Python只会使用实际的__add__方法,而不是仅因__getattr__而“存在”的方法。

添加__add__ = (10).__add__时效果很好。

所以你要做的就是添加代理方法:

def __add__(self, *args): return self.data.__add__(*args)
def __sub__(self, *args): return self.data.__sub__(*args)
# ...

答案 1 :(得分:1)

你可以试试这个,但它并不漂亮:

import numbers

def redirect(name):
    def redirected(self, *args):
        assert len(args) <= 1
        if len(args) == 1 and isinstance(args[0], Number):
            return getattr(self._data, name)(args[0]._data)
        else:
            return getattr(self._data, name)(*args)
    return redirected

names = set(['__gt__', '__ge__'])
names.update(numbers.Real.__abstractmethods__)
methods = dict((name, redirect(name)) for name in names)
methods.update(
    {'__init__': lambda self, data: setattr(self, '_data', float(data))})
Number = type("Number", (), methods)

你可以像这样使用它:

>>> a = Number(5)
>>> b = Number(7)
>>> a + 10
15.0
>>> a + b
12.0
>>> a > b
False
>>> a == Number(5.0)
True

请注意,算术运算符返回的类型为float而不是Number(可能与您预期的一样,也可能不是。)

如果您想拥有整数,可以移除float()来电并将numbers.Real更改为numbers.Integral。 但请注意,算术运算符将停止使用浮点值。