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


class P01_Welcome(Page):
    def is_displayed(self):
        return self.round_number == 1


class P02_T1_Inst(Page):
    def before_next_page(self):
        self.group.set_roles()

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


class P03_T1_Inst1(Page):

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


class P04_T1_Inst2(Page):
    def is_displayed(self):
        return self.round_number == 1


class  P07_T1_Control(Page):
    form_model = 'player'
    form_fields = ["control1"]
    def is_displayed(self):
        return self.round_number == 1
    
class P09_T1_Role(Page):
    def is_displayed(self):
        return self.round_number == 1


class P17_T2_Role(Page):
    def is_displayed(self):
        return self.round_number == 1


class W03_T1_ShuffleWait(WaitPage):
    wait_for_all_groups = True
    def after_all_players_arrive(self):
        self.subsession.group_randomly(fixed_id_in_group=True)

    def is_displayed(self):
        return self.round_number >= 2


class W04_T1_GroupWait(WaitPage):
    def after_all_players_arrive(self):
         self.group.T2_offer()


class P17_T1_Teilnehmer1(Page):
    form_model = 'player'
    form_fields = ["buyer_choice"]
    def vars_for_template(self):
        product1 = self.group.get_player_by_id(1).T2_product1
        product2 = self.group.get_player_by_id(1).T2_product2
        payoff1 = self.group.get_player_by_id(1).T2_payoff1
        payoff2 = self.group.get_player_by_id(1).T2_payoff2
        period = self.round_number
        return dict(
            product1=product1,
            product2=product2,
            payoff1=payoff1,
            payoff2=payoff2,
            period=period
        )
    def is_displayed(self):
        return self.player.type == "1"


class W06_T1_PayoffWait(WaitPage):
    def after_all_players_arrive(self):
        self.group.T2_payoff_calc()


class P19_T1_Feedback(Page):
    def vars_for_template(self):
        buyer_choice = self.group.get_player_by_id(1).buyer_choice
        buyer = self.group.get_player_by_id(1).T2_payoff
        TC = self.group.get_player_by_id(2).T2_payoff
        period = self.round_number
        return dict(
            buyer_choice=buyer_choice,
            buyer=buyer,
            TC=TC,
            period=period
                    )


class P20_T2_Intro(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds


class P21_T2_Q1(Page):
    form_model = 'player'
    form_fields = ['age',
                   'sex',
                   'experience',
                   'studies',
                   'semester']

    def is_displayed(self):
        return self.round_number == Constants.num_rounds


class P22_T2_Q2(Page):
    form_model = 'player'
    form_fields = ['markets1',
                   'markets2',
                   'markets3',
                   'markets4',
                   'markets5',
                   'markets6']

    def is_displayed(self):
        return self.round_number == Constants.num_rounds


class P23_T2_Q31(Page):
    form_model = 'player'
    form_fields = ['verstanden',
                   'quality_check',
                   'not_harming',
                   'ownpayoff',
                   'utilitarian']

    def is_displayed(self):
        return self.round_number == Constants.num_rounds and self.player.type == '1'


class P23_T2_Q32(Page):
    form_model = 'player'
    form_fields = ['verstanden', 'quality_check']

    def is_displayed(self):
        return self.round_number == Constants.num_rounds and self.player.type == '2'


class P24_T2_Q4(Page):
    form_model = 'player'
    form_fields = ['partei', 'pol_selbstein']

    def is_displayed(self):
        return self.player.round_number == Constants.num_rounds


class P25_T2_Q5(Page):
    form_model = 'player'
    form_fields = ['allocation1', 'allocation2', 'allocation3', 'allocation4', 'allocation5', 'allocation6']

    def is_displayed(self):
        return self.player.round_number == Constants.num_rounds


class W07_T3_ResultsWait(WaitPage):
    def after_all_players_arrive(self):
        self.group.set_payoff()
    def is_displayed(self):
        return self.round_number == Constants.num_rounds


class P26_T3_Results(Page):
    def vars_for_template(self):
        part2_T1 = self.group.get_player_by_id(1).in_round(self.group.round_payed).T2_payoff
        part2_T2 = self.group.get_player_by_id(2).in_round(self.group.round_payed).T2_payoff
        final_payoff = self.player.payoff.to_real_world_currency(self.session)
        return dict(
            part2_T1=part2_T1,
            part2_T2=part2_T2,
            final_payoff=final_payoff
        )

    def is_displayed(self):
        return self.round_number == Constants.num_rounds


class P27_T3_Payment(Page):
    form_model = 'player'
    form_fields = ['name', 'surname', 'method']

    def is_displayed(self):
        return self.round_number == Constants.num_rounds


class P28_T3_PaymentB(Page):
    form_model = 'player'
    form_fields = ['IBAN']

    def is_displayed(self):
        return self.player.method == "Banküberweisung" and self.round_number == Constants.num_rounds


class P28_T3_PaymentP(Page):
    form_model = 'player'
    form_fields = ['mail']

    def is_displayed(self):
        return self.player.method == "PayPal" and self.round_number == Constants.num_rounds


class P29_T3_End(Page):
    pass

    def is_displayed(self):
        return self.round_number == Constants.num_rounds


page_sequence = [P01_Welcome,
                 P02_T1_Inst,
                 #P03_T1_Inst1,
                 #P04_T1_Inst2,
                 #P07_T1_Control,
                 P09_T1_Role,
                 W03_T1_ShuffleWait,
                 W04_T1_GroupWait,
                 P17_T1_Teilnehmer1,
                 W06_T1_PayoffWait,
                 P19_T1_Feedback,
                 #P20_T2_Intro,
                 #P21_T2_Q1,
                 #P22_T2_Q2,
                 #P23_T2_Q31,
                 #P23_T2_Q32,
                 #P24_T2_Q4,
                 #P25_T2_Q5,
                 #W07_T3_ResultsWait,
                 #P26_T3_Results
                 #P27_T3_Payment,
                 #P28_T3_PaymentB,
                 #P28_T3_PaymentP,
                 #P29_T3_End
                ]