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



class Instructions(Page):
    # instructions and control questions
    form_model = 'player'
    form_fields = ['q1','q2','q3','q4']


    def is_displayed(self):
        return self.subsession.round_number == 1
    def error_message(self, values):
        if values['q1'] != 4:
            self.player.errors += 1
            return 'La respuesta a la pregunta nº 1 no es correcta.'
        if values['q2'] != 1:
            self.player.errors += 1
            return 'La respuesta a la pregunta nº 2 no es correcta.'
        if values['q3'] != 0:
            self.player.errors += 1
            return 'La respuesta a la pregunta nº 3 no es correcta.'
        if values['q4'] != 0:
            self.player.errors += 1
            return 'La respuesta a la pregunta nº 4 no es correcta.'
#
class WaitPage1(WaitPage):
    wait_for_all_groups = True


class Practice_intro(Page):
    # intro to the practice phase

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


class  Price_setting_CP_gen (Page):
    # choice of the price by CP generalist
        form_model = 'group'
        form_fields = ['p_gen_input']


        def js_vars(self):
            # data for the plots of price and quantity sold
            if self.subsession.round_number > 1:
                return dict(
                     p_gen_list=self.player.participant.vars['list_p_gen'],
                     p_spec_list=self.player.participant.vars['list_p_spec'],
                     demand_gen_list=self.player.participant.vars['list_demand_gen'],
                     demand_spec_list=self.player.participant.vars['list_demand_spec']
                    )
            else:
                return ()

        def is_displayed(self):
            return self.player.id_in_group == 1

        def vars_for_template(self):
            self.group.recupera_dati()  # recollect data from previous rounds


            if self.subsession.round_number == 1:
                return{
                'round': self.subsession.round_number,
                }
            else:
                return{
                    'role': self.player.role,
                    'round': self.subsession.round_number,
                    'cum_payoff': self.group.cum_payoff_gen,
                    'tarif_1_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_1,
                    'tarif_2_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_1,
                    'tarif_1_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_2,
                    'tarif_2_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_2,
                    'tarif_1_ISP_1_sold_old':self.group.tarif_1_sold_old_ISP_1,
                    'tarif_2_ISP_1_sold_old':self.group.tarif_2_sold_old_ISP_1,
                    'tarif_1_ISP_2_sold_old':self.group.tarif_1_sold_old_ISP_2,
                    'tarif_2_ISP_2_sold_old':self.group.tarif_2_sold_old_ISP_2,
                }


class  Price_setting_CP_spec (Page):
    # choice of the price by CP specialist
        form_model = 'group'
        form_fields = ['p_spec_input']


        def is_displayed(self):
            return self.player.id_in_group == 2

        def vars_for_template(self):
            self.group.recupera_dati() # recollect data from previous round


            if self.subsession.round_number == 1:
                return{
                'round': self.subsession.round_number,
                }
            else:
                return{
                    'role': self.player.role,
                    'round': self.subsession.round_number,
                    'tarif_1_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_1,
                    'tarif_2_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_1,
                    'tarif_1_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_2,
                    'tarif_2_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_2,
                    'cum_payoff': self.group.cum_payoff_spec,
                    'tarif_1_ISP_1_sold_old':self.group.tarif_1_sold_old_ISP_1,
                    'tarif_2_ISP_1_sold_old':self.group.tarif_2_sold_old_ISP_1,
                    'tarif_1_ISP_2_sold_old':self.group.tarif_1_sold_old_ISP_2,
                    'tarif_2_ISP_2_sold_old':self.group.tarif_2_sold_old_ISP_2,
                }
        def js_vars(self):
            # data for the plots of price and quantity sold
            if self.subsession.round_number > 1:
                    return dict(
                     p_gen_list=self.player.participant.vars['list_p_gen'],
                     p_spec_list=self.player.participant.vars['list_p_spec'],
                     demand_gen_list=self.player.participant.vars['list_demand_gen'],
                     demand_spec_list=self.player.participant.vars['list_demand_spec'],

                    )
            else:
                return ()


class Price_setting_ISP_1(Page):
    # choice of the tariff by ISP_1
    form_model = 'group'
    form_fields = ['p1_ISP_1', 'p2_ISP_1']



    def is_displayed(self):
            return self.player.id_in_group == 3

    def error_message(self, values):
        if values['p1_ISP_1'] == values ['p2_ISP_1']:

            return 'Escoge dos tarifas diferentes'



    def vars_for_template(self):
        self.group.recupera_dati() # recollect data from previous round


        if self.subsession.round_number == 1:
            return{
            'round': self.subsession.round_number,
            }
        else:
            return{
            'p_gen_price': self.player.participant.vars['list_p_gen'][self.subsession.round_number-2],
            'p_spec_price': self.player.participant.vars['list_p_spec'][self.subsession.round_number-2],
            'round': self.subsession.round_number,
            'role': self.player.role,
            'round': self.subsession.round_number,
            'tarif_1_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_1,
            'tarif_2_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_1,
            'tarif_1_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_2,
            'tarif_2_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_2,
            'tarif_1_ISP_1_sold_old':self.group.tarif_1_sold_old_ISP_1,
            'tarif_2_ISP_1_sold_old':self.group.tarif_2_sold_old_ISP_1,
            'tarif_1_ISP_2_sold_old':self.group.tarif_1_sold_old_ISP_2,
            'tarif_2_ISP_2_sold_old':self.group.tarif_2_sold_old_ISP_2,
            'cum_payoff': self.group.cum_payoff_ISP_1
                }





class Price_setting_ISP_2(Page):
    # choice of the tariff by ISP_2
    form_model = 'group'
    form_fields = ['p1_ISP_2', 'p2_ISP_2']

    def old_data(self):
        self.group.recupera_dati()

    def error_message(self, values):
        if values['p1_ISP_2'] == values ['p2_ISP_2']:

            return 'Escoge dos tarifas diferentes'

    def is_displayed(self):

            return self.player.id_in_group == 4


    def vars_for_template(self):
            self.group.recupera_dati() # recollect data from previous rounds
            self.player.participant.vars['tarifs_ISP_2_old'].sort()
            self.player.participant.vars['tarifs_ISP_2_old'].sort()
            if self.subsession.round_number == 1:
                return{
                'round': self.subsession.round_number,
                }
            else:
                return{
                    'p_gen_price': self.player.participant.vars['list_p_gen'][self.subsession.round_number-2],
                    'p_spec_price': self.player.participant.vars['list_p_spec'][self.subsession.round_number-2],
                    'round': self.subsession.round_number,
                    'role': self.player.role,
                    'round': self.subsession.round_number,
                    'tarif_1_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_1,
                    'tarif_2_ISP_1_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_1,
                    'tarif_1_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p1_ISP_2,
                    'tarif_2_ISP_2_old':self.group.in_round(self.subsession.round_number-1).p2_ISP_2,
                    'tarif_1_ISP_1_sold_old':self.group.tarif_1_sold_old_ISP_1,
                    'tarif_2_ISP_1_sold_old':self.group.tarif_2_sold_old_ISP_1,
                    'tarif_1_ISP_2_sold_old':self.group.tarif_1_sold_old_ISP_2,
                    'tarif_2_ISP_2_sold_old':self.group.tarif_2_sold_old_ISP_2,

                    'cum_payoff': self.group.cum_payoff_ISP_2
                        }



class ResultsWaitPage(WaitPage):
# compute demand, utilities and payoff for the current round
    def after_all_players_arrive (self):
        self.group.demand()
        self.group.set_payoff()


class Results (Page):
    # show results of the current round
    def vars_for_template(self):


        return{
            'demand_CP_spec': round(self.group.demand_spec_TOT),
            'demand_CP_gen': round(self.group.demand_gen_TOT),
            'p_gen': self.group.p_gen,
            'p_spec': self.group.p_spec,
            'revenue_gen': round(self.group.revenue_CP_gen),
            'revenue_spec': round(self.group.revenue_CP_spec),
            'revenue_ISP_1': round(self.group.revenue_ISP_1),
            'revenue_ISP_2': round(self.group.revenue_ISP_2),
            'round': self.subsession.round_number,
            'demand_ISP_1': round(self.group.demand_ISP_1),
            'demand_ISP_2': round(self.group.demand_ISP_2),
            'tarif_1_ISP_1':self.group.p1_ISP_1,
            'tarif_2_ISP_1':self.group.p2_ISP_1,
            'tarif_1_ISP_2':self.group.p1_ISP_2,
            'tarif_2_ISP_2':self.group.p2_ISP_2,
            'tarif_1_sold_ISP_1': self.player.participant.vars['tarifs_sold_ISP_1_TOT'][self.group.p1_ISP_1-1],
            'tarif_2_sold_ISP_1': self.player.participant.vars['tarifs_sold_ISP_1_TOT'][self.group.p2_ISP_1-1],
            'tarif_1_sold_ISP_2': self.player.participant.vars['tarifs_sold_ISP_2_TOT'][self.group.p1_ISP_2-1],
            'tarif_2_sold_ISP_2': self.player.participant.vars['tarifs_sold_ISP_2_TOT'][self.group.p2_ISP_2-1],
            'demand_spec_low' :round(self.group.demand_spec_L),
            'demand_spec_high' : round(self.group.demand_spec_H),
            'demand_gen_low' : round(self.group.demand_gen_L),
            'demand_gen_high' : round(self.group.demand_gen_H),
             'role': self.player.role,
            'cum_payoff_gen': round(self.group.cum_payoff_gen),
            'cum_payoff_spec': round(self.group.cum_payoff_spec),
            'cum_payoff_ISP_1': round(self.group.cum_payoff_ISP_1),
            'cum_payoff_ISP_2': round(self.group.cum_payoff_ISP_2),

           }






page_sequence = [
    Instructions,
    Practice_intro,
    WaitPage1,
    Price_setting_CP_gen,
    Price_setting_CP_spec,
    Price_setting_ISP_1,
    Price_setting_ISP_2,
    ResultsWaitPage,
    Results,
    #MyPage

]