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


doc = """
This is an iterated Prisoner's Dilemma where players play against a computer player that executes
a particular strategy.
"""


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    age = models.IntegerField(
        label='What is your age?',
    )

    gender = models.StringField(
        choices=['Male', 'Female', 'Other'],
        label='What is your gender?',
        widget=widgets.RadioSelect
    )

    continent = models.StringField(
        choices=['Africa', 'Asia', 'Europe', 'North America', 'South America', 'Oceania'],
        label='Which continent are you from?',
        widget=widgets.RadioSelect
    )

    studied = models.BooleanField(
        choices=[
            [True, 'Yes'],
            [False, 'No'],
        ],
        label='Have you studied game theory before?',
        widget=widgets.RadioSelect
    )

    personality1 = models.StringField(
        label='What is your "Mind" personality aspect?',
        choices=[
            ['I', '(I) Introverted'],
            ['E', '(E) Extroverted'],
        ],
        widget=widgets.RadioSelect
    )

    personality2 = models.StringField(
        label='What is your "Energy" personality aspect?',
        choices=[
            ['S', '(S) Sensing/Observant'],
            ['N', '(N) Intuition'],
        ],
        widget=widgets.RadioSelect
    )

    personality3 = models.StringField(
        label='What is your "Nature" personality aspect?',
        choices=[
            ['T', '(T) Thinking'],
            ['F', '(F) Feeling'],
        ],
        widget=widgets.RadioSelect
    )

    personality4 = models.StringField(
        label='What is your "Tactics" personality aspect?',
        choices=[
            ['J', '(J) Judging'],
            ['P', '(P) Perceiving/Prospecting'],
        ],
        widget=widgets.RadioSelect
    )

    personality5 = models.StringField(
        label='What is your "Identity" personality aspect?',
        choices=[
            ['A', '(-A) Assertive'],
            ['T', '(-T) Turbulent'],
        ],
        widget=widgets.RadioSelect
    )

    comments = models.StringField(
        label='Please note down any other comments you may have in the box below:',
        widget=widgets.Textarea,
        blank=True
    )