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


class Results(Page):

    def vars_for_template(self):
        self.player.id_random = int(self.participant.vars['id_random'])
        self.player.supergroup_task1 = int(self.participant.vars['supergroup_task1'])
        self.player.supergroup_task2 = int(self.participant.vars['supergroup_task2'])
        self.player.supergroup_task3 = int(self.participant.vars['supergroup_task3'])
        self.player.payoff_final_PD_task1 = self.participant.vars['payoff_final_PD_task1']
        self.player.payoff_final_PD_task2 = self.participant.vars['payoff_final_PD_task2']
        self.player.payoff_final_PD = self.player.payoff_final_PD_task1 + self.player.payoff_final_PD_task2
        self.player.payoff_final_DG = self.participant.vars['payoff_final_DG']
        self.player.payoff_final_PD_to_show = round(self.player.payoff_final_PD, ndigits=2)
        self.player.payoff_final_DG_to_show = round(self.player.payoff_final_DG, ndigits=2)

        self.player.payoff_overall = self.player.payoff_final_PD + self.player.payoff_final_DG + Constants.show_up_fee
        self.player.payoff_overall_to_show = round(self.player.payoff_overall, ndigits=2)
        self.participant.vars['payoff_overall'] = self.player.payoff_overall

        return{'payoff_final_PD_to_show': self.player.payoff_final_PD_to_show,
               'payoff_final_DG_to_show': self.player.payoff_final_DG_to_show,
               'payoff_overall_to_show': self.player.payoff_overall_to_show,
               }


page_sequence = [
    Results
]