from otree.api import *


doc = """
Post-survey
"""


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    # Math ability
    math3 = models.StringField(blank=True,
                               choices=[[1, '540'], [2, '570'], [3, '600']],
                               widget=widgets.RadioSelect,
                               label='',
                               )

    # Decision-making logic
    logic_EV_U = models.StringField(blank=True,
                                  choices=[[1, 'Yes'], [2, 'No']],
                                  widget=widgets.RadioSelect,
                                  label='1-1. Did you attempt to calculate the expected value of your income in games where '
                                        'information on probabilities and income values was NOT provided?',
                                  )

    logic_EV_R = models.StringField(blank=True,
                                  choices=[[1, 'Yes'], [2, 'No']],
                                  widget=widgets.RadioSelect,
                                  label='1-2. In the games where such information was available, '
                                        'did you calculate the expected value of your income?',
                                  )

    logic_PT_G = models.StringField(blank=True,
                                    choices=[[1, 'Yes'], [2, 'No'], [3, 'Partly']],
                                    widget=widgets.RadioSelect,
                                    label='2-1. When incomes started declining, were you motivated to remain playing '
                                          'until you restored your original income flow?',
                                    )

    logic_PT_L = models.StringField(blank=True,
                                    choices=[[1, 'Yes'], [2, 'No'], [3, 'Partly']],
                                    widget=widgets.RadioSelect,
                                    label='2-2. When incomes started declining, were you worried about losing '
                                          'what you had and so decided to quit earlier than you might otherwise have done?',
                                    )

    logic_Factor = models.LongStringField(
        blank=True,
        label='3. What factors were important in making the exit decisions?'
              ' (e.g., Initial income, exit cost, expected income value, etc.) '
              'Did the factors you mentioned have the same importance in games where information on probabilities '
              'and income values was NOT provided and games where such information was given?',
    )

    logic_Inc = models.LongStringField(
        blank=True,
        label='4. In what way did the initial income of $40 play a role in shaping your decisions?',
    )

    logic_ERvLA = models.LongStringField(
        blank=True,
        label='5. While playing the games, have you thought that you decided to exit too late or too early? '
              'If so, please describe what you were thinking at that time.'
    )

# PAGES
class Lq(Page):
    form_model = 'player'
    form_fields = ['math3', 'logic_EV_U', 'logic_EV_R', 'logic_PT_G', 'logic_PT_L', 'logic_Factor',
                   'logic_Inc', 'logic_ERvLA']

page_sequence = [Lq]