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.task1_supergroup = int(self.participant.vars['task1_supergroup'])
        self.player.task2_supergroup = int(self.participant.vars['task2_supergroup'])
        self.player.task1_payoff_final_PD = self.participant.vars['task1_payoff_final_PD']
        self.player.task2_payoff_final_PD = self.participant.vars['task2_payoff_final_PD']
        self.player.payoff_final_PD = self.player.task1_payoff_final_PD + self.player.task2_payoff_final_PD
        self.player.payoff_final_PD_to_show = round(self.player.payoff_final_PD, ndigits=2)

        self.player.payoff_overall = self.player.payoff_final_PD + self.player.show_up_fee + self.player.quest_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_overall_to_show': self.player.payoff_overall_to_show,
               }


page_sequence = [
    Results
]