创建一个方便使用的包装器类

时间:2019-01-30 14:44:18

标签: python wrapper

为了跟踪循环中不断更新的某些变量的当前值和先前值,我创建了一个包装器类,该类在更新时将旧值自动存储在其中:

class Updatable:
    def __init__(self, value=None, prev_value=None):
        # Initial conditions
        self._value = value
        self._prev_value = prev_value

    @property
    def value(self):
        return self._value

    @property
    def prev_value(self):
        return self._prev_value

    def update(self, new_value):
        self._prev_value = self._value
        self._value = new_value

    def has_value(self):
        return self._value is not None

    def has_prev_value(self):
        return self._prev_value is not None

但是,为了使它更像普通变量一样工作,我希望所有普通运算符都能按预期工作(除了赋值,在Python中不能重载),因此我定义了许多特殊方法,班级可以有:

    # Object representation
    def __str__(self):
        return str(self._value)

    def __repr__(self):
        return repr(self._value)

    # Iteration
    def __len__(self):
        return len(self._value)

    def __getitem__(self, key):
        return self._value[key]

    def __reversed__(self):
        return reversed(self._value)

    # Arithmetic operations
    def __add__(self, other):
        return self._value + other

    def __sub__(self, other):
        return self._value - other

    def __mul__(self, other):
        return self._value * other

    def __matmul__(self, other):
        return self._value @ other

    def __truediv__(self, other):
        return self._value / other

    def __floordiv__(self, other):
        return self._value // other

    def __mod__(self, other):
        return self._value % other

    def __divmod__(self, other):
        return divmod(self._value, other)

    def __pow__(self, other, *args):
        return pow(self._value, other, *args)

    def __lshift__(self, other):
        return self._value << other

    def __rshift__(self, other):
        return self._value >> other

    def __and__(self, other):
        return self._value & other

    def __xor__(self, other):
        return self._value ^ other

    def __or__(self, other):
        return self._value | other

    # Comparison operators
    def __eq__(self, other):
        return self._value == other

    def __ne__(self, other):
        return self._value != other

    def __lt__(self, other):
        return self._value < other

    def __gt__(self, other):
        return self._value > other

    def __le__(self, other):
        return self._value <= other

    def __ge__(self, other):
        return self._value >= other

尽管这似乎是许多特殊方法,但它与all special methods仍然相距甚远,而支持所有这些方法将意味着比我已经拥有的样板代码还要多。

此外,定义这些特殊方法不允许我将实例用作二进制运算中的第二个操作数。例如,代码

a = Updatable(2)
a + 3

有效,代码

a = Updatable(2)
3 + a

导致错误TypeError: unsupported operand type(s) for +: 'int' and 'Updatable',如果使用起来很方便,这是不希望的。

还有其他创建包装类的方法,该类与使用直接包装的数据类型(几乎)一样方便使用(忽略了以下事实:使用裸数据类型时,您将无法访问前一个除非您明确存储它的值)?

1 个答案:

答案 0 :(得分:1)

将此添加到您的班级定义中:

# Arithmetic operations
def __radd__(self, other):
    return other + self._value

def __add__(self, other):
    return self._value + other 

然后

a = Updatable(2)
3 + a

输出:

5

使用原始功能定义__add__

a = Updatable(2)
a + 3

输出:

5