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



class Assessment(Page):
    form_model = 'player'
    form_fields = ['p2', 'p3', 'p4',
                   'p5_1', 'p5_2', 'p5_3',
                   'p6_1', 'p6_2', 'p6_3', 'p6_4',
                   'p7']

class BigFive(Page):
    form_model = 'player'
    form_fields = ['q1', 'q2', 'q3', 'q4', 'q5',
                   'q6', 'q7', 'q8', 'q9', 'q10',
                   ]

class MyPage(Page):
    form_model = 'player'
    form_fields = ['yr', 'std_sex', 'std_race', 'hs_gpa', 'gpa', 'std_college']


class Major(Page):
    form_model = 'player'

    def get_form_fields(self):
        if self.player.std_college == "Agriculture":
            return ['major_ag']
        elif self.player.std_college == "Business":
            return ['major_bz']
        elif self.player.std_college == "Health and Human Sciences":
            return ['major_hs']
        elif self.player.std_college == "Liberal Arts":
            return ['major_la']
        elif self.player.std_college == "Natural Sciences":
            return ['major_ns']
        elif self.player.std_college == "Veterinary Medicine and Biomedical Sciences":
            return ['major_vm']
        elif self.player.std_college == "Engineering":
            return ['major_eg']
        elif self.player.std_college == "Natural Resources":
            return ['major_nr']
        else:
            return ["major_other"]


class Personal(Page):
    form_model = 'player'
    form_fields = ['std_id', 'first_name', 'last_name', 'email', 'followup']



class Results(Page):
    pass


page_sequence = [
    Assessment,
    BigFive,
    MyPage,
    Major,
    Personal,
]