在python中创建状态机

from transitions import Machine

class MyStateMachine(object):
    states = ['start', 'middle', 'end']

    def __init__(self):
        self.machine = Machine(model=self, states=MyStateMachine.states, initial='start')
        self.machine.add_transition('move_to_middle', 'start', 'middle')
        self.machine.add_transition('move_to_end', 'middle', 'end')

# 创建状态机对象
my_machine = MyStateMachine()

# 触发状态转换
my_machine.move_to_middle()
print(my_machine.state)  # 输出 'middle'
my_machine.move_to_end()
print(my_machine.state)  # 输出 'end'

import pygraphviz as pgv

# 创建有向图
graph = pgv.AGraph(directed=True)

# 添加状态节点
graph.add_node('start')
graph.add_node('middle')
graph.add_node('end')

# 添加状态转换边
graph.add_edge('start', 'middle')
graph.add_edge('middle', 'end')

# 保存图形
graph.draw('state_machine.png')