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


author = 'Your name here'

doc = """
Your app description
"""


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


class Subsession(BaseSubsession):
    def creating_session(self):
        # randomize to treatments
        import random
        import itertools
        tratamiento = itertools.cycle([1,2,3 ])
        for p in self.get_players():
            p.tratamiento = next(tratamiento)


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    participa = models.BooleanField(label = "A CONTINUACIÓN DEBES CONFIRMAR SER UN RESIDENTE DE MÉXICO Y TENER ENTRE 18 Y 25 AÑOS DE EDAD PARA CONTESTAR NUESTRA ENCUESTA",
        choices=[[True,'Sí, me gustaría contestar la encuesta, y confimo que soy un RESIDENTE PERMANENTE DE MÉXICO y tengo entre 18 y 25 años de edad'],[False, 'No, prefiero no participar']]
                                )
    tratamiento = models.IntegerField()
    edad = models.IntegerField(choices = [[18,"18 años"],[19,"19 años"],[20,"20 años"],[21,"21 años"],[22,"22 años"],[23,"23 años"],[24,"24 años"],[25,"25 años"],])
    sexo = models.IntegerField(label="",
                                choices=[
                                    [1, "Hombre"],
                                    [2, "Mujer"],
                                     ]
                                )
    es_civil = models.IntegerField(label="",
                                        choices=[
                                            [1,"Casado"],
                                            [2, "Soltero"],
                                            [3, "Unión libre"],
                                            [4, "Separado/Divociado"],
                                            [5, "Viudo"],
                                             ]
                                        )

    raza = models.IntegerField(label="",
                                   choices=[
                                       [1, "Mexicano Indígena"],
                                       [2, "Mexicano Afrodescendiente/Negro"],
                                       [3, "Mexicano Blanco"],
                                       [4, "Mexicano Asiático"],
                                       [5, "Otro"],
                                   ]
                                   )
    nacimiento = models.IntegerField(label="",
                               choices=[
                                   [1, "Sí"],
                                   [2, "No"],
                               ]
                               )
    estado = models.IntegerField(label="",
                                   choices=[
                                       [1, "Aguascalientes"],
                                       [2, "Baja California"],
                                       [3, "Baja California Sur"],
                                       [4, "Campeche"],
                                       [5, "Coahuila de Zaragoza"],
                                       [6, "Colima"],
                                       [7, "Chiapas"],
                                       [8, "Chihuahua"],
                                       [9, "Distrito Federal"],
                                       [10, "Durango"],
                                       [11, "Guanajuato"],
                                       [12, "Guerrero"],
                                       [13, "Hidalgo"],
                                       [14, "Jalisco"],
                                       [15, "México"],
                                       [16, "Michoacán de Ocampo"],
                                       [17, "Morelos"],
                                       [18, "Nayarit"],
                                       [19, "Nuevo León"],
                                       [20, "Oaxaca"],
                                       [21, "Puebla"],
                                       [22, "Querétaro"],
                                       [23, "Quintana Roo"],
                                       [24, "San Luis Potosí"],
                                       [25, "Sinaloa"],
                                       [26, "Sonora"],
                                       [27, "Tabasco"],
                                       [28, "Tamaulipas"],
                                       [29, "Tlaxcala"],
                                       [30, "Veracruz"],
                                       [31, "Yucatán"],
                                       [32, "Zacatecas"],
                                   ]
                                   )
    educacion = models.IntegerField(label="",
                               choices=[
                                   [1, "No asistió"],
                                   [2, "Primaria"],
                                   [3, "Secundaria técnica"],
                                   [4, "Secundaria general"],
                                   [5, "Preparatoria técnica"],
                                   [6, "Preparatoria general"],
                                   [7, "Técnica o comercial con secundaria"],
                                   [8, "Técnica o comercial con preparatoria"],
                                   [9, "Normal básica (con primaria o secundaria)"],
                                   [10, "Normal de licenciatura"],
                                   [11, "Profesional (licenciatura o ingeniería)"],
                                   [12, "Postgrado (maestría o doctorado)"],
                               ]
                               )
    empleo = models.IntegerField(label="",
                                   choices=[
                                       [1, "Empleado a tiempo completo"],
                                       [2, "Empleado a tiempo parcial"],
                                       [3, "Cuenta propia o emprendedor"],
                                       [4, "Desempleado y buscando trabajo"],
                                       [5, "Estudiante"],
                                       [6, "No estoy en la fuerza laboral(por ejemplo es padre de tiempo completo)"],
                                   ]
                                   )

    ingreso_hogar = models.IntegerField(label="Esta pregunta tiene la dificultad de que podemos inducir a los participantes sobre ciertas distribuciones del ingreso. Discutir.",
                                        choices=[
                                            [1, "$0 - $9,999"],
                                            [2, "$10,000 - $15,999"],
                                            [3, "$16,000 - $20,999"],
                                            [4, "$21,000 - $25,999"],
                                            [5, "$26,000 - $31,999"],
                                            [6,"$32,000 - $38,999"],
                                            [7, "$39,000 - $46,999"],
                                            [8, "$47,000 - $58,999"],
                                            [9, "$59,000 - $69,999"],
                                            [10, "$70,000 - $81,999"],
                                            [11, "$82,000 - $93,999"],
                                            [12, "$94,000 - $109,999"],
                                            [13, "$110,000 - $126,999"],
                                            [14, "$127,000 - $140,999"],
                                            [15, "$141,000 - $156,999"],
                                            [16, "$157,000 - $172,999"],
                                            [17, "$173,000 - $187,999"],
                                            [18, "$188,000+"],

                                        ]
                                   )
    percepcion_1 = models.IntegerField(label="",
        choices=[0,1, 2, 3,4,5,6,7,8,9,10],
        widget=widgets.RadioSelectHorizontal
    )

    perspectiva = models.IntegerField(label="",
                        choices=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
                        widget=widgets.RadioSelectHorizontal
                        )

    ingreso_pobre = models.PositiveIntegerField(label="",
                                      )
    monto_sacrificar= models.IntegerField(label="Monto",min=0, max=10000
                        )
    sociedad_nivel = models.IntegerField(label="Ahora, quiero pedirle que elija, desde su punto de vista, cuál de las imágenes representa a la sociedad mexicana.PARA NUESTRO ESTUDIO DEBEMOS ADAPTARLO AL CASO MEXICANO",
                                 choices=[
                                     [1, "Tipo A"],
                                     [2, "Tipo B"],
                                     [3, "Tipo C"],
                                     [4, "Tipo D"],
                                     [5, "Tipo E"],
                                 ]
                                 )