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


class Get_info_previous_round (WaitPage):
    """Get Endowment and Multiplier from previous application"""

    after_all_players_arrive = 'my_method'



class Calc_Info (WaitPage):
    """Get Endowment and Multiplier from previous application"""

    after_all_players_arrive = 'load_data'

    wait_for_all_groups = True



class Get_info_previous_round2 (WaitPage):
    """Get Endowment and Multiplier from previous application"""

    after_all_players_arrive = 'my_method2'



class PaymentInfo(Page):
    def vars_for_template(self):
        me = self.player
        return dict(
            my_accumulated_wealth = me.accumulated_wealth,
            my_con_other1 = me.con_other1,
            my_con_other2 = me.con_other2,
            my_con_other3 = me.con_other3,
            my_con_other4 = me.con_other4,
            my_earnings_euros = me.earnings_euros,
            my_beliefs_HH = me.beliefs_HH,
            my_beliefs_HL = me.beliefs_HL,
            my_beliefs_LH = me.beliefs_LH,
            my_beliefs_LL = me.beliefs_LL,
            my_earnings_beliefs = me.earnings_beliefs,
            my_con_other1_avg = me.con_other1_avg,
            my_con_other2_avg = me.con_other2_avg,
            my_con_other3_avg = me.con_other3_avg,
            my_con_other4_avg = me.con_other4_avg,
            my_total_earnings = me.total_earnings,
            my_svo_payoff = me.svo_payoff,
            my_svo_payoff_eur_good = me.svo_payoff_eur,
            my_total_earnings_euros = me.total_earnings_euros,
            my_beliefs_partnerchoice1 = me.beliefs_partnerchoice1,
            my_beliefs_partnerchoice2 = me.beliefs_partnerchoice2,
            my_beliefs_partnerchoice3 = me.beliefs_partnerchoice3,
            my_beliefs_partnerchoice4 = me.beliefs_partnerchoice4,
            my_coupled_HH = me.coupled_HH,
            my_coupled_HL = me.coupled_HL,
            my_coupled_LH = me.coupled_LH,
            my_coupled_LL = me.coupled_LL,
            my_bonus_units = me.bonus_units,
            my_bonus_units_beliefs_HH = me.bonus_units_beliefs_HH,
            my_bonus_units_beliefs_HL = me.bonus_units_beliefs_HL,
            my_bonus_units_beliefs_LH = me.bonus_units_beliefs_LH,
            my_bonus_units_beliefs_LL = me.bonus_units_beliefs_LL,
            my_bonus_units_beliefs_partnerchoice1 = me.bonus_units_beliefs_partnerchoice1,
            my_bonus_units_beliefs_partnerchoice2 = me.bonus_units_beliefs_partnerchoice2,
            my_bonus_units_beliefs_partnerchoice3 = me.bonus_units_beliefs_partnerchoice3,
            my_bonus_units_beliefs_partnerchoice4 = me.bonus_units_beliefs_partnerchoice4,
            my_condition = me.condition
            )




page_sequence = [Get_info_previous_round, Calc_Info, Get_info_previous_round2, PaymentInfo]