1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
|
import dataclasses
import typing
from automat import TypeMachineBuilder
class Inputs(typing.Protocol):
def behavior1(self) -> None: ...
def behavior2(self) -> None: ...
class Nothing: ...
builder = TypeMachineBuilder(Inputs, Nothing)
start = builder.state("start")
@start.upon(Inputs.behavior1).loop()
def one(inputs: Inputs, core: Nothing) -> None:
print("starting behavior 1")
inputs.behavior2()
print("ending behavior 1")
@start.upon(Inputs.behavior2).loop()
def two(inputs: Inputs, core: Nothing) -> None:
print("behavior 2")
machineFactory = builder.build()
machineFactory(Nothing()).behavior1()
|