Python设计模式-状态模式

zx钟

发布于 2019-07-18 21:51:11

7250

发布于 2019-07-18 21:51:11

文章被收录于专栏：测试游记测试游记

状态模式可以看作是在运行时改变对象行为的一种方式。

假设有一台笔记本电脑，它有以下几个状态：

开机「On」
关机「Off」
挂起「Suspend」
休眠「Hibernate」

封装对象行为的接口

首先需要一个行为与对象状态关联的接口

它有两个属性：

状态
允许的状态

它有一个方法：

切换状态

切换状态的时候会使用self.__class__将它动态切换到状态对应的对象，这个稍后再看。

class ComputerState(object):
    name = "状态"
    allowed = []

    def switch(self, state):
        if state.name in self.allowed:
            print(f"状态由「{self}」切换到新的状态「{state.name}」。")
            self.__class__ = state
        else:
            print(f"状态由「{self}」切换到新的状态{state.name}是不可能的。")

    def __str__(self):
        return self.name

定义几种状态

name对应当前类的状态
allowed对应当前状态支持的转换状态

class Off(ComputerState):
    name = '关闭'
    allowed = ['打开']


class On(ComputerState):
    name = '打开'
    allowed = ['关闭', '休眠', '挂起']


class Suspemd(ComputerState):
    name = '挂起'
    allowed = ['打开']


class Hibernate(ComputerState):
    name = '休眠'
    allowed = ['打开']

定义一个计算机

class Computer(object):
    def __init__(self):
        self.state = Off()

    def change(self, state):
        self.state.switch(state)

一个计算机的初始状态就是关机状态咯，所以实例化的时候把Off() 赋给self.state

测试

if __name__ == '__main__':
    computer = Computer()
    computer.change(On)
    computer.change(Off)
    computer.change(On)
    computer.change(Suspemd)
    computer.change(Hibernate)
    computer.change(On)
    computer.change(Off)

实例化一下之后，