from otree.api import *


doc = """
Your app description
"""


class C(BaseConstants):
    NAME_IN_URL = 'socialdilemma'
    PLAYERS_PER_GROUP = 5
    NUM_ROUNDS = 10
    ENDOWMENT = 1000
    MULTIPLIER = 3


class Subsession(BaseSubsession):
    pass


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

    def set_payoffs(self):
        players = self.get_players()
        contributions = [p.contribution for p in players]
        self.total_contribution = sum(contributions)
        self.individual_share = self.total_contribution * C.MULTIPLIER / C.PLAYERS_PER_GROUP
        for player in players:
            player.payoff = C.ENDOWMENT - player.contribution + self.individual_share


class Player(BasePlayer):
    contribution = models.IntegerField(
        min=0,
        max=C.ENDOWMENT,
        label="どれほど公共財に投資しますか？"
    )

# PAGES
class MyPage(Page):
    form_model = 'player'
    form_fields = ['contribution']


class ResultsWaitPage(WaitPage):
    def after_all_players_arrive(self):
        self.group.set_payoffs()


class Results(Page):
    pass


page_sequence = [MyPage, ResultsWaitPage, Results]