Question

我想创建一个包装类，其行为与包装对象完全相同（具有一些特定的例外）。我目前遇到的问题是内置函数。我怎样才能将内置函数重定向到包装对象？

class Wrapper:
    def __init__(self, wrapped):
        object.__setattr__(self, '_wrapped', wrapped)
    def __getattr__(self, name):
        return getattr(object.__getattribute__(self, '_wrapped'), name)

class Foo:
    def __init__(self, val):
        self.val = val
    def __abs__(self):
        return abs(self.val)

foo = Wrapper(Foo(-1))
print(foo.val) # Okay
print(abs(foo)) # TypeError: bad operand type for abs(): 'Wrapper'

Answer 1

您可以动态创建一个新类，它是Wrapper和Foo的子类，因此您将拥有所需的所有属性：

class Wrapper:
    def __new__(self, wrapped):
        cls = type(wrapped)
        new_type = type(cls.__name__ + '_wrapped', (Wrapper, cls), {})
        return object.__new__(new_type)

    def __init__(self, wrapped):
        self._wrapped = wrapped
    def __getattr__(self, name):
        return getattr(self._wrapped, name)

现在你可以这样做：

>>> foo = Wrapper(Foo(-1))
>>> abs(foo)
1
>>> type(foo)
<class '__main__.Foo_wrapped'>

PS：

object.__getattr__和__setattr__函数中的__init__（或__getattr__）不需要获取和设置此属性。
您可能希望缓存此操作以避免在每个新对象上创建新类。

将内置函数重定向到包装类

1 个答案: