python cookbook第三版学习笔记十七：委托属性

我们想在访问实例的属性时能够将其委托到一个内部持有的对象上，这经常用到代理机制上

class A:
    def spam(self,x):
        print("class_A:"+str(x))
    def foo(self):
        pass

class B:
    def __init__(self):
        self._a=A()
    def bar(self):
        pass
    def __getattr__(self, item):
        return getattr(self._a,item)

 

b=B()

b.bar()

b.spam(42)

运行结果：

class_A:42

在这里，当调用b.spam的时候，由于查找不到这个属性，因此调用__getattr__来查找所有的属性。在上面的代码中，在访问B中未定义的方法时就把这个操作委托给A。

 

对这种方法进行扩展一下，我们可以实现带有状态的对象或状态机。代码如下：

class connection():

    def __init__(self):

        self.new_state(ClosedConnection)

    def new_state(self,newstate):

        self.__class__=newstate

    def read(self):

        raise NotImplementedError()

    def write(self):

        raise NotImplementedError()

    def open(self):

        raise NotImplementedError

    def close(self):

        raise NotImplementedError

 

class ClosedConnection(connection):

    def read(self):

        raise RuntimeError('not open')

    def write(self,data):

        raise RuntimeError('not open')

    def open(self):

        self.new_state(OpenConnection)

    def close(self):

        raise RuntimeError('Already closed')

 

class OpenConnection(connection):

    def read(self):

        print('reading')

    def write(self,data):

        print('writing')

    def open(self):

        raise RuntimeError('Already open')

    def close(self):

        self.new_state(ClosedConnection)

 

c=connection()

print(c)

c.read()

执行结果如下，初始状态为ClosedConnection, 调用read的时候提示not open

<__main__.ClosedConnection object at 0x00000250CD984DD8>

Traceback (most recent call last):

  File "D:/py_prj/test2/cookbook.py", line 152, in <module>

    c.read()

  File "D:/py_prj/test2/cookbook.py", line 130, in read

    raise RuntimeError('not open')

RuntimeError: not open

 

c=connection()

print(c)

c.open()

print(c)

c.read()

c.write('abc')

c.close()

print(c)

调用c.open后状态转移到OpenConnection。此时调用read和write方法则可以正常调用。调用close方法后状态转移到ClosedConnection

<__main__.ClosedConnection object at 0x000001C495E94DA0>

<__main__.OpenConnection object at 0x000001C495E94DA0>

reading

writing

<__main__.ClosedConnection object at 0x000001C495E94DA0>

通过这种方法减少了在代码分支中大量使用ifelse的调用。