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 First(Page):
    form_model = 'player'
    form_fields = ['un_contribution']


class First_Estimate(Page):
       form_model = 'player'
       form_fields = ['first_estimate_1',
                      'first_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,
                 First,
                 First_Estimate,
                 Final_Questionnaire,
                 Thanks
                 ]