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

class FaceSheet(Page):
    form_model = 'player'
class Demographics(Page):
    form_model = 'player'
    form_fields = ['age', 'gender']
class Csr_Explanation(Page):
    form_model = 'player'
    form_fields = ['csr1', 'csr2', 'csr3', 'csr4', 'csr5']
    def error_message(self, values):
        if values['csr1'] == "0" or values['csr2'] == "1" or values['csr3'] == "1" or values['csr4'] == "1" or values['csr5'] == "0":
            return 'Please answer again!'
class Positive_Top_Message(Page):
    form_model = 'player'
    def is_displayed(self):
        return self.player.treatment_id == 'positive'
class Negative_Top_Message(Page):
    form_model = 'player'
    def is_displayed(self):
        return self.player.treatment_id == 'negative'
class Decision_Making(Page):
    form_model = 'player'
    form_fields = ['short_price', 'long_price', 'allocation']
class Questions(Page):
    form_model = 'player'
    form_fields = ['q1', 'q2', 'q3', 'q4', 'q5', 'q6']
page_sequence = [FaceSheet, Demographics, Csr_Explanation, Positive_Top_Message, Negative_Top_Message, Decision_Making, Questions]