状态机的实现

 1 """
 2 状态机的实现
 3 修改实例的__class__属性
 4 """
 5 
 6 
 7 class Connection:
 8     def __init__(self):
 9         self.new_state(CloseState)
10 
11     def new_state(self, state):
12         self.__class__ = state
13 
14     def read(self):
15         raise NotImplementedError
16 
17     def write(self, data):
18         raise NotImplementedError
19 
20     def open(self):
21         raise NotImplementedError
22 
23     def close(self):
24         raise NotImplementedError
25 
26 
27 class CloseState(Connection):
28     def read(self):
29         raise RuntimeError("Not Open")
30 
31     def write(self, data):
32         raise RuntimeError("Not Open")
33 
34     def open(self):
35         self.new_state(OpenState)
36 
37     def close(self):
38         raise RuntimeError("Already close")
39 
40 
41 class OpenState(Connection):
42     def read(self):
43         print("reading")
44 
45     def write(self, data):
46         print("writing")
47 
48     def open(self):
49         raise RuntimeError("Already open")
50 
51     def close(self):
52         self.new_state(CloseState)
53 
54 
55 if __name__ == "__main__":
56     c = Connection()
57     print(c)
58     c.open()
59     print(c)
60     c.read()
61     c.close()
62     print(c)