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






class Welcome(Page):
    form_model = 'player'
    form_fields = ['control_question']

    def vars_for_template(self):
        self.player.Prolific_ID = self.player.participant.label

class ProlificID(Page):
    form_model = 'player'
    form_fields = ['Prolific_ID']

class General_Instructions(Page):
    form_model = 'player'


class Description(Page):
    form_model = 'player'


class EmotionMeasure1(Page):
    form_model = 'player'
    form_fields = ['warmth_before',
                   'anger_before',
                   'fear_before',
                   'envy_before',
                   'sadness_before',
                   'happiness_before',
                   'shame_before',
                   'irritation_before',
                   'contempt_before',
                   'guilt_before',
                   'joy_before',
                   'jealousy_before',
                   'surprise_before',]

class Content(Page):
    form_model = 'player'

class Videos(Page):
    form_model = 'player'

class EmotionMeasure2(Page):
    form_model = 'player'
    form_fields = ['warmth_after',
                   'anger_after',
                   'fear_after',
                   'envy_after',
                   'sadness_after',
                   'happiness_after',
                   'shame_after',
                   'irritation_after',
                   'contempt_after',
                   'guilt_after',
                   'joy_after',
                   'jealousy_after',
                   'surprise_after',]

class Second (Page):
    form_model = 'player'
    form_fields = ['un_contribution']

    def vars_for_template(self):
        number_list = [20, 15, 20, 8, 20, 20, 18, 5, 12]
        if self.group.id_in_subsession == 1:
            self.group.decision = number_list[0]
        elif self.group.id_in_subsession == 2:
            self.group.decision = number_list[1]
        elif self.group.id_in_subsession == 3:
            self.group.decision = number_list[2]
        elif self.group.id_in_subsession == 4:
            self.group.decision = number_list[3]
        elif self.group.id_in_subsession == 5:
            self.group.decision = number_list[4]
        elif self.group.id_in_subsession == 6:
            self.group.decision = number_list[5]
        elif self.group.id_in_subsession == 7:
            self.group.decision = number_list[6]
        elif self.group.id_in_subsession == 8:
            self.group.decision = number_list[7]
        elif self.group.id_in_subsession == 9:
            self.group.decision = number_list[8]






class Second_Estimate(Page):
       form_model = 'player'
       form_fields = ['second_estimate_1',
                      'second_estimate_2'
                      ]


class Final_Questionnaire(Page):
    form_model = 'player'
    form_fields = ['gender',
                   'age',
                   'language',
                   'max_income',
                   'trust',
                   'risk',
                   'instructions',
                   'same',
                   'econ_exp',
                   'feedback']


class Thanks(Page):
    form_model = 'player'



page_sequence = [Welcome,
                 ProlificID,
                 General_Instructions,
                 Description,
                 EmotionMeasure1,
                 Content,
                 Videos,
                 EmotionMeasure2,
                 Second_Estimate,
                 Second,
                 Final_Questionnaire,
                 Thanks
                 ]