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


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


class P02_T1_Inst(Page):
    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 P05_T1_Inst3(Page):
    def is_displayed(self):
        return self.round_number == 1


class P07_T1_Control(Page):
    form_model = 'player'
    form_fields = ["control1",
                   "control2",
                   "control3",
                   "control4",
                   "control5",
                   "control6"]

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


class P09_T1_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.set_offer_order()
        self.group.product_offered()


class P15_T1_SellerOfferX(Page):
    form_model = 'player'
    form_fields = ["seller_price_X"]
    def vars_for_template(self):
        seller1_product = self.group.get_player_by_id(1).T2_product
        seller2_product = self.group.get_player_by_id(2).T2_product
        period = self.round_number
        return dict(
            seller1_product=seller1_product,
            seller2_product=seller2_product,
            period=period
        )
    def is_displayed(self):
        return self.player.type == 'seller' and self.player.T2_product == 'X'


class P15_T1_SellerOfferY(Page):
    form_model = 'player'
    form_fields = ["seller_price_Y"]
    def vars_for_template(self):
        seller1_product = self.group.get_player_by_id(1).T2_product
        seller2_product = self.group.get_player_by_id(2).T2_product
        period = self.round_number
        return dict(
            seller1_product=seller1_product,
            seller2_product=seller2_product,
            period=period
        )
    def is_displayed(self):
        return self.player.type == 'seller' and self.player.T2_product == 'Y'


class W05_T1_MarketWait(WaitPage):
    pass


class P16_T1_Seller1Info(Page):
    def vars_for_template(self):
        seller1_product = self.group.get_player_by_id(1).T2_product
        seller2_product = self.group.get_player_by_id(2).T2_product
        price_1X = self.group.get_player_by_id(1).field_maybe_none('seller_price_X')
        price_1Y = self.group.get_player_by_id(1).field_maybe_none('seller_price_Y')
        price_2X = self.group.get_player_by_id(2).field_maybe_none('seller_price_X')
        price_2Y = self.group.get_player_by_id(2).field_maybe_none('seller_price_Y')
        period = self.round_number
        return dict(
            seller1_product=seller1_product,
            seller2_product=seller2_product,
            price_1X=price_1X,
            price_1Y=price_1Y,
            price_2X=price_2X,
            price_2Y=price_2Y,
            period=period
        )
    def is_displayed(self):
        return self.player.id_in_group == 1

class P16_T1_Seller2Info(Page):
    def vars_for_template(self):
        seller1_product = self.group.get_player_by_id(1).T2_product
        seller2_product = self.group.get_player_by_id(2).T2_product
        price_1X = self.group.get_player_by_id(1).field_maybe_none('seller_price_X')
        price_1Y = self.group.get_player_by_id(1).field_maybe_none('seller_price_Y')
        price_2X = self.group.get_player_by_id(2).field_maybe_none('seller_price_X')
        price_2Y = self.group.get_player_by_id(2).field_maybe_none('seller_price_Y')
        period = self.round_number
        return dict(
            seller1_product=seller1_product,
            seller2_product=seller2_product,
            price_1X=price_1X,
            price_1Y=price_1Y,
            price_2X=price_2X,
            price_2Y=price_2Y,
            period=period
        )
    def is_displayed(self):
        return self.player.id_in_group == 2


class P17_T1_Buyer(Page):
    form_model = 'player'
    form_fields = ["buyer_choice"]

    def vars_for_template(self):
        offer_order = self.group.offer_order
        seller1_product = self.group.get_player_by_id(1).T2_product
        seller2_product = self.group.get_player_by_id(2).T2_product
        price_1X = self.group.get_player_by_id(1).field_maybe_none('seller_price_X')
        price_1Y = self.group.get_player_by_id(1).field_maybe_none('seller_price_Y')
        price_2X = self.group.get_player_by_id(2).field_maybe_none('seller_price_X')
        price_2Y = self.group.get_player_by_id(2).field_maybe_none('seller_price_Y')
        period = self.round_number
        return dict(
            offer_order=offer_order,
            seller1_product=seller1_product,
            seller2_product=seller2_product,
            price_1X=price_1X,
            price_1Y=price_1Y,
            price_2X=price_2X,
            price_2Y=price_2Y,
            period=period
        )

    def is_displayed(self):
        return self.player.type == "buyer"


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):
        offer_order = self.group.offer_order
        buyer_choice = self.group.get_player_by_id(3).buyer_choice
        seller1 = self.group.get_player_by_id(1).T2_payoff
        seller2 = self.group.get_player_by_id(2).T2_payoff
        buyer = self.group.get_player_by_id(3).T2_payoff
        TC = self.group.get_player_by_id(4).T2_payoff
        seller1_product = self.group.get_player_by_id(1).T2_product
        seller2_product = self.group.get_player_by_id(2).T2_product
        price_1X = self.group.get_player_by_id(1).field_maybe_none('seller_price_X')
        price_1Y = self.group.get_player_by_id(1).field_maybe_none('seller_price_Y')
        price_2X = self.group.get_player_by_id(2).field_maybe_none('seller_price_X')
        price_2Y = self.group.get_player_by_id(2).field_maybe_none('seller_price_Y')
        period = self.round_number
        return dict(
            offer_order=offer_order,
            buyer_choice=buyer_choice,
            seller1=seller1,
            seller2=seller2,
            buyer=buyer,
            TC=TC,
            seller1_product=seller1_product,
            seller2_product=seller2_product,
            price_1X=price_1X,
            price_1Y=price_1Y,
            price_2X=price_2X,
            price_2Y=price_2Y,
            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_Q3B(Page):
    form_model = 'player'
    form_fields = ['verstanden',
                   'quality_check',
                   'not_harming',
                   'ownpayoff',
                   'utilitarian',
                   'belief_buyer_sellers']

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


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

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


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

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


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):
        buyer_choice = self.group.get_player_by_id(3).in_round(self.group.round_payed).buyer_choice
        offer_order = self.group.in_round(self.group.round_payed).offer_order
        seller_1X = self.group.get_player_by_id(1).in_round(self.group.round_payed).field_maybe_none('seller_price_X')
        seller_2X = self.group.get_player_by_id(2).in_round(self.group.round_payed).field_maybe_none('seller_price_X')
        seller_1Y = self.group.get_player_by_id(1).in_round(self.group.round_payed).field_maybe_none('seller_price_Y')
        seller_2Y = self.group.get_player_by_id(2).in_round(self.group.round_payed).field_maybe_none('seller_price_Y')
        T2_product_1 = self.group.get_player_by_id(1).in_round(self.group.round_payed).T2_product
        T2_product_2 = self.group.get_player_by_id(2).in_round(self.group.round_payed).T2_product
        T2_payoff = self.player.in_round(self.group.round_payed).T2_payoff
        final_payoff = self.player.payoff.to_real_world_currency(self.session)
        return dict(
            buyer_choice = buyer_choice,
            offer_order = offer_order,
            seller_1X = seller_1X,
            seller_2X = seller_2X,
            seller_1Y = seller_1Y,
            seller_2Y = seller_2Y,
            T2_product_1 = T2_product_1,
            T2_product_2 = T2_product_2,
            T2_payoff = T2_payoff,
            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,
                 #P05_T1_Inst3,
                 #P07_T1_Control,
                 P09_T1_Role,
                 W03_T1_ShuffleWait,
                 W04_T1_GroupWait,
                 P15_T1_SellerOfferX,
                 P15_T1_SellerOfferY,
                 W05_T1_MarketWait,
                 P16_T1_Seller1Info,
                 P16_T1_Seller2Info,
                 P17_T1_Buyer,
                 W06_T1_PayoffWait,
                 P19_T1_Feedback,
                 #P20_T2_Intro,
                 #P21_T2_Q1,
                 #P22_T2_Q2,
                 #P23_T2_Q3B,
                 #P23_T2_Q3C,
                 #P23_T2_Q3S,
                 #P24_T2_Q4,
                 #P25_T2_Q5,
                 #W07_T3_ResultsWait,
                 #P26_T3_Results
                 # P27_T3_Payment,
                 # P28_T3_PaymentB,
                 # P28_T3_PaymentP,
                 # P29_T3_End
                ]