0

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

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'
4

1 回答 1

3

您可以动态创建一个新类，它是 and 的子类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__and__getattr__函数中获取和设置此属性。
  • 您可能希望缓存此操作以避免在每个新对象上创建一个新类。
于 2012-10-21T20:37:42.950 回答