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


author = 'Your name here'

doc = """
Your app description
"""


class Constants(BaseConstants):
    name_in_url = 'first_publicgood'
    players_per_group = 3
    num_rounds = 1

    endowment = 20

    multiplier = 2

class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    total_contribution = models.IntegerField()

    individual_share = models.IntegerField()

    def set_payoffs(self):
        players = self.get_players()
        contributions = [p.contributions for p in players]

        self.total_contribution = sum(contributions)

        self.individual_share = (self.total_contribution * Constants.multiplier)/Constants.players_per_group

        for p in players:
            p.payoff = Constants.endowment - p.contribution + self.individual_share



class Player(BasePlayer):
    contribution = models.IntegerField(min = 0, max = Constants.endowment, label = "How much will you contribute?")