from otree.api import *
c = cu

doc = ''
class Constants(BaseConstants):
    name_in_url = 'QFpublic_goods_simple2'
    players_per_group = 4
    num_rounds = 10
    endowment = cu(1000)
    multiplier = 3
    heiho = 0.5
    nibai = 2
    subsidy = 3000
class Subsession(BaseSubsession):
    pass
def set_payoffs(group):
    players = group.get_players()
    contributions = [p.contribution ** Constants.heiho for p in players]
    group.total_contribution = sum(contributions)
    group.total_contribution2 = group.total_contribution ** Constants.nibai
    group.individual_share = ( 
       group.total_contribution2 / Constants.players_per_group
    )
    for p in players:
        p.payoff = Constants.endowment - p.contribution + group.individual_share
class Group(BaseGroup):
    total_contribution2 = models.CurrencyField()
    individual_share = models.CurrencyField()
    total_contribution = models.CurrencyField()
class Player(BasePlayer):
    contribution = models.CurrencyField(label='あなたは、どの位グループに貢献を行いますか？', max=Constants.endowment, min=0)
class Contribute(Page):
    form_model = 'player'
    form_fields = ['contribution']
class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs
    title_text = '他の参加者が入力し終わるまで、しばしお待ちください。'
class Results(Page):
    form_model = 'player'
class Ends(Page):
    form_model = 'player'
page_sequence = [Contribute, ResultsWaitPage, Results, Ends]