class FactorFloat(float):
def _factor_scale(f):
def wrapper(self, *args, **kwargs):
scaled = float.__mul__(self, FACTOR)
result = f(scaled, *args, **kwargs)
# if you want to return FactorFloats when possible:
if isinstance(result, float):
result = type(self)(result/FACTOR)
return result
return wrapper
def __repr__(self):
return '%s(%s)' % (type(self).__name__, float.__repr__(self))
__str__ = _factor_scale(float.__str__)
__mul__ = _factor_scale(float.__mul__)
__div__ = _factor_scale(float.__div__)
__add__ = _factor_scale(float.__add__)
__sub__ = _factor_scale(float.__sub__)
f = FactorFloat(3.)
FACTOR = 10.
print f # 30.0
print f-1 # 29.0
FACTOR = 2.
print f # 6.0
print f-1 # 5.0
print repr(f)
для:
30.0
29.0
6.0
5.0
FactorFloat(3.0)
РЕДАКТИРОВАТЬ:
В ответ на вопрос в комментарии;сделать вещи немного более общими и автоматизированными, используя декоратор классов.Я бы не зацикливался на dir(baseclass)
, а вместо этого явно перечислял бы методы, которые я хотел обернуть.В приведенном ниже примере я перечисляю их в переменной класса _scale_methods
.
def wrap_scale_methods(cls):
Base = cls.__base__
def factor_scale(f):
def wrapper(self, *args, **kwargs):
scaled = Base.__mul__(self, FACTOR)
result = f(scaled, *args, **kwargs)
if isinstance(result, Base):
result = type(self)(result/FACTOR)
return result
return wrapper
for methodname in cls._scale_methods:
setattr(cls, methodname, factor_scale(getattr(Base, methodname)))
return cls
@wrap_scale_methods
class FactorComplex(complex):
_scale_methods = '__str__ __mul__ __div__ __add__ __sub__'.split()
def __repr__(self):
return '%s(%s)' % (type(self).__name__, complex.__repr__(self)[1:-1])