example.py

import state as StateManagment


class Game:
    def __init__(self):
        self.states = ("state_1", "state_2")

        state_dict = {
            self.states[0]: StateManagment.ExampleState1,
            self.states[1]: StateManagment.ExampleState2
        }
        self.game_state = StateManagment.StateMachine(self, state_dict)

        self.game_state.change('state_1')

    @staticmethod
    def close():
        exit()

    def print_state_list(self):
        print("*" * 20)
        print("States list:")
        for s in self.states:
            print("--" + s)
        print("*" * 20)

    def run(self):
        print("Welcome!")
        self.print_state_list()
        while True:
            self.update()
            print()
            self.draw()

    def update(self):
        self.game_state.update()

    def draw(self):
        self.game_state.draw()


if __name__ == "__main__":
    g = Game()
    try:
        g.run()
    except KeyboardInterrupt:
        print("\nPress enter to exit..")
        input()