Enum campaign machine

A StateChart that demonstrates declaring States from Enum types as source for States definition.

from enum import Enum

from statemachine.states import States

from statemachine import StateChart


class CampaignStatus(Enum):
    DRAFT = 1
    PRODUCING = 2
    CLOSED = 3


class CampaignMachine(StateChart):
    "A workflow machine"

    states = States.from_enum(
        CampaignStatus,
        initial=CampaignStatus.DRAFT,
        final=CampaignStatus.CLOSED,
    )

    add_job = states.DRAFT.to(states.DRAFT) | states.PRODUCING.to(states.PRODUCING)
    produce = states.DRAFT.to(states.PRODUCING)
    deliver = states.PRODUCING.to(states.CLOSED)

Asserting campaign machine declaration

assert CampaignMachine.states.DRAFT.initial
assert not CampaignMachine.states.DRAFT.final

assert not CampaignMachine.states.PRODUCING.initial
assert not CampaignMachine.states.PRODUCING.final

assert not CampaignMachine.states.CLOSED.initial
assert CampaignMachine.states.CLOSED.final

Testing our campaign machine

sm = CampaignMachine()
res = sm.send("produce")

assert CampaignStatus.DRAFT not in sm.configuration_values
assert CampaignStatus.PRODUCING in sm.configuration_values
assert CampaignStatus.CLOSED not in sm.configuration_values
assert CampaignStatus.PRODUCING in sm.configuration_values