from ._builtin import Page, WaitPage
from .models import Constants

class Post_Task(WaitPage):
    wait_for_all_groups = True

    def is_displayed(self):
        return self.round_number == 1     

    #def before_next_page(self):
        #self.player.s1_player_payoff = self.participant.vars['s1_player_payoff']
        #self.player.player_payoff = self.player.participant.vars['player_payoff']

class Final_Results(Page):

    def is_displayed(self):
        return self.round_number == 1 

    #def vars_for_template(self):
        #self.player.s1_player_payoff = float(self.participant.vars['s1_player_payoff'])
        #self.player.player_payoff = float(self.player.participant.vars['player_payoff']) 
        #self.player.total_game_payoff = (self.player.player_payoff + self.player.s1_player_payoff)
        #self.player.final_payoff = self.player.total_game_payoff + 3

        #return {
            #'s1_player_payoff': self.player.s1_player_payoff,
            #'player_payoff': self.player.player_payoff,
            #'final_payoff_rou': round(self.player.final_payoff, 2)
        #}   

class comp_1(Page):
    form_model = 'player'
    form_fields = ['serv_1a','serv_1b','serv_1c']

    def is_displayed(self):
        return self.round_number == 1           

class comp_2(Page):
    form_model = 'player'
    form_fields = ['serv_2','serv_3','serv_4','serv_5','serv_6','serv_7a','serv_7b','serv_7c','serv_8','serv_9','serv_10']

    def is_displayed(self):
        return self.round_number == 1                      

class Final_Hold(Page):

    def is_displayed(self):
        return self.round_number == 1     

    # Pass on the variable question_correct (or score) to next app sequence, this is needed to calculate final payments     

page_sequence = [Post_Task, Final_Results, comp_1, comp_2, Final_Hold]