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

author = 'LBouman'

doc = """
SPG 
"""

class Constants(BaseConstants):
    name_in_url = 'SPG'
    players_per_group = 5
    num_rounds = 1 # in exp set to 15 for spg session decision page

    num_other_players = players_per_group - 1

    general_benefit = c(15)
    investment_cost = c(10)

class Subsession(BaseSubsession):

    share_distribution = ""
    information_condition = ""

# group formation
    def creating_session(self):

        self.share_distribution = self.session.config['share_distribution']  # homogeneous / heterogeneous
        self.information_condition = self.session.config['information_condition']  # complete / incomplete

        if self.round_number == 1:
            self.group_randomly()
        else:
            self.group.like.round(1)

        # role assignment
        shares = {'heterogeneous': [2, 8, 20, 20, 50], 'homogeneous': [20, 20, 20, 20, 20]}

        for p in self.get_players():
            p.share = shares[self.share_distribution][p.in_round(1).id_in_group - 1]


class Group(BaseGroup):
# SPG global group vars
    num_investors = models.IntegerField()
    sum_shares = models.IntegerField()

    def good_is_produced(self):
        players = self.get_players()
        self.num_investors = sum([p.invest for p in players])
        self.sum_shares = sum([p.share for p in players if p.invest])

        if self.sum_shares > 51:
            return True
        else:
            return False

    def set_payoffs(self):
        players = self.get_players()
        if self.good_is_produced():
            baseline_amount = Constants.general_benefit
        else:
            baseline_amount = c(0)
        for p in players:
            p.payoff = baseline_amount
            if not p.invest:
                p.payoff += Constants.investment_cost


class Player(BasePlayer):
# SGP model code
    invest = models.BooleanField()
    share = models.IntegerField()


# efficacy questions
    E1 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
        [7, "7. Certain"],
    ],
        verbose_name="To what extent is your investment necessary to produce the group project? It is ... that my investment is necessary to produce the group project."
    )
    E2 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
        [7, "7. Certain"],
    ],
        verbose_name=" To what extent is your investment sufficient to produce the group project? It is ... that my investment is sufficient to produce the group project. "
    )
    E3 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
        [7, "7. Certain"],
    ],
        verbose_name="To what extent does your investment determine whether or not the group project is produced? It is ... that my investment determines whether or not the group project is produced. "
    )

    E4 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
        [7, "7. Certain"],
    ],
    verbose_name="How likely do you think it is that another group member will invest, given your share? It is ... that another groupmember will invest."
    )

    E5 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
        [7, "7. Certain"],
    ],
    verbose_name="How likely do you think it is that the group project will be produced within your group?It is ... that the group project will be produced in my group. "
    )

# add HET efficacy questions

    E6 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
		[7, "7. Certain"],
    ],
        verbose_name = "How likely do you think it is that an individual whose share is 2 will invest his or her 10 points?
						It is ... that an individual whose share is 2 will invest his or her 10 points. "
	)

    E7 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
		[7, "7. Certain"],
    ],
        verbose_name = "How likely do you think it is that an individual whose share is 16 will invest his or her 10 points? It is ... that an individual whose share is 16 will invest his or her 10 points. "
	)
	E8 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
		[7, "7. Certain"],
    ],
        verbose_name = "How likely do you think it is that an individual whose share is 50 will invest his or her 10 points? It is ... that an individual whose share is 50 will invest his or her 10 points. "
	)

	E9 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
		[7, "7. Certain"],
    ],
        verbose_name = "To what extent is your investment necessary to produce the group project? It is ... that my investment is necessary to produce the group project."
	)

	E10 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
		[7, "7. Certain"],
    ],
        verbose_name = "To what extent is your investment sufficient to produce the group project? It is ... that my investment is sufficient to produce the group project. "
	)

	E11 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
		[7, "7. Certain"],
    ],
        verbose_name = "To what extent does your investment determine whether or not the group project is produced? It is ... that my investment determines whether or not the group project is produced. "
	)

	E12 = models.IntegerField(choices=[
        [1, "1. Impossible"],
        [2, "2. Very unlikely"],
        [3, "3. Unlikely"],
        [4, "4. A 50/50 chance"],
        [5, "5. Likely"],
        [6, "6. Very likely"],
		[7, "7. Certain"],
    ],
        verbose_name = "How likely do you think it is that the group project will be produced within your group?It is ... that the group project will be produced in my group."
	)