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


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


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    bank_account = models.StringField()
    treatment_order = models.LongStringField()
    podcast_announce = models.IntegerField()
    game_finished = models.IntegerField()
    paying_round = models.IntegerField()
    correct_adding1 = models.IntegerField()
    correct_adding2 = models.IntegerField()
    correct_adding3 = models.IntegerField()
    time_spent = models.FloatField()
    points = models.IntegerField(min=-1000)
    total_green = models.IntegerField()
    total_gray = models.IntegerField()
    total_wrong = models.IntegerField()
    total_right = models.IntegerField()
    wrong_green = models.IntegerField()  #correct should be green
    wrong_gray = models.IntegerField()   #correct should be gray
    right_green = models.IntegerField()
    right_gray = models.IntegerField()
    board_history = models.LongStringField() #if each cell is correct or not (1 for correct color, 0 for wrong color)
    time_between_moves = models.LongStringField()
    solve_order = models.LongStringField()
    h1 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label= "I see myself as someone who worries a lot.",

        widget=widgets.RadioSelectHorizontal
    )

    h2 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who gets nervous easily.",

        widget=widgets.RadioSelectHorizontal
    )

    h3 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who remains calm in tense situations.",

        widget=widgets.RadioSelectHorizontal
    )

    h4 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is talkative.",

        widget=widgets.RadioSelectHorizontal
    )

    h5 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is outgoing, sociable.",

        widget=widgets.RadioSelectHorizontal
    )

    h6 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is reserved.",

        widget=widgets.RadioSelectHorizontal
    )

    h7 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is original, comes up with new ideas.",

        widget=widgets.RadioSelectHorizontal
    )

    h8 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who values artistic, aesthetic experiences.",
        widget=widgets.RadioSelectHorizontal
    )

    h9 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who has an active imagination.",
        widget=widgets.RadioSelectHorizontal
    )

    h10 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is sometimes rude to others.",
        widget=widgets.RadioSelectHorizontal
    )

    h11 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who has a forgiving nature.",
        widget=widgets.RadioSelectHorizontal
    )

    h12 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is considerate and kind to almost anyone.",
        widget=widgets.RadioSelectHorizontal
    )

    h13 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who does a thorough job.",
        widget=widgets.RadioSelectHorizontal
    )

    h14 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who tends to be lazy.",
        widget=widgets.RadioSelectHorizontal
    )

    h15 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who does things efficiently.",
        widget=widgets.RadioSelectHorizontal
    )

    h16 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who enjoys working under time pressure.",
        widget=widgets.RadioSelectHorizontal
    )

    h17 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is productive under time pressure.",
        widget=widgets.RadioSelectHorizontal
    )

    h18 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is competitive.",
        widget=widgets.RadioSelectHorizontal
    )

    h19 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I see myself as someone who is willing to take risks.",
        widget=widgets.RadioSelectHorizontal
    )

    h20 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I dislike working in distracting environments.",
        widget=widgets.RadioSelectHorizontal
    )

    h21 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I can focus well in noisy environments.",
        widget=widgets.RadioSelectHorizontal
    )

    h22 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I can easily concentrate after being interrupted.",
        widget=widgets.RadioSelectHorizontal
    )

    h23 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I am good at working on several projects at the same time.",
        widget=widgets.RadioSelectHorizontal
    )

    h24 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I enjoy working on several projects at the same time.",
        widget=widgets.RadioSelectHorizontal
    )

    h25 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I have trouble limiting my phone usage.",
        widget=widgets.RadioSelectHorizontal
    )

    h26 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I constantly check my phone while studying.",
        widget=widgets.RadioSelectHorizontal
    )

    h27 = models.StringField(
        choices=["Strongly Disagree", "Disagree", "Slightly Disagree", "Neutral", "Slightly Agree", "Agree", "Strongly Agree"],

        label="I am easily distracted.",
        widget=widgets.RadioSelectHorizontal
    )


#h1, h2, h3(R): neuroticism
#h4, h5, h6(R): extraversion
#h7, h8, h9: openness
#h10(R), h11, h12: agreeableness
#h13, h14(R), h15: conscientiousness
#h16, h17: time pressure
#h18: competitiveness
#h19: risk taking
#h20(R), h21, h22, h27: distractions
#h23, h24: multitasking
#h25(R), h26(R): phone using