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

from django import forms
from django.core.validators import MinLengthValidator
from django.core.validators import RegexValidator

author = 'Alexander Coutts, Boon Han Koh, Zahra Murad'

doc = """
Feedback/Gender Experiment: Demographic Survey (All)
"""


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

states = [
    ['AK', 'Alaska'],
    ['AL', 'Alabama'],
    ['AR', 'Arkansas'],
    ['AS', 'American Samoa'],
    ['AZ', 'Arizona'],
    ['CA', 'California'],
    ['CO', 'Colorado'],
    ['CT', 'Connecticut'],
    ['DC', 'District of Columbia'],
    ['DE', 'Delaware'],
    ['FL', 'Florida'],
    ['GA', 'Georgia'],
    ['GU', 'Guam'],
    ['HI', 'Hawaii'],
    ['IA', 'Iowa'],
    ['ID', 'Idaho'],
    ['IL', 'Illinois'],
    ['IN', 'Indiana'],
    ['KS', 'Kansas'],
    ['KY', 'Kentucky'],
    ['LA', 'Louisiana'],
    ['MA', 'Massachusetts'],
    ['MD', 'Maryland'],
    ['ME', 'Maine'],
    ['MI', 'Michigan'],
    ['MN', 'Minnesota'],
    ['MO', 'Missouri'],
    ['MP', 'Northern Mariana Islands'],
    ['MS', 'Mississippi'],
    ['MT', 'Montana'],
    ['NA', 'National'],
    ['NC', 'North Carolina'],
    ['ND', 'North Dakota'],
    ['NE', 'Nebraska'],
    ['NH', 'New Hampshire'],
    ['NJ', 'New Jersey'],
    ['NM', 'New Mexico'],
    ['NV', 'Nevada'],
    ['NY', 'New York'],
    ['OH', 'Ohio'],
    ['OK', 'Oklahoma'],
    ['OR', 'Oregon'],
    ['PA', 'Pennsylvania'],
    ['PR', 'Puerto Rico'],
    ['RI', 'Rhode Island'],
    ['SC', 'South Carolina'],
    ['SD', 'South Dakota'],
    ['TN', 'Tennessee'],
    ['TX', 'Texas'],
    ['UT', 'Utah'],
    ['VA', 'Virginia'],
    ['VI', 'Virgin Islands'],
    ['VT', 'Vermont'],
    ['WA', 'Washington'],
    ['WI', 'Wisconsin'],
    ['WV', 'West Virginia'],
    ['WY', 'Wyoming'],
    ['OTHER', 'Other (please state below)'],
]

resid_state_list = states
grew_state_list = states
grew_state_list.insert(0,['N/A', 'Did not spend first 18 years in US at all'])

countries = [
    ['US', 'United States of America'],
    ['AF', 'Afghanistan'],
    ['AL', 'Albania'],
    ['DZ', 'Algeria'],
    ['AD', 'Andorra'],
    ['AO', 'Angola'],
    ['AG', 'Antigua and Barbuda'],
    ['AR', 'Argentina'],
    ['AM', 'Armenia'],
    ['AU', 'Australia'],
    ['AT', 'Austria'],
    ['AZ', 'Azerbaijan'],
    ['BS', 'Bahamas'],
    ['BH', 'Bahrain'],
    ['BD', 'Bangladesh'],
    ['BB', 'Barbados'],
    ['BY', 'Belarus'],
    ['BE', 'Belgium'],
    ['BZ', 'Belize'],
    ['BJ', 'Benin'],
    ['BT', 'Bhutan'],
    ['BO', 'Bolivia (Plurinational State of)'],
    ['BA', 'Bosnia and Herzegovina'],
    ['BW', 'Botswana'],
    ['BR', 'Brazil'],
    ['BN', 'Brunei Darussalam'],
    ['BG', 'Bulgaria'],
    ['BF', 'Burkina Faso'],
    ['BI', 'Burundi'],
    ['CV', 'Cabo Verde'],
    ['KH', 'Cambodia'],
    ['CM', 'Cameroon'],
    ['CA', 'Canada'],
    ['CF', 'Central African Republic'],
    ['TD', 'Chad'],
    ['CL', 'Chile'],
    ['CN', 'China'],
    ['CO', 'Colombia'],
    ['KM', 'Comoros'],
    ['CG', 'Congo'],
    ['CK', 'Cook Islands'],
    ['CR', 'Costa Rica'],
    ['HR', 'Croatia'],
    ['CU', 'Cuba'],
    ['CY', 'Cyprus'],
    ['CZ', 'Czechia'],
    ['CI', 'Côte d\'Ivoire'],
    ['KP', 'Democratic People\'s Republic of Korea'],
    ['CD', 'Democratic Republic of the Congo'],
    ['DK', 'Denmark'],
    ['DJ', 'Djibouti'],
    ['DM', 'Dominica'],
    ['DO', 'Dominican Republic'],
    ['EC', 'Ecuador'],
    ['EG', 'Egypt'],
    ['SV', 'El Salvador'],
    ['GQ', 'Equatorial Guinea'],
    ['ER', 'Eritrea'],
    ['EE', 'Estonia'],
    ['SZ', 'Eswatini'],
    ['ET', 'Ethiopia'],
    ['FO', 'Faroe Islands'],
    ['FJ', 'Fiji'],
    ['FI', 'Finland'],
    ['FR', 'France'],
    ['GA', 'Gabon'],
    ['GM', 'Gambia'],
    ['GE', 'Georgia'],
    ['DE', 'Germany'],
    ['GH', 'Ghana'],
    ['GR', 'Greece'],
    ['GD', 'Grenada'],
    ['GT', 'Guatemala'],
    ['GN', 'Guinea'],
    ['GW', 'Guinea-Bissau'],
    ['GY', 'Guyana'],
    ['HT', 'Haiti'],
    ['HN', 'Honduras'],
    ['HU', 'Hungary'],
    ['IS', 'Iceland'],
    ['IN', 'India'],
    ['ID', 'Indonesia'],
    ['IR', 'Iran (Islamic Republic of)'],
    ['IQ', 'Iraq'],
    ['IE', 'Ireland'],
    ['IL', 'Israel'],
    ['IT', 'Italy'],
    ['JM', 'Jamaica'],
    ['JP', 'Japan'],
    ['JO', 'Jordan'],
    ['KZ', 'Kazakhstan'],
    ['KE', 'Kenya'],
    ['KI', 'Kiribati'],
    ['KW', 'Kuwait'],
    ['KG', 'Kyrgyzstan'],
    ['LA', 'Lao People\'s Democratic Republic'],
    ['LV', 'Latvia'],
    ['LB', 'Lebanon'],
    ['LS', 'Lesotho'],
    ['LR', 'Liberia'],
    ['LY', 'Libya'],
    ['LT', 'Lithuania'],
    ['LU', 'Luxembourg'],
    ['MG', 'Madagascar'],
    ['MW', 'Malawi'],
    ['MY', 'Malaysia'],
    ['MV', 'Maldives'],
    ['ML', 'Mali'],
    ['MT', 'Malta'],
    ['MH', 'Marshall Islands'],
    ['MR', 'Mauritania'],
    ['MU', 'Mauritius'],
    ['MX', 'Mexico'],
    ['FM', 'Micronesia (Federated States of)'],
    ['MC', 'Monaco'],
    ['MN', 'Mongolia'],
    ['ME', 'Montenegro'],
    ['MA', 'Morocco'],
    ['MZ', 'Mozambique'],
    ['MM', 'Myanmar'],
    ['NA', 'Namibia'],
    ['NR', 'Nauru'],
    ['NP', 'Nepal'],
    ['NL', 'Netherlands'],
    ['NZ', 'New Zealand'],
    ['NI', 'Nicaragua'],
    ['NE', 'Niger'],
    ['NG', 'Nigeria'],
    ['NU', 'Niue'],
    ['MK', 'North Macedonia'],
    ['NO', 'Norway'],
    ['OM', 'Oman'],
    ['PK', 'Pakistan'],
    ['PW', 'Palau'],
    ['PA', 'Panama'],
    ['PG', 'Papua New Guinea'],
    ['PY', 'Paraguay'],
    ['PE', 'Peru'],
    ['PH', 'Philippines'],
    ['PL', 'Poland'],
    ['PT', 'Portugal'],
    ['QA', 'Qatar'],
    ['KR', 'Republic of Korea'],
    ['MD', 'Republic of Moldova'],
    ['RO', 'Romania'],
    ['RU', 'Russian Federation'],
    ['RW', 'Rwanda'],
    ['KN', 'Saint Kitts and Nevis'],
    ['LC', 'Saint Lucia'],
    ['VC', 'Saint Vincent and the Grenadines'],
    ['WS', 'Samoa'],
    ['SM', 'San Marino'],
    ['ST', 'Sao Tome and Principe'],
    ['SA', 'Saudi Arabia'],
    ['SN', 'Senegal'],
    ['RS', 'Serbia'],
    ['SC', 'Seychelles'],
    ['SL', 'Sierra Leone'],
    ['SG', 'Singapore'],
    ['SK', 'Slovakia'],
    ['SI', 'Slovenia'],
    ['SB', 'Solomon Islands'],
    ['SO', 'Somalia'],
    ['ZA', 'South Africa'],
    ['SS', 'South Sudan'],
    ['ES', 'Spain'],
    ['LK', 'Sri Lanka'],
    ['SD', 'Sudan'],
    ['SR', 'Suriname'],
    ['SE', 'Sweden'],
    ['CH', 'Switzerland'],
    ['SY', 'Syrian Arab Republic'],
    ['TJ', 'Tajikistan'],
    ['TH', 'Thailand'],
    ['TL', 'Timor-Leste'],
    ['TG', 'Togo'],
    ['TK', 'Tokelau'],
    ['TO', 'Tonga'],
    ['TT', 'Trinidad and Tobago'],
    ['TN', 'Tunisia'],
    ['TR', 'Turkey'],
    ['TM', 'Turkmenistan'],
    ['TV', 'Tuvalu'],
    ['UG', 'Uganda'],
    ['UA', 'Ukraine'],
    ['AE', 'United Arab Emirates'],
    ['GB', 'United Kingdom of Great Britain and Northern Ireland'],
    ['TZ', 'United Republic of Tanzania'],
    ['UY', 'Uruguay'],
    ['UZ', 'Uzbekistan'],
    ['VU', 'Vanuatu'],
    ['VE', 'Venezuela (Bolivarian Republic of)'],
    ['VN', 'Viet Nam'],
    ['YE', 'Yemen'],
    ['ZM', 'Zambia'],
    ['ZW', 'Zimbabwe'],
    ['OTHER', 'Other (please state below)'],
]

grew_country_list = countries

class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):

    yearbirth = models.IntegerField(
        label   = "What is your year of birth?",
        min     = 1900,
        max     = 2005,
    )

    quarterbirth = models.IntegerField(
        label   = "What is your month of birth?",
        choices = [
            [1,'Between January and March'],
            [2,'Between April and June'],
            [3,'Between July and September'],
            [4,'Between October and December'],
        ]
    )

    female = models.IntegerField(
        label   = "What is your gender?",
        choices = [
            [0, 'Male'],
            [1, 'Female'],
            [-1, 'Trans/Intersex/Other'],
        ]
    )

    ethnicity = models.StringField(
        label   = "What is your ethnicity?",
        choices = [
            ['White', 'White/ Caucasian'],
            ['Black', 'Black/ African-American'],
            ['Hispanic', 'Latino or Hispanic'],
            ['Asian', 'Asian'],
            ['Native American', 'Native American'],
            ['Islander', 'Native Hawaiian or Pacific Islander'],
            ['Mixed', 'Mixed/ Multiple ethnic groups'],
            ['OTHER', 'Other ethnic group (please state below)'],
        ]
    )
    other_ethnicity = models.StringField(
        label   = '',
        blank   = True
    )

    educ = models.IntegerField(
        label   = "What is the highest education qualification you have attained?",
        choices = [
            [1, 'Did not complete High School'],
            [2, 'Graduated from High School'],
            [3, 'Bachelor’s Degree'],
            [4, 'Master’s Degree'],
            [5, 'Ph.D. or higher'],
        ]
    )

    income = models.IntegerField(
        label   = "What is your household annual income?",
        choices = [
            [1, 'Less than $20,000'],
            [2, '$20,000-$39,999'],
            [3, '$40,000-$59,999'],
            [4, '$60,000-$79,999'],
            [5, '$80,000-$99,999'],
            [6, '$100,000 or more'],
        ]
    )

    residence_state = models.StringField(
        label   = "In which US state/territory do you currently live?",
        choices = resid_state_list
    )
    other_residence_state = models.StringField(
        label   = '',
        blank   = True
    )

    grewup_country = models.StringField(
        label   = "In which country did you spend the most time for the first 18 years of your life?",
        choices = grew_country_list
    )
    other_grewup_country = models.StringField(
        label   = '',
        blank   = True
    )

    grewup_state = models.StringField(
        label   = "In which US state/territory did you spend the most time for the first 18 years of your life?",
        choices = grew_state_list
    )
    other_grewup_state = models.StringField(
        label   = '',
        blank   = True
    )

    fave_color = models.StringField(
        label   = "Of the following, which is your favorite color?",
        choices = [
            'Red',
            'Green',
            'Blue',
            'Purple',
        ]
    )

    fave_beverage = models.StringField(
        label="Of the following, which is your favorite choice of hot beverage?",
        choices = [
            'Coffee',
            'Tea',
            'Hot chocolate',
        ]
    )