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

author = ''

doc = """

"""


class Constants(BaseConstants):
    name_in_url = 'G3'
    players_per_group = 3
    num_rounds = 1
    answers_Motiv_a = ['b', 'c', 'b', 'a']
    answers_NoMotiv = ['b', 'c', 'b', 'b']
    pre_pilot = False
    outcome_part1 = 50
    outcome_norm_pun = 50
    outcome_norm_nopun = 50


class Subsession(BaseSubsession):

    def creating_session(self):
        if self.round_number == 1:
            for p in self.get_players():
                # p.random_role = random.randint(1, 3)
                p.random_part = random.randint(1, 2)
                p.random_order = random.randint(1, 2)
                p.computer_n = random.randint(0, 100)
                p.lottery_draw = random.randint(0, 100)
                p.lottery_outcome_draw = random.randint(0, 100)
        for group in self.get_groups():
            group.random_number = random.randint(1, 5)
            if group.random_number == 1:
                group.random_game = 1
            else:
                group.random_game = 2
            # group.random_game = 1  # manually set game 1 or 2

        # new_structure_a = [[1, 13, 14], [2, 15, 16], [3, 17, 18], [4, 19, 20],
        #                    [5, 21, 22], [6, 23, 24], [7, 25, 26], [8, 27, 28],
        #                    [9, 29, 30], [10, 31, 32], [11, 33, 34], [12, 35, 36]]

        # new_structure_b = [[13, 1, 14], [15, 2, 16], [17, 3, 18], [19, 4, 20],
        #                   [21, 5, 22], [23, 6, 24], [25, 7, 26], [27, 8, 28],
        #                   [29, 9, 30], [31, 10, 32], [33, 11, 34], [35, 12, 36]]

        # new_structure_c = [[14, 13, 1], [16, 15, 2], [18, 17, 3], [20, 19, 4],
        #                    [22, 21, 5], [24, 23, 6], [26, 25, 7], [28, 27, 8],
        #                    [30, 29, 9], [32, 31, 10], [34, 33, 11], [36, 35, 12]]

        # self.set_group_matrix(new_structure_a)
        # self.set_group_matrix(new_structure_b)
        # self.set_group_matrix(new_structure_c)


class Group(BaseGroup):
    random_game = models.IntegerField()
    random_number = models.IntegerField()


class Player(BasePlayer):
    accept = models.BooleanField()
    ProlificID = models.StringField()
    paid = models.BooleanField()  # selected 1 in 20 to be paid
    # random_role = models.IntegerField()  # player a, b or c
    random_part = models.IntegerField()  # either part 1 or part 2 for payment
    random_order = models.IntegerField()
    # karni mechanism
    computer_n = models.IntegerField()  # computer draws number (N)
    lottery_played = models.BooleanField() # whether simple lottery is played or outcome lottery
    lottery_played_norm_pun = models.BooleanField()
    lottery_played_norm_nopun = models.BooleanField()
    # if N>=guess, play simple lottery
    lottery_draw = models.IntegerField()  # draw a number to determine lottery outcome
    lottery_win = models.BooleanField()
    lottery_win_norm_pun = models.BooleanField()  # if lottery_draw <= computer_n, win, else, lose
    lottery_win_norm_nopun = models.BooleanField()
    # if N<guess, play outcome lottery
    lottery_outcome_draw = models.IntegerField()  # draw a number to determine outcome lottery outcome
    # if lottery_outcome_draw <= actual share, win, else, lose
    lottery_outcome_win = models.BooleanField()
    lottery_outcome_win_norm_pun = models.BooleanField()
    lottery_outcome_win_norm_nopun = models.BooleanField()

    # Comprehension questions:
    q1 = models.StringField(
        label='',
        choices=[['a', 'Player A'],
                 ['b', 'Player A and Player C'],
                 ['c', 'Player B'],
                 ['d', 'All Players']],
        widget=widgets.RadioSelect)
    q2 = models.StringField(
        label='',
        choices=[['a', 'Player A: £6, Player B: £10, and Player C: £10'],
                 ['b', 'Player A: £12, Player B: £12, and Player C: £6'],
                 ['c', 'Player A: £10, Player B: £10, and Player C: £6'],
                 ['d', 'Player A: £6, Player B: £12, and Player C: £12']],
        widget=widgets.RadioSelect)
    q3 = models.StringField(
        label='',
        choices=[['a', 'Between Player B and C: 8 in 10 chance, Between Player A and B: 2 in 10 chance'],
                 ['b', 'Between Player B and C: 2 in 10 chance, Between Player A and B: 8 in 10 chance'],
                 ['c', 'Between Player B and C: 5 in 10 chance, Between Player A and B: 5 in 10 chance']],
        widget=widgets.RadioSelect)
    q4 = models.StringField(
        label='',
        choices=[['a', 'Player A'],
                 ['b', 'Player B'],
                 ['c', 'Player C']],
        widget=widgets.RadioSelect
    )
    q1a = models.StringField(
        label='',
        choices=[['a', 'Player A'],
                 ['b', 'Player A and Player C'],
                 ['c', 'Player B'],
                 ['d', 'All Players']],
        widget=widgets.RadioSelect)
    q2a = models.StringField(
        label='',
        choices=[['a', 'Player A: £6, Player B: £10, and Player C: £10'],
                 ['b', 'Player A: £12, Player B: £12, and Player C: £6'],
                 ['c', 'Player A: £10, Player B: £10, and Player C: £6'],
                 ['d', 'Player A: £6, Player B: £12, and Player C: £12']],
        widget=widgets.RadioSelect)
    q3a = models.StringField(
        label='',
        choices=[['a', 'Between Player B and C: 8 in 10 chance, Between Player A and B: 2 in 10 chance'],
                 ['b', 'Between Player B and C: 2 in 10 chance, Between Player A and B: 8 in 10 chance'],
                 ['c', 'Between Player B and C: 5 in 10 chance, Between Player A and B: 5 in 10 chance']],
        widget=widgets.RadioSelect)
    q4a = models.StringField(
        label='',
        choices=[['a', 'Player A'],
                 ['b', 'Player B'],
                 ['c', 'Player C']],
        widget=widgets.RadioSelect
    )

    answers_correct = models.IntegerField(initial=0)
    answers_correct_a = models.IntegerField(initial=0)

    # Motiv for Player A only
    def check_answers_motiv(self):
        answers_given = [self.q1, self.q2, self.q3, self.q4]
        for i in range(1, 5):
            self.answers_correct += (answers_given[i - 1] or "").lower() == Constants.answers_Motiv_a[i - 1].lower()

    def check_answers_motiv_a(self):
        answers_given = [self.q1a, self.q2a, self.q3a, self.q4a]
        for i in range(1, 5):
            self.answers_correct_a += (answers_given[i - 1] or "").lower() == Constants.answers_Motiv_a[i - 1].lower()

    # NoMotiv, Motiv for Player B and C
    def check_answers_nomotiv(self):
        answers_given = [self.q1, self.q2, self.q3, self.q4]
        for i in range(1, 5):
            self.answers_correct += (answers_given[i - 1] or "").lower() == Constants.answers_NoMotiv[i - 1].lower()

    def check_answers_nomotiv_a(self):
        answers_given = [self.q1a, self.q2a, self.q3a, self.q4a]
        for i in range(1, 5):
            self.answers_correct_a += (answers_given[i - 1] or "").lower() == Constants.answers_NoMotiv[i - 1].lower()

    # Beliefs (Part 1)
    posterior_belief = models.IntegerField(label='', min=0, max=100, initial=0)
    posterior_belief_inverse = models.IntegerField(label='', min=0, max=100)
    norm1_pun_EP = models.IntegerField(label='',
                                         choices=[[1, 'Very inappropriate'],
                                                  [2, 'Somewhat inappropriate'],
                                                  [3, 'Neither inappropriate nor appropriate'],
                                                  [4, 'Somewhat appropriate'],
                                                  [5, 'Very appropriate']])
    norm1_nopun_EP = models.IntegerField(label='',
                                           choices=[[1, 'Very inappropriate'],
                                                    [2, 'Somewhat inappropriate'],
                                                    [3, 'Neither inappropriate nor appropriate'],
                                                    [4, 'Somewhat appropriate'],
                                                    [5, 'Very appropriate']])
    norm1_pun_CP = models.IntegerField(label='',
                                       choices=[[1, 'Very inappropriate'],
                                                [2, 'Somewhat inappropriate'],
                                                [3, 'Neither inappropriate nor appropriate'],
                                                [4, 'Somewhat appropriate'],
                                                [5, 'Very appropriate']])
    norm1_nopun_CP = models.IntegerField(label='',
                                         choices=[[1, 'Very inappropriate'],
                                                  [2, 'Somewhat inappropriate'],
                                                  [3, 'Neither inappropriate nor appropriate'],
                                                  [4, 'Somewhat appropriate'],
                                                  [5, 'Very appropriate']])
    norm2_pun_EP = models.IntegerField(label='',
                                       choices=[[1, 'Very inappropriate'],
                                                [2, 'Somewhat inappropriate'],
                                                [3, 'Neither inappropriate nor appropriate'],
                                                [4, 'Somewhat appropriate'],
                                                [5, 'Very appropriate']])
    norm2_nopun_EP = models.IntegerField(label='',
                                         choices=[[1, 'Very inappropriate'],
                                                  [2, 'Somewhat inappropriate'],
                                                  [3, 'Neither inappropriate nor appropriate'],
                                                  [4, 'Somewhat appropriate'],
                                                  [5, 'Very appropriate']])
    norm2_pun_CP = models.IntegerField(label='',
                                       choices=[[1, 'Very inappropriate'],
                                                [2, 'Somewhat inappropriate'],
                                                [3, 'Neither inappropriate nor appropriate'],
                                                [4, 'Somewhat appropriate'],
                                                [5, 'Very appropriate']])
    norm2_nopun_CP = models.IntegerField(label='',
                                         choices=[[1, 'Very inappropriate'],
                                                  [2, 'Somewhat inappropriate'],
                                                  [3, 'Neither inappropriate nor appropriate'],
                                                  [4, 'Somewhat appropriate'],
                                                  [5, 'Very appropriate']])
    emp_norm_pun_EP = models.IntegerField(label='',
                                             min=0, max=100)
    emp_norm_nopun_EP = models.IntegerField(label='',
                                             min=0, max=100)
    confirm = models.StringField(label='',
                                 choices=[['yes', 'Yes, continue to Part 2.'],
                                          ['no', 'No, I wish to revise my answer.']],
                                 widget=widgets.RadioSelect)
    more_info_part1 = models.BooleanField(label='')  # = 1 if "see more details" was clicked
    more_info_report = models.BooleanField(label='')
    more_info_norm = models.BooleanField(label='')
    more_info_norm_report = models.BooleanField(label='')

    # Task (Part 2) decisions
    impose_fee = models.BooleanField(label='')
    project_choice_pun = models.StringField(label='Suppose Player A decided to impose a fee '
                                                  'against you for choosing the Exclusive Project. '
                                                  'Please select a project.',
                                            choices=[['CP', 'Communal Project'],
                                                     ['EP', 'Exclusive Project']])
    project_choice_nopun = models.StringField(label='Suppose Player A decided NOT to impose a fee '
                                                    'against you for choosing the Exclusive Project. '
                                                    'Please select a project.',
                                              choices=[['CP', 'Communal Project'],
                                                       ['EP', 'Exclusive Project']])

    # Survey questions
    gender = models.StringField(label='What is your gender?',
                                choices=[['male', 'Male'],
                                         ['female', 'Female'],
                                         ['non-binary', 'Non-binary'],
                                         ['gender-diverse', 'Gender Diverse'],
                                         ['na', 'My gender is not listed'],
                                         ['not-say', 'Prefer not to say']])
    age = models.IntegerField(label='What is your age?')
    education = models.StringField(label='What is the highest level of education you have completed?',
                                   choices=[['none', 'No formal qualifications'],
                                            ['highschool', 'High school diploma'],
                                            ['communitycollege', 'Technical/community college'],
                                            ['undergrad', 'Undergraduate degree'],
                                            ['graduate', 'Graduate degree'],
                                            ['NA', 'Not applicable'],
                                            ['not-say', 'Prefer not to say']])
    religiosity = models.StringField(label='My religion is very important to me',
                                     choices=[['SD', 'Strongly disagree'],
                                              ['D', 'Disagree'],
                                              ['N', 'Neither agree nor disagree'],
                                              ['A', 'Agree'],
                                              ['SA', 'Strongly agree'],
                                              ['not-say', 'Prefer not to say']])
    income = models.StringField(label='What is your average household income per year?',
                                choices=[['<20k', 'Less than £20,000'],
                                         ['<40>k', '£20,000 - £39,000'],
                                         ['<60>k', '£40,000 - £59,000'],
                                         ['<80>k', '£60,000 - £79,000'],
                                         ['<100k', '£80,000 - £99,000'],
                                         ['>100k', 'More than £100,000'],
                                         ['not-say', 'Prefer not to say'] ])
    political = models.StringField(label='What is your political orientation?',
                                   choices=['Left', 'Centre-Left', 'Centre-Right', 'Right', 'Other',
                                            'Prefer not to say'])
    open_ended = models.LongStringField(label='')
    open_ended_B = models.LongStringField(label='')
    feedback = models.LongStringField(label='Any general comments or feedback '
                                            '(e.g. anything we can improve in the instructions)?', blank=True)