from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c, currency_range
)


author = 'Eugene'

doc = """
Control questionnaire
"""


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):
    investment = models.FloatField(label="On average, how much would each player have to transfer "
                                         "to the group account in total for the group "
                                         "to achieve its target of £30?",
                                     choices=[0, 3, 7.50, 10],
                                     widget=widgets.RadioSelectHorizontal
                                     )
    takehome = models.IntegerField(label="Imagine that the group succeeds in reaching the target of £30 "
                                         "after the 10 rounds, and you have £5 remaining"
                                         " in your private account. How much money will you"
                                         " take home from the game?",
                                   choices=[0, 2.50, 5, 10],
                                   widget=widgets.RadioSelectHorizontal
                                   )
    p_ruin = models.IntegerField(label="Assume that the group target of £30 is not reached, "
                                              "and you have £5 remaining in your private account. "
                                              "What is the probability that you will lose this "
                                              "remaining amount in your private account?",
                                        choices=[0, 10, 50, 100],
                                        widget=widgets.RadioSelectHorizontal
                                        )
    contributions_fair_p1 = models.FloatField(label="£20?")
    contributions_fair_p2 = models.FloatField(label="£10?")

    contributions_pred_p1 = models.FloatField(label="Players starting with £20:")
    contributions_pred_p2 = models.FloatField(label="Players starting with £10:")

    confidence = models.IntegerField(label="How confident are you that the group will achieve "
                                           "its target after 10 rounds? (1 = Not confident at all,"
                                           " 7 = Extremely confident)",
                                           choices=[1,2,3,4,5,6,7])
                # Add slider 0-100% confident here
                # Just put a dropdown in for now - can do a slider if needed later
                # That's fine - have added labels and made it 1-7