from otree.api import *


doc = """
Public good experiment according to oTree tutorial
"""


class C(BaseConstants):
    NAME_IN_URL = 'first_public_good'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1
    ENDOWMENT = cu(1000)
    MULTIPLIER = 2




class Subsession(BaseSubsession):
    pass

class Player(BasePlayer):
    name = models.StringField()
    age = models.IntegerField()
    contribution = models.CurrencyField(
        min=0,
        max=C.ENDOWMENT,
        label="How much will you contribute?"
    )

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

#functions
def set_payoffs(group: Group):
    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
    )
    for p in players:
        p.payoff = C.ENDOWMENT - p.contribution + group.individual_share

#def set_payoffs(group: Group):
 #   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
    #for player in players:
     #   player.payoff = C.ENDOWMENT-Player.contribution+group.individual_share




# otree update_my_code to remove old syntax
# otree remove_self to remove syntax including self
# otree upcase_constants

# PAGES
class Survey(Page):
    form_model = 'player'
    form_fields = ['name', 'age']

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

class ResultsWaitPage(WaitPage):
    after_all_players_arrive ='set_payoffs'



class Results(Page):
    pass


page_sequence = [Survey, Contribute, ResultsWaitPage, Results]