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

author = 'Your name here'

doc = """

"""


class Constants(BaseConstants) :
    name_in_url = 'public_goods_scenario'
    players_per_group = 6
    num_rounds = 5
    mpcr = 0.8
    endowment = c(50)


class Subsession(BaseSubsession) :
    pass


class Group(BaseGroup) :
    total_contribution = models.CurrencyField()
    individual_share = models.CurrencyField()

    def set_payoffs(self) :

        players = self.get_players()
        r = models.CurrencyField()
        q = 0
        contributions = [p.contribution for p in players]
        self.total_contribution = sum(contributions)
        self.individual_share = self.total_contribution * Constants.mpcr

        for p in self.get_players() :
            p.payoff = Constants.endowment - p.contribution + self.individual_share

        while self.round_number == 5 :
            for p in self.get_players() :
                l = [p.in_round(5).payoff, p.in_round(4).payoff, p.in_round(3).payoff, p.in_round(1).payoff,
                     p.in_round(2).payoff]
            for p in self.get_players() :
                p.final_payoff = random.choice(l, replace=True)
            self.round_number = self.round_number + 1


class Player(BasePlayer) :
    contribution = models.CurrencyField(min=0,
                                        label="How much you want to contribute",
                                        max=Constants.endowment)
    final_payoff = models.CurrencyField(initial=0, label='Player earnings')