from otree.api import *

c = Currency

doc = """
Treatment Questionnaire
"""


class Constants(BaseConstants):
    name_in_url = 'questionnaire_time_lags'
    players_per_group = None
    num_rounds = 1
    completion_code = '3078432'


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    understanding = models.IntegerField(
        choices=[
            [1, 'Absolutely not'],
            [2, 'More no than yes'],
            [3, 'More yes than no'],
            [4, 'Absolutely yes'],
        ],
        label="1) Did you understand how the experiment worked?",
    )
    past_experience = models.BooleanField(
        choices=[
            [True, 'Yes'],
            [False, 'No'],
        ],
        label="2) Have you participated in similar experiments in the past?",
    )
    group_discussion = models.IntegerField(
        choices=[
            [1, 'Completely disagree'],
            [2, 'Disagree'],
            [3, 'Agree'],
            [4, 'Completely agree'],
        ],
        label="3) In the group, we discussed how much everyone should harvest."
    )
    delay_others = models.IntegerField(
        choices=[
            [1, 'More aggressively'],
            [2, 'A bit more aggressively'],
            [3, 'A bit more cautiously'],
            [4, 'More cautiously'],
        ],
        label="4) Given the presence of a delayed change in stock B, how did you think the other group members would "
              "harvest?"
    )
    delay_you = models.IntegerField(
        choices=[
            [1, 'More aggressively'],
            [2, 'A bit more aggressively'],
            [3, 'A bit more cautiously'],
            [4, 'More cautiously'],
        ],
        label="5) Given the presence of a delayed change in stock B, how did you decide to harvest?"
    )
    cooperation_stock = models.IntegerField(
        choices=[
            [1, 'Completely disagree'],
            [2, 'Disagree'],
            [3, 'Agree'],
            [4, 'Completely agree'],
        ],
        label="6) I believe that cooperation between the group members was necessary for sustaining the stock of "
              "resources."
    )
    cooperation_payoff = models.IntegerField(
        choices=[
            [1, 'Completely disagree'],
            [2, 'Disagree'],
            [3, 'Agree'],
            [4, 'Completely agree'],
        ],
        label="7) I believe that cooperation between the group members was necessary for maximizing my payoff."
    )
    equality_stock = models.IntegerField(
        choices=[
            [1, 'Completely disagree'],
            [2, 'Disagree'],
            [3, 'Agree'],
            [4, 'Completely agree'],
        ],
        label="8) I believe that equally distributed harvest shares were necessary to sustain the stock of resources."
    )
    equality_payoff = models.IntegerField(
        choices=[
            [1, 'Completely disagree'],
            [2, 'Disagree'],
            [3, 'Agree'],
            [4, 'Completely agree'],
        ],
        label="9) I believe that equally distributed harvest shares were necessary for maximizing my payoff."
    )
    strategy = models.BooleanField(
        choices=[
            [True, 'Yes'],
            [False, 'No'],
        ],
        label="10) Did the group develop a (any kind of) strategy or agreement?"
    )
    timing_of_strategy = models.IntegerField(
        choices=[
            [0, 'Never'],
            [1, 'The beginning (round 1 to 3)'],
            [2, 'A bit after the beginning (round 4 to 6)'],
            [3, 'The middle of the game (round 7 to 9)'],
            [4, 'Late in the game (round 10 onwards)'],
        ],
        label="11) The group stuck to the common strategy or agreement throughout the game since:"
    )
    broken_agreements_you = models.IntegerField(
        choices=[
            [0, 'Not applicable (no agreement)'],
            [1, 'Never'],
            [2, 'Yes, few times'],
            [3, 'Yes, most of times'],
            [4, 'Yes, at all times'],
        ],
        label="12) Did you respect the agreement?"
    )
    broken_agreements_others = models.IntegerField(
        choices=[
            [0, 'Not applicable (no agreement)'],
            [1, 'Never'],
            [2, 'Yes, few times'],
            [3, 'Yes, most of times'],
            [4, 'Yes, at all times'],
        ],
        label="13) Did you feel the other group members respected the agreement?"
    )
    collapse = models.IntegerField(
        choices=[
            [1, 'No'],
            [2, 'Once'],
            [3, 'More than once'],
        ],
        label="14) Did the stock ever fall below 20?"
    )
    agreement_before_collapse = models.IntegerField(
        choices=[
            [0, 'Not applicable (always above 20)'],
            [1, 'No'],
            [2, 'Yes'],
        ],
        label="15) Did the group reach an agreement before the stock fell below 20?"
    )
    agreement_after_collapse = models.IntegerField(
        choices=[
            [0, 'Not applicable (always above 20)'],
            [1, 'No'],
            [2, 'Yes'],
        ],
        label="16) Did the group reach an agreement after the stock fell below 20?"
    )
    trust = models.IntegerField(
        choices=[
            [1, 'Completely disagree'],
            [2, 'Disagree'],
            [3, 'Agree'],
            [4, 'Completely agree'],
        ],
        label="17) I trusted my group members."
    )
    timing_of_trust = models.IntegerField(
        choices=[
            [0, 'Never'],
            [1, 'The beginning (round 1 to 3)'],
            [2, 'A bit after the beginning (round 4 to 6)'],
            [3, 'The middle of the game (round 7 to 9)'],
            [4, 'Late in the game (round 10 onwards)'],
        ],
        label="18) I trusted my group members since:"
    )
    realistic_payment = models.IntegerField(
        choices=[
            [1, 'More aggressively'],
            [2, 'A bit more aggressively'],
            [3, 'A bit more cautiously'],
            [4, 'More cautiously'],
        ],
        label="19) If you were paid with real money for how much you harvested during the experiment, how would have you played?"
    )
    gender = models.IntegerField(
        choices=[
            [1, 'Female'],
            [2, 'Male'],
            [3, 'Other'],
            [4, 'Prefer not to say'],
        ],
        label="20) Please specify your gender."
    )
    age = models.IntegerField(
        label="21) Please specify your age."
    )
    education = models.IntegerField(
        choices=[
            [1, 'No formal education'],
            [2, 'Primary school'],
            [3, 'Secondary school'],
            [4, 'Undergraduate degree'],
            [5, 'Graduate degree or higher'],
        ],
        label="22) What is your level of education?"
    )
    comments = models.LongStringField(
        label="23) Do you have any comments (e.g. misunderstandings, suggestions for improvements, errors, etc.)?"
    )


# PAGES
class Questionnaire(Page):
    form_model = "player"
    form_fields = ["understanding", "past_experience", "group_discussion", "delay_others", "delay_you",
                   "cooperation_stock", "cooperation_payoff", "equality_stock", "equality_payoff", "strategy", "timing_of_strategy", "broken_agreements_you", "broken_agreements_others",
                   "collapse", "agreement_before_collapse", "agreement_after_collapse", "trust",
                   "timing_of_trust", "realistic_payment", "gender", "age", "education", "comments"]


page_sequence = [Questionnaire]