from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer
)


author = 'Eveline Vandewal'

doc = """
Questionnaire
"""


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    randomized_order = models.IntegerField()
    colors_reversed = models.IntegerField()
    roles_reversed = models.IntegerField()

    cog_un1 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        widget=widgets.RadioSelect
    )
    cog_un2 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        widget=widgets.RadioSelect
    )
    cog_un3 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        widget=widgets.RadioSelect
    )

    rank1_1 = models.IntegerField(
        label='You earn 260 points and your interaction partner earns 200 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank1_2 = models.IntegerField(
        label='You earn 230 points and your interaction partner earns 170 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank1_3 = models.IntegerField(
        label='You earn 200 points and your interaction partner earns 170 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank1_4 = models.IntegerField(
        label='You earn 140 points and your interaction partner earns 230 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank1_5 = models.IntegerField(
        label='Your interaction partner earns 200 points and you earn 260 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank1_6 = models.IntegerField(
        label='Your interaction partner earns 170 points and you earn 230 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank1_7 = models.IntegerField(
        label='Your interaction partner earns 170 points and you earn 200 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank1_8 = models.IntegerField(
        label='Your interaction partner earns 230 points and you earn 140 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )

    rank2_1 = models.IntegerField(
        label='You earn 410 points and your interaction partner earns 360 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2_2 = models.IntegerField(
        label='You earn 460 points and your interaction partner earns 460 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2_3 = models.IntegerField(
        label='You earn 510 points and your interaction partner earns 510 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2_4 = models.IntegerField(
        label='You earn 410 points and your interaction partner earns 560 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2_5 = models.IntegerField(
        label='Your interaction partner earns 360 points and you earn 410 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2_6 = models.IntegerField(
        label='Your interaction partner earns 460 points and you earn 460 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2_7 = models.IntegerField(
        label='Your interaction partner earns 510 points and you earn 510 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2_8 = models.IntegerField(
        label='Your interaction partner earns 560 points and you earn 410 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )

    rank3_1 = models.IntegerField(
        label='You earn 650 points and your interaction partner earns 720 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3_2 = models.IntegerField(
        label='You earn 720 points and your interaction partner earns 860 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3_3 = models.IntegerField(
        label='You earn 790 points and your interaction partner earns 580 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3_4 = models.IntegerField(
        label='You earn 650 points and your interaction partner earns 790 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3_5 = models.IntegerField(
        label='Your interaction partner earns 720 points and you earn 650 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3_6 = models.IntegerField(
        label='Your interaction partner earns 860 points and you earn 720 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3_7 = models.IntegerField(
        label='Your interaction partner earns 580 points and you earn 790 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3_8 = models.IntegerField(
        label='Your interaction partner earns 790 points and you earn 650 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )

    rank4_1 = models.IntegerField(
        label='You earn 880 points and your interaction partner earns 880 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4_2 = models.IntegerField(
        label='You earn 1060 points and your interaction partner earns 790 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4_3 = models.IntegerField(
        label='You earn 700 points and your interaction partner earns 790 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4_4 = models.IntegerField(
        label='You earn 970 points and your interaction partner earns 970 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4_5 = models.IntegerField(
        label='Your interaction partner earns 880 points and you earn 880 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4_6 = models.IntegerField(
        label='Your interaction partner earns 790 points and you earn 1060 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4_7 = models.IntegerField(
        label='Your interaction partner earns 790 points and you earn 700 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4_8 = models.IntegerField(
        label='Your interaction partner earns 970 points and you earn 970 points.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )

    age = models.IntegerField()
    gender = models.IntegerField(
        blank=True,
        choices=[
            [1, 'Female'],
            [2, 'Male'],
            [3, 'Other']
        ],
        widget=widgets.RadioSelect
    )
    nationality = models.StringField()
    study = models.StringField()
    course = models.IntegerField(
        choices=[
            [1, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect
    )

    question6 = models.IntegerField()
    question7 = models.IntegerField(
        choices=[
            [1, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect
    )
    question8 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        widget=widgets.RadioSelect
    )
    question9 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        widget=widgets.RadioSelect
    )
    question10 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        widget=widgets.RadioSelect
    )
    question11 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        widget=widgets.RadioSelect
    )
    question12 = models.IntegerField(
        choices=[
            [1, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect
    )
    question13 = models.IntegerField(
        choices=[
            [1, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect
    )
    question14 = models.IntegerField(
        blank=True,
        choices=[
            [1, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect
    )

    first_name = models.StringField()
    last_name = models.StringField()
    address = models.StringField()
    city = models.StringField()
    country = models.StringField()
    dob = models.StringField()
    iban = models.StringField()
    email = models.StringField(blank=True)