如何编写适应成员类接口的容器/包装类?

时间:2011-05-24 12:58:55

标签: python containers wrapper

我正在编写一个包装另一个类的对象的类。打算改变它的一些方法行为,同时能够扩展其所有其他接口。我没有使用继承,因为内部类对象可能会死亡,外部类需要能够在不破坏自身的情况下用实时替换它。

所以我有:

class Inner():
    def foo(): pass
    def goo(): pass

class Outer():
    self.inner = InnerFactory(innerType)
    def foo(): 
         try:
             self.inner.foo() 
         except:
             del self.inner
             self.inner = InnerFactory(innerType)
             self.inner.foo()

问题是如何扩展goo w / o明确重写,因为我可能有很多其他我不知道的方法。

实际上,在阅读了下面的一些反馈之后,我意识到我没有使用强大的功能 getattr 。但是,我并不完全理解为什么下面的建议似乎都使用了这么复杂的版本。为什么不能这么简单:

def __getattr__( self, name ):
    if self.inner:
          return getattr( self.inner, name )
    else:
          raise Exception( 'attribute %s not found' % name ) 

2 个答案:

答案 0 :(得分:2)

下面的代码就像你想要的那样,但是:1)它很难看; 2)它不是线程安全的; 3)它会进入一个循环,直到Inner的某个方法引发异常(这不是由于实现,而是由于发布了初始想法); 4)避免使用它的更多理由:)

class Inner:
  def foo(self):
    print "foo"
  def bar(self):
    print "bar"

class Outer:
  def __init__(self):
    self.inner = Inner()

  def __getattr__(self, name):
    ret = getattr(self.inner, name)
    def wrapper(*args):
      try:
        ret(*args)
      except:
        del self.inner
        self.inner = Inner()
        updated = self.__getattr__(name)
        updated(*args)

    return wrapper

  def blah(self):
    print "Blah"

outer = Outer()

outer.foo()
outer.bar()
outer.blah()
outer.nosuchattr()

答案 1 :(得分:0)

我的解决方案与@khachik类似,加上一些方法缓存。

  • 小心使用__ getattr__很容易进入无限循环。
  • 您还可以根据需要添加线程锁

未经测试的代码将其视为伪代码。

class Outer(object):
    def __init__(self):
        self.method_cache = {}
        self.methods_to_override = ['foo', 'goo']

    def __getattr__(self, method_name):
        if method_name in self.methods_to_override:
            if method_name in self.method_cache:
                return self.method_cache[method_name]
            else:
                def wrapper(*args, **kw):
                    wrapped = getattr(self.inner, method_name)
                    try:
                        return wrapped(*args, **kw)
                    except InnerDiedError:
                        self.inner = self.InnerFactory(innerType)
                        wrapped = getattr(self.inner, method_name)
                        return wrapped(*args, **kw)

                self.method_cache[method_name] = wrapper
                return wrapper