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


author = 'Eveline Vandewal'

doc = """
Part 3
"""


def make_likert_field(label):

    return models.IntegerField(
        label=label,
        choices=[0, 1, 2, 3],
        widget=widgets.RadioSelect
    )


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    statement1 = make_likert_field('1. I am always trying to figure myself out.')
    statement2 = make_likert_field('2. I am concerned about my style of doing things.')
    statement3 = make_likert_field('3. It takes me time to get over my shyness in new situations.')
    statement4 = make_likert_field('4. I think about myself a lot.')
    statement5 = make_likert_field('5. I care a lot about how I present myself to others.')
    statement6 = make_likert_field('6. I often daydream about myself.')
    statement7 = make_likert_field('7. It is hard for me to work when someone is watching me.')
    statement8 = make_likert_field('8. I never take a hard look at myself.')
    statement9 = make_likert_field('9. I get embarrassed very easily.')
    statement10 = make_likert_field('10. I am self-conscious about the way I look.')
    statement11 = make_likert_field('11. It is easy for me to talk to strangers.')
    statement12 = make_likert_field('12. I generally pay attention to my inner feelings.')
    statement13 = make_likert_field('13. I usually worry about making a good impression.')
    statement14 = make_likert_field('14. I am constantly thinking about my reasons for doing things.')
    statement15 = make_likert_field('15. I feel nervous when I speak in front of a group.')
    statement16 = make_likert_field('16. Before I leave my house, I check how I look.')
    statement17 = make_likert_field('17. I sometimes step back in my mind to examine myself from a distance.')
    statement18 = make_likert_field('18. I am concerned about what other people think of me.')
    statement19 = make_likert_field('19. I am quick to notice changes in my mood.')
    statement20 = make_likert_field('20. I am usually aware of my appearance.')
    statement21 = make_likert_field('21. I know the way my mind works when I work through a problem.')
    statement22 = make_likert_field('22. Large groups make me nervous.')

    treatment = models.IntegerField()

    rank1 = models.IntegerField(
        label='You choose Option A and the other participant chooses Option A.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank2 = models.IntegerField(
        label='You choose Option A and the other participant chooses Option B.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank3 = models.IntegerField(
        label='You choose Option B and the other participant chooses Option A.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )
    rank4 = models.IntegerField(
        label='You choose Option B and the other participant chooses Option B.',
        choices=[1, 2, 3, 4],
        widget=widgets.RadioSelect
    )

    choice = models.IntegerField(
        choices=[
            [1, 'Option A'],
            [2, 'Option B']
        ],
        widget=widgets.RadioSelect
    )
    belief = models.IntegerField()

    question1 = models.IntegerField()
    question2 = models.IntegerField(
        choices=[
            [1, 'Female'],
            [2, 'Male'],
            [3, 'Other'],
            [4, 'Prefer Not to Answer']
        ],
        widget=widgets.RadioSelect
    )
    question3 = models.IntegerField(
        choices=[
            [1, 'British'],
            [2, 'Other European'],
            [3, 'African'],
            [4, 'Asian'],
            [5, 'Latin American'],
            [6, 'North American'],
            [7, 'Other']
        ],
        widget=widgets.RadioSelect
    )
    question4 = models.IntegerField(
        choices=[
            [0, 'No Formal Eduction'],
            [1, 'Primary Education'],
            [2, 'Secondary Education'],
            [3, 'Further Education'],
            [4, 'Higher Education']
        ],
        widget=widgets.RadioSelect
    )
    question5 = models.IntegerField()
    question6 = models.IntegerField(
        choices=[
            [1, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect
    )
    question7 = models.IntegerField(
        choices=[1, 2, 3, 4, 5, 6, 7],
        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, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect
    )
    question12 = models.BooleanField(
        choices=[
            [True, 'Yes'],
            [False, 'No']
        ],
        widget=widgets.RadioSelect
    )
    question13 = models.IntegerField(
        choices=[
            [1, 'Yes'],
            [2, 'No']
        ],
        widget=widgets.RadioSelect,
        blank=True
    )

    participant_vars_dump = models.StringField()