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


author = 'David Lucius'

doc = """
created by: David Lucius (david.lucius@posteo.de)
"""


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    # Screen: AmbiguityAversion
    CASE_A = models.IntegerField(
        label='Case A: Choose Urn 1 containing 20 balls that are either red or blue OR '
              'choose Urn 2 containing 16 red balls and 4 blue balls.',
        choices=[
            [1, 'Urn 1'],
            [2, 'Urn 2']
        ],
        widget=widgets.RadioSelect
    )

    # Screen: AmbiguityAversion
    CASE_B = models.IntegerField(
        label='Case B: Choose Urn 1 containing 20 balls that are either red or blue OR '
              'choose Urn 2containing 14 red balls and 6 blue balls.',
        choices=[
            [1, 'Urn 1'],
            [2, 'Urn 2']
        ],
        widget=widgets.RadioSelect
    )

    # Screen: AmbiguityAversion
    CASE_C = models.IntegerField(
        label='Case C: Choose Urn 1 containing 20 balls that are either red or blue OR '
              'choose Urn 2 containing 12 red balls and 8 blue balls.',
        choices=[
            [1, 'Urn 1'],
            [2, 'Urn 2']
        ],
        widget=widgets.RadioSelect
    )

    # Screen: AmbiguityAversion
    CASE_D = models.IntegerField(
        label='Case D: Choose Urn 1 containing 20 balls that are either red or blue OR '
              'choose Urn 2containing 10 red balls and 10 blue balls.',
        choices=[
            [1, 'Urn 1'],
            [2, 'Urn 2']
        ],
        widget=widgets.RadioSelect
    )

    # Screen: AmbiguityAversion
    CASE_E = models.IntegerField(
        label='Case E: Choose Urn 1 containing 20 balls that are either red or blue OR '
              'choose Urn 2 containing 8 red balls and 12 blue balls.',
        choices=[
            [1, 'Urn 1'],
            [2, 'Urn 2']
        ],
        widget=widgets.RadioSelect
    )

    # Screen: AmbiguityAversion
    CASE_F = models.IntegerField(
        label='Case F: Choose Urn 1 containing 20 balls that are either red or blue OR '
              'choose Urn 2 containing 6 red balls and 14 blue balls.',
        choices=[
            [1, 'Urn 1'],
            [2, 'Urn 2']
        ],
        widget=widgets.RadioSelect
    )

    # Screen: AmbiguityAversion
    CASE_G = models.IntegerField(
        label='Case G: Choose Urn 1 containing 20 balls that are either red or blue OR '
              'choose Urn 2 containing 4 red balls and 16 blue balls.',
        choices=[
            [1, 'Urn 1'],
            [2, 'Urn 2']
        ],
        widget=widgets.RadioSelect
    )