transitions狀態(tài)機(jī)如何在python中使用

pip install transitions

# 連接協(xié)議狀態(tài)機(jī)
from transitions.extensions import HierarchicalMachine as Machine
from transitions.extensions.nesting import NestedState
 
class ConnectionStateMachine:
    """Connection state machine."""
 
    def __init__(self, callbacks=None):
        """
        :param callbacks: callbacks for the state machine
        """
        self.callbacks = {}
 
        # 定義狀態(tài)
        self.states = ["STATE_NOT_CONNECTED",
                       {
                            'name': "STATE_CONNECTED",   # 狀態(tài)名
                            'on_enter': self._on_enter_CONNECTED,  # 進(jìn)入狀態(tài)觸發(fā)
                            'on_exit': self._on_exit_CONNECTED,    # 退出狀態(tài)觸發(fā)
                            'children': [     # 狀態(tài)嵌套
                                "STATE_NOT_SELECTED",
                                {
                                    'name': "STATE_SELECTED",
                                    'on_enter': self._on_enter_CONNECTED_SELECTED
                                }
                            ]
                       }]
 
        # transition 1
        self.machine = Machine(model=self, states=self.states, initial="STATE_NOT_CONNECTED", auto_transitions=False)
 
        if callbacks:
            self.callbacks = callbacks
 
        # 定義狀態(tài)變換
        self.machine.add_transition('connect', "STATE_NOT_CONNECTED", "STATE_CONNECTED_NOT_SELECTED")  # transition 2
        self.machine.add_transition('disconnect', "STATE_CONNECTED", "STATE_NOT_CONNECTED")  # transition 3
        self.machine.add_transition('select', "STATE_CONNECTED_NOT_SELECTED", "STATE_CONNECTED_SELECTED")  # transition 4
        self.machine.add_transition('deselect', "STATE_CONNECTED_SELECTED", "STATE_CONNECTED_NOT_SELECTED")  # transition 5
        self.machine.add_transition('timeoutT7', "STATE_CONNECTED_NOT_SELECTED", "STATE_NOT_CONNECTED")  # transition 6
 
    # 事件觸發(fā)的動作
    def _on_enter_CONNECTED(self):
        if "on_enter_CONNECTED" in self.callbacks:
            self.callbacks["on_enter_CONNECTED"]()
 
    def _on_exit_CONNECTED(self):
        if "on_exit_CONNECTED" in self.callbacks:
            self.callbacks["on_exit_CONNECTED"]()
 
    def _on_enter_CONNECTED_SELECTED(self):
        if "on_enter_CONNECTED_SELECTED" in self.callbacks:
            self.callbacks["on_enter_CONNECTED_SELECTED"]()

# 一般都是兩種：不嵌套、嵌套
from transitions import Machine   #不嵌套
from transitions.extensions import HierarchicalMachine as Machine # 嵌套
 
# 檢查狀態(tài)
machine.state
machine.is_<state_name>()

# 狀態(tài)可以是三種類型: 對象、字符串、字典
from transitions import State
states = [
    State(name="solid"),    # 對象
    "liquid",               # 字符串
    {"name": "gas"}         # 字典
]

# 定義函數(shù)
machine.add_transition(trigger, source, dest, )
trigger(str)         方法名,觸發(fā)狀態(tài)轉(zhuǎn)換
source(str or list)  源狀態(tài)
dest(str)            目標(biāo)狀態(tài)
 
# 加上變換
machine.add_transition(trigger="wake_up", source="asleep", dest="hanging out")
machine.add_transition('work_out', 'hanging out', 'hungry')
machine.add_transition('nap', '*', 'asleep')   # 從任意狀態(tài)變?yōu)閍sleep