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


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    gpa = models.StringField(
        choices=['0-1.00', '1.01-2.00', '2.01-2.59', '2.60-3.00', '3.01-3.49', '3.50-4.00'],
        label='What is your GPA? Please choose a range below?'
    )
    age = models.IntegerField(label='What is your age?', min=18, max=125)

    gender = models.StringField(
        choices = ['Male', 'Female', 'Non-binary', 'Non-conforming', 'Do not want to disclose'],
        label='What is your gender?',
        widget = widgets.RadioSelect,
    )

    weed = models.StringField(
        choices = ['Yes', 'No', 'I used it before but not anymore', 'I have not used it before but I am not against to try it', 'Do not want to disclose'],
        label='Do you use marijuana for recreational purpose?',
        widget = widgets.RadioSelect,
    )

    agree = models.IntegerField(label='How many of these statements bother you? Please enter an integer. If none of them bothers you, please enter 0.', min=0, max=4)

    drive = models.StringField(
        choices = ['Yes', 'No', 'Do not want to disclose'],
        label='Do you drive or possess a car yourself?',
        widget = widgets.RadioSelect,
    )

    race = models.StringField(
        choices = ['Asian and Pacific Islander', 'Black or African American', 'Hispanic or Latino', 'Native American', 'White', 'Do not want to disclose'],
        label='What is your racial background?'
    )

    major = models.StringField(label='What is your major?')

    crt_bat = models.IntegerField(
        label='''
        A bat and a ball cost 22 dollars in total.
        The bat costs 20 dollars more than the ball.
        How many dollars does the ball cost?'''
    )

    crt_widget = models.IntegerField(
        label='''
        "If it takes 5 machines 5 minutes to make 5 widgets,
        how many minutes would it take 100 machines to make 100 widgets?"
        '''
    )

    crt_lake = models.IntegerField(
        label='''
        In a lake, there is a patch of lily pads.
        Every day, the patch doubles in size.
        If it takes 48 days for the patch to cover the entire lake,
        how many days would it take for the patch to cover half of the lake?
        '''
    )