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


class EndOfGame(Page):

    def before_next_page(self):
        self.player.set_payoffs()


class ResultsWaitPage(WaitPage):
    def after_all_players_arrive(self):
        pass


class Results(Page):

    def vars_for_template(self):
        payoff1 = self.participant.vars['payoff_1']
        payoff2 = self.participant.vars['payoff_2']
        random_choice = self.participant.vars['random_task'] + 1
        total_points = self.player.final_payoff
        conversion_rate = Constants.points_to_money
        show_up_fee = Constants.show_up_fee
        final_money_payoff = self.participant.payoff
        return dict(
            payoff1=payoff1,
            payoff2=payoff2,
            random_choice=random_choice,
            total_points=total_points,
            conversion_rate=conversion_rate,
            show_up_fee=show_up_fee,
            final_money_payoff=final_money_payoff
        )


page_sequence = [EndOfGame, Results]