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


class Stage1_to_Stage2(Page):
    def before_next_page(self):
        self.subsession.creat_group()
        self.group.set_participant_list()
        self.player.score_num1()
        self.player.score_num2()
        self.player.score_num3()
        self.player.set_endowment_UF()
        self.player.set_endowment()
        self.group.endowment_in_stage2()
        
    
class Feedback(Page):
     def vars_for_template(self):
        return dict(
            participant_treatment = self.participant.vars['treatment'],
            participant_endowment = self.participant.vars['endowment_in_stage2']
        )

page_sequence = [Stage1_to_Stage2,Feedback]