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”之间的区别是什么,如何挂钩运算符“+”?
答案 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
。
但请注意,算术运算符将停止使用浮点值。