simplestate 0.1.0

A simple stupid finite state machine.

simplestate

Simplestate helps developers create and manage finite state machines in the simplest way possible—focusing solely on states and transitions, leaving out complex event systems.

Features

• Focus on states and transitions.
• Minimalistic and easy to use.
• Supports entering-state handlers for additional functionality.

Installation

To install simplestate, use poetry:

poetry add simplestate

Usage Example

Here's a basic example of using simplestate:

from simplestate import StateMachine

# Define the machine with an initial state
m = StateMachine("loading")

# Define transitions
m["loading"] + "error" >> "failed"   # Transition from loading to failed on "error"
m["loading"] + "ok" >> "success"     # Transition from loading to success on "ok"
m["?"] + "back" >> "loading"         # Any state transitions to loading on "back"

# Add handlers for entering states
m.add_callbacks({
    "loading": lambda previous: print(f"{previous} >> loading"),
    "failed": lambda previous, error: print(f"{previous} >> failed: {error}"),
    "success": lambda previous: print(f"{previous} >> success"),
})

# Start the machine
m.start()  # adjust inital state as you wish i.e. m.start(at_state="ok")
assert m.current == "loading"

# Trigger transitions
m.handle("error", error="Something went wrong")
assert m.current == "failed"

m.handle("back")
assert m.current == "loading"

Why Simplestate?

The goal of simplestate is to keep finite state machines simple and focused. By eliminating complex event systems, you can easily manage states and transitions while keeping the code testable and maintainable.

Event Handling on State Entry

Simplestate allows event handling when entering a state using .add_callbacks(). These callbacks take previous and an optional **input_context.

FAQ

Q: How can I handle state exit events?
A: You can use the entering handler for the next state to handle actions when leaving the previous state.

def on_state_x(previous, **input_context):
    if previous == "a":
        do_this()
    elif previous == "b":
        do_that()

Q: Can I represent the state machine as a class?
A: Yes! Think of simplestate as a state manager. You can compose it within your own class for handling side effects:

class TrafficLight:
    def __init__(self):
        brain = Stateachine("red")
        ["red"] + "next" >> "yellow_green"
        m["yellow_green"] + "next" >> "green"
        m["green"] + "next" >> "yellow_red"
        m["yellow_red"] + "next" >> "red"
        m.start("red")

        self.machine = m
    
    def display(self) -> str:
        return self.machine.current
    
    def next(self) -> None:
        self.machine.handle("next")

Tests

To run the tests:

poetry install --with test
poetry run pytest

Contribution

Contributions are welcome! Feel free to open an issue or submit a pull request.

1. Install devcontainer
2. Open this project in dev container
3. Run poetry shell
4. Run poetry install --with test
5. Run poetry run pytest

License

This project is licensed under a free, open-source license.