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


class ResultsWaitPage(WaitPage):
    pass


class ResultsFinal(Page):

    def vars_for_template(self):
        #
        # Payoff - Part 1
        self.player.part1_payoff = self.participant.vars['part1_payoff']
        self.player.part1_payoff_eur_to_show = self.participant.vars['part1_payoff_eur_to_show']
        #
        # Payoff - Part 2
        self.player.part2_payoff = self.participant.vars['part2_payoff']
        self.player.part2_payoff_eur_to_show = self.participant.vars['part2_payoff_eur_to_show']
        #
        # Payoff - Part 3
        self.player.part3_payoff = self.participant.vars['part3_payoff']
        self.player.part3_payoff_eur_to_show = self.participant.vars['part3_payoff_eur_to_show']
        #
        #
        self.player.total_payoff_pt1to3 = self.player.part1_payoff + self.player.part2_payoff + self.player.part3_payoff
        self.player.total_payoff_pt1to3_eur_to_show = self.player.part1_payoff_eur_to_show + self.player.part2_payoff_eur_to_show + self.player.part3_payoff_eur_to_show
        self.player.total_payoff = 5 + self.player.total_payoff_pt1to3
        self.player.total_payoff_eur_to_show = 5 + self.player.total_payoff_pt1to3_eur_to_show

    def before_next_page(self):
        self.participant.vars['total_payoff'] = self.player.total_payoff
        self.participant.vars['total_payoff_eur_to_show'] = self.player.total_payoff_eur_to_show


class End(Page):
    pass


page_sequence = [
    ResultsFinal,
    End
]