from otree.api import *



class C(BaseConstants):
    NAME_IN_URL = 'public_goods_simple'
    PLAYERS_PER_GROUP = 3   #1グループ何人か
    NUM_ROUNDS = 3          #ラウンド数は何回か
    ENDOWMENT = cu(100)     #初期保有はいくらか
    MULTIPLIER = 1.8        #公共財の係数


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    total_contribution = models.CurrencyField() #投資合計額
    individual_share = models.CurrencyField()   #投資量からのリターン


class Player(BasePlayer):
    contribution = models.CurrencyField(        #投資量
        min=0, max=C.ENDOWMENT, label="投資額をいくらにしますか？"
    )


# FUNCTIONS
#def creating_session(subsession):   #毎ラウンド異なるマッチング
#    subsession.group_randomly()

#1ラウンド目だけランダムマッチング、その後は固定
def creating_session(subsession):
    if subsession.round_number == 1:
        subsession.group_randomly()
    else:
        subsession.group_like_round(1)


def set_payoffs(group: Group):
    
    #1. total_contribution（グループ内の投資額の合計）、individual_share（投資による利益分）を計算
    players = group.get_players()
    contributions = [p.contribution for p in players]
    group.total_contribution = sum(contributions)
    group.individual_share = (
        group.total_contribution * C.MULTIPLIER / C.PLAYERS_PER_GROUP
    )

    #2. グループ内で各々の利益を計算
    for p in players:
        p.payoff = C.ENDOWMENT - p.contribution + group.individual_share


# PAGES
class Introduction(Page):
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number==1

class Contribute(Page):
    form_model = 'player'
    form_fields = ['contribution']


class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs


class Results(Page):
    @staticmethod
    def vars_for_template(player):
        return dict(kept = C.ENDOWMENT - player.contribution)


page_sequence = [Introduction, Contribute, ResultsWaitPage, Results]