Python-сообщество

def automaton():
    """Main loop of the automaton"""
    money = 0
    state = 0
    def next_state(state, inp):
        """
        Calculate the successor state for a given state and input.
        Args:
        state (int): a natural number, representing the automaton's current state
        inp (string): a string representing the automaton's input
        Returns:
        int: a natural number, representing the automaton's successor state
        """
        d = {'inp50':50, 'inp100':100, 'inp200':200}
        nonlocal money
        if inp[:3] == 'inp' and money < 200:
            money += d[inp]
            return 0
        elif inp[:3] == 'inp' and money > 200:
            return 0
        if money >= 150 and inp == 'reqCoke':
            money -=150
            return 1
        if money >= 200 and inp == 'reqEner':
            money -= 200
            return 2
    def output(state):
        """
        Calculate the output in a given state.
        Args:
        state (int): a natural number, representing the automaton's current state
        Returns:
        string: the output of the automaton
        """
        if state == 0:
            return str(money) + 'ct'
        elif state == 1:
            return 'COKE'
        elif state == 2:
            return 'ENERGETIC'
 
    while True:
        state = next_state(state, input('> '))
        print(output(state))
if __name__ == '__main__':
     automaton()