from ._builtin import Page, WaitPage
from .models import Constants




############################################################
#                    WELCOME
############################################################

# general instructions and only shown in round 1

class Introduction(Page):
    def is_displayed(self):
        return self.round_number == 1


############################################################
#                    QUESTIONNAIRE
############################################################

# questionnaire on risk aversion, aversion against machines, ...
# before actual task to avoid priming during sessions

class Questionnaire(Page):

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

    form_model = 'player'
    form_fields = [
        'q_age',
        'q_female',
        'q_work',
        'q_degree',
        'q_risk_aversion_1',
        'q_risk_aversion_2',
        'q_risk_aversion_3',
        'q_algo_aversion_1',
        'q_algo_aversion_2',
        'q_algo_aversion_3',
        'q_algo_aversion_4',
        'q_tech_anxiety_1',
        'q_tech_anxiety_2',
        'q_tech_anxiety_3',
        'q_tech_anxiety_4',
        'q_finance_1',
        'trusting_human',
        'trusting_machine',
        'q_cs_study',
        'q_coding_skill',
        'ml_knowledge'
    ]

############################################################
#             INTRODUCTION MARKET GAME
############################################################

# instructions about the experimental market game
# only shown at the beginning
# creates the random state of the asset

class Instructions(Page):
    def is_displayed(self):
        return self.round_number == 1


############################################################
#                   TRIAL 1
############################################################

class Trial1(Page):

    # decisions
    form_model = 'player'
    form_fields = [
        'trial_1'
    ]


class TrialResult1(Page):

    def vars_for_template(self):
        return dict(
            trial=self.player.trial_1
        )


############################################################
#                   TRIAL 2
############################################################

class Trial2(Page):

    # decisions
    form_model = 'player'
    form_fields = [
        'trial_2'
    ]

class TrialResult2(Page):

    def vars_for_template(self):
        return dict(
            trial=self.player.trial_2
        )




page_sequence = [Introduction, Questionnaire, Instructions,
                 Trial1, TrialResult1,
                 Trial2, TrialResult2]