rough state machine with actions and states

statemachine.py

from abc import abstractmethod, ABC

class State(ABC):
    actions = [] # action(s) to take in this state
    next_states = [] # all states reachable from this state

    def test(self, entity):
        """
        Perform some test using the entity data
        to see if this state is valid
        """
        return True

    def next(self, entity):
        """
        Perform some test of entity data to determine which state to
        return from the transitions. The returned state is the new state for
        this entity.
        """
        return self.next_states[0]

class Action(ABC):

    @abstractmethod
    def do(self, entity):
        # perform some transformation on the
        # entity or world
        pass


class StateMachine:
    current_state = None

    def think(self, entity):
        # test all states in current_state and select.
        
        pass


class Entity:
    x_pos = 1
    y_pos = 1
    hunger = 0.5
    tiredness = 0.1