from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants


class GameOutcome(Page):
    def vars_for_template(self):
        move_label = {0: 'None', 1: 'A', 2: 'B'}

        data = {'games': []}
        for i, game in enumerate(self.participant.vars['payment_games']):
            data['games'].append({
                'round': game['round'],
                'game': game['game'],
                'payoff': game['payoff'],
                'random_time': game['random_time'],
                'row_move': move_label[game['row_move']],
                'column_move': move_label[game['column_move']],
                'payoff_dollars': game['payoff'].to_real_world_currency(self.session),
            })

            data['final_payoff'] = self.participant.payoff.to_real_world_currency(self.session)

        return data


page_sequence = [
    GameOutcome,
]