Python State Machine - Simple State Machine

simple_device.py

from states import LockedState

class SimpleDevice(object):
    """ 
    A simple state machine that mimics the functionality of a device from a 
    high level.
    """

    def __init__(self):
        """ Initialize the components. """

        # Start with a default state.
        self.state = LockedState()

    def on_event(self, event):
        """
        This is the bread and butter of the state machine. Incoming events are
        delegated to the given states which handle the state.
        """

        # The next state will be the result of the on_event function.
        self.state = self.state.on_event(event)