from gamelib.gamestate import Game


class ReplGame(object):
    def __init__(self, init_data=None):
        self.game = Game(init_data=init_data)
        self._begin_turn()

    def _begin_turn(self):
        self.game.start_turn()
        self.display_state()

    def display_state(self):
        print "Game:", self.game
        print "Science:"
        for science in self.game.lab.science:
            print " %s %s (%s)" % (
                "*" if science.can_spend(self.game.lab) else "-",
                science.NAME, science.points)

    def next_turn(self, research_list, missions):
        self.game.cur_allocation.extend(research_list)
        self.game.cur_missions.extend(missions)
        self.game.end_turn()
        self._begin_turn()