from otree.api import *
import pytz

author = 'Steffen Hilpisch'

doc = """
Post experiment survey - meant after belief survey.
"""


class Constants(BaseConstants):
    name_in_url = 'questionnaire'
    players_per_group = None
    num_rounds = 1


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    # Memory questions
    belief_memory1 = models.IntegerField(doc='',
                                          label='At the end of the previous/last round (Round 3), what did you '
                                                'think were the chances (in %) that the BLUE URN was picked in the '
                                                'beginning?',
                                          min=0,
                                          max=100)
    belief_memory2 = models.IntegerField(doc='',
                                          label='At the end of the second to last round (Round 2), what did you think '
                                                'were the chances (in %) that the BLUE URN was picked in the '
                                                'beginning?',
                                          min=0,
                                          max=100)

    # Belief Questions
    belief_strategy = models.IntegerField(doc='Strategy subjects followed to make guesses',
                                          label='What method did you mainly use to arrive at some estimate?',
                                          choices=[
                                              [1, 'I tried to calculate precise probabilities.'],
                                              [2, 'I used some rule of thumb for calculations.'],
                                              [3, 'I mainly relied on my intuition.'],
                                              [4, 'I had no idea and entered random numbers.'],
                                              [5, 'Other.']
                                          ])
    belief_optimal = models.IntegerField(doc='Subjective probability regarding optimality of own action',
                                         label='In theory, after each report that was shown to you it was possible to calculate '
                                               'the exact percentage chance the selected urn was blue. '
                                               'How close (in percentage points) do you think you were on average to this correct probability? '
                                               '(Zero means you always reported the correct number)',
                                         choices=[
                                             [1, '0%'],
                                             [2, '1-5%'],
                                             [3, '6-10%'],
                                             [4, '11-20%'],
                                             [5, '21-30%'],
                                             [6, '31-40%'],
                                             [7, '41-50%'],
                                             [8, 'More than 50%']
                                         ])
    belief_fake_blue = models.IntegerField(doc='',
                                           label='How likely is it that an uninformative ball is blue? (in %)',
                                           min=0,
                                           max=100)
    belief_fake_red = models.IntegerField(doc='',
                                          label='How likely is it that an uninformative ball is red? (in %)',
                                          min=0,
                                          max=100)

    # CRT
    crt1 = models.IntegerField(doc='CRT1 - Running',
                               label='1. If you are running a race and you pass the person in second place, what place are you in?',
                               choices=[
                                   [1, 'First.'],
                                   [2, 'Second.'],
                                   [3, 'Third.'],
                                   [4, 'Not enough information.']
                               ],
                               widget=widgets.RadioSelect)
    crt2 = models.IntegerField(doc='CRT2 - ',
                               label='2. A farmer had 15 sheep and all but 8 died. How many are left?',
                               choices=[
                                   [1, '15.'],
                                   [2, '8.'],
                                   [3, '7.'],
                                   [4, 'Not enough information.']
                               ],
                               widget=widgets.RadioSelect)
    crt3 = models.IntegerField(doc='CRT3 - ',
                               label="3. Emily's father has three daughters. The first two are named April and May. What is the third daughter's name?",
                               choices=[
                                   [1, 'June.'],
                                   [2, 'July.'],
                                   [3, 'Emily.'],
                                   [4, 'Not enough information.']
                               ],
                               widget=widgets.RadioSelect)
    crt4 = models.IntegerField(doc='CRT4 - lethal heart attacks',
                               label='4. How many lethal heart attacks did you survive this far?',
                               choices=[
                                   [1, 'Zero.'],
                                   [2, '1.'],
                                   [3, '2.'],
                                   [4, '3 or more.']
                               ],
                               widget=widgets.RadioSelect)
    # Demographics
    age = models.IntegerField(doc='Age',
                              label='How old are you?',
                              min=16,
                              max=100)
    gender = models.IntegerField(doc='Gender',
                                 label='What is your gender?',
                                 choices=[
                                     [1, 'Female'],
                                     [2, 'Male'],
                                     [3, 'Other']
                                 ])
    edu = models.IntegerField(doc='Level of educations',
                              label='What is the highest level of education you have completed?',
                              choices=[
                                  [1, 'No completed education.'],
                                  [2, 'High school'],
                                  [3, 'Bachelor or equivalent'],
                                  [4, 'Master or equivalent'],
                                  [5, 'PhD'],
                                  [6, 'Other']
                              ])
    occ = models.IntegerField(doc='Occupations',
                              label='What best describes your main occupation at this moment?',
                              choices=[
                                  [1, 'Student'],
                                  [2, 'Unemployed'],
                                  [3, 'Working part time'],
                                  [4, 'Working full time'],
                                  [5, 'Other']
                              ])
    country = models.StringField(doc='Country of residence',
                                 label='What is your country of residence?',
                                 choices=pytz.country_names.items()
                                 )
    prob_fam = models.IntegerField(doc='Level of familiarity with probabilities',
                                   label='How familiar would you describe yourself with probabilities?',
                                   choices=[
                                       [1, 'I have no idea about it.'],
                                       [2, 'I have heard of it, but I do not know much about it.'],
                                       [3, 'I know a bit about it.'],
                                       [4, 'I understand it well.']
                                   ])


# PAGES
class MemoryQuestions(Page):
    form_model = 'player'
    form_fields = ['belief_memory1', 'belief_memory2']


class Transition(Page):
    pass


class BeliefQuestions(Page):
    form_model = 'player'
    form_fields = ['belief_strategy', 'belief_optimal', 'belief_fake_blue', 'belief_fake_red']


class CRT(Page):
    form_model = 'player'
    form_fields = ['crt1', 'crt2', 'crt3', 'crt4']


class Demographics(Page):
    form_model = 'player'
    form_fields = ['age', 'gender', 'edu', 'occ', 'country', 'prob_fam']


class FinalPage(Page):
    # feedback on correct urn and payment.
    pass


page_sequence = [MemoryQuestions, Transition, BeliefQuestions, CRT, Demographics, FinalPage]