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

#from otree_tools.models.fields import OtherModelField

import random
import itertools

author = 'FX CHEN'

doc = """
This is a TPPG with leader not punishing.
"""


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

    instructions_template = 'htlj3/aInstructions.html'

    endowment = c(90)
    initial_endowment = c(90)

    pool_multiplier = c(1.6)

    answer_increment = c(2)
    take_away_endo = c(90)
    leader_decision = currency_range(0, take_away_endo, answer_increment)
    leader_decision_count = len(leader_decision)



class Subsession(BaseSubsession):

    def creating_session(self):
        players = self.get_players()
        self.group_randomly()


class Group(BaseGroup):

    total_removed = models.CurrencyField()
    removed = models.CurrencyField()
    total_contribution = models.CurrencyField()

    individual_share = models.CurrencyField()

    decision_member1 = models.CurrencyField(choices=Constants.leader_decision,
                                            label='')
    decision_member2 = models.CurrencyField(choices=Constants.leader_decision,
                                            label='')
    decision_member3 = models.CurrencyField(choices=Constants.leader_decision,
                                            label='')

    current_payoff = models.CurrencyField()


    p1_current = models.CurrencyField()
    p2_current = models.CurrencyField()
    p3_current = models.CurrencyField()

    p1_payoff2 = models.CurrencyField()
    p2_payoff2 = models.CurrencyField()
    p3_payoff2 = models.CurrencyField()

    def set_payoffs(self):
        self.total_contribution = sum([p.contribution for p in self.get_players()]) + 60
        self.individual_share = self.total_contribution * Constants.pool_multiplier / 4
        self.total_removed = (self.decision_member3 + self.decision_member1 + self.decision_member2)/2
        self.removed = 55 - sum([p.contribution for p in self.get_players()])
        for p in self.get_players():
            if p.contribution >= 55:
                p.p1_payoff2 = sum([+73, -39,
                                     -self.decision_member1,
                                     +self.individual_share,
                                     ])
                p.p2_payoff2 = sum([+66, -30,
                                     -self.decision_member2,
                                     +self.individual_share,
                                     ])
                p.p3_payoff2 = sum([+71, -39,
                                     -self.decision_member3,
                                     +self.individual_share,
                                     ])
                p.payoff = Constants.endowment - p.contribution + self.individual_share - (self.decision_member1/2) - (self.decision_member2/2) - (self.decision_member3/2) - 0
                p.payoff2 = Constants.endowment - p.contribution + self.individual_share - self.removed
                p.payoff3 = (Constants.endowment - p.contribution + self.individual_share - self.removed) * 2
                p.removed = 0
            if p.contribution < 55:
                p.p1_payoff2 = sum([+73, -39,
                                     -self.decision_member1,
                                     +self.individual_share,
                                     ])
                p.p2_payoff2 = sum([+66, -30,
                                     -self.decision_member2,
                                     +self.individual_share,
                                     ])
                p.p3_payoff2 = sum([+71, -39,
                                     -self.decision_member3,
                                     +self.individual_share,
                                     ])
                p.payoff = Constants.endowment - p.contribution + self.individual_share - (self.decision_member1/2) - (self.decision_member2/2) - (self.decision_member3/2) - self.removed
                p.payoff2 = Constants.endowment - p.contribution + self.individual_share - self.removed
                p.payoff3 = (Constants.endowment - p.contribution + self.individual_share - self.removed) * 2
                p.removed = self.removed


    def current(self):
        self.total_contribution= sum([p.contribution for p in self.get_players()]) + 60
        self.individual_share = self.total_contribution * Constants.pool_multiplier / 4
        self.p1_current = sum([+73,+self.individual_share, -39])
        self.p2_current = sum([+66,+self.individual_share, -30])
        self.p3_current = sum([+71,+self.individual_share, -39])
        self.removed = 55 - sum([p.contribution for p in self.get_players()])
        for p in self.get_players():
            if p.contribution >= 55:
                p.p1_payoff2 = sum([+73, -39,
                                    +self.individual_share,
                                    ])
                p.p2_payoff2 = sum([+66, -30,
                                    +self.individual_share,
                                    ])
                p.p3_payoff2 = sum([+71, -39,
                                    +self.individual_share,
                                    ])
                p.payoff2 = Constants.endowment - p.contribution + self.individual_share - 0
                p.payoff3 = (Constants.endowment - p.contribution + self.individual_share - 0)*2
                p.removed = 0
            elif p.contribution < 55:
                p.p1_payoff2 = sum([+73, -39,
                                     +self.individual_share,
                                     ])
                p.p2_payoff2 = sum([+66, -30,
                                     +self.individual_share,
                                     ])
                p.p3_payoff2 = sum([+71, -39,
                                     +self.individual_share,
                                     ])
                p.payoff2 = Constants.endowment - p.contribution + self.individual_share  - self.removed
                p.payoff3 = (Constants.endowment - p.contribution + self.individual_share - self.removed) * 2
                p.removed = self.removed





class Player(BasePlayer):
    total_removed = models.CurrencyField()
    removed = models.CurrencyField()
    pointremoved = models.CurrencyField()

    current_payoff = models.CurrencyField()
    current_payoff2 = models.CurrencyField()

    payoff2 = models.CurrencyField()
    payoff3 = models.CurrencyField()

    p1_payoff2 = models.CurrencyField()
    p2_payoff2 = models.CurrencyField()
    p3_payoff2 = models.CurrencyField()

    p1_current = models.CurrencyField()
    p2_current = models.CurrencyField()
    p3_current = models.CurrencyField()

    contribution = models.CurrencyField(
        min=0, max=Constants.endowment,
        doc="""The amount contributed by the player""",
    )

    def member_id(self):
        # player 1 is the leader, so member 1 is actually player 2
        return (self.id_in_group)