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


doc = """
Your app description
"""


class Constants(BaseConstants):
    name_in_url = 'my_pub_goods'
    players_per_group = 2
    num_rounds = 2
    MPCR1 = 0.4
    MPCR2 = 0.6
    endowment=100


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    MPCR = models.FloatField()
# made an mpcr which will be a floating number


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


# FUNCTIONS
def creating_session(subsession):
    print('in creating session')

    # Establish total earnings variable
    #this is a loop
    for p in subsession.get_players():
        if subsession.round_number == 1:
            p.paticipant.vars['totalEarnings'] = 0

    # Assign varying MPCR
    for g in subsession.get_groups():
        print('round', subsession.round_number)
        print('num_rounds/2', int(Constants.num_rounds/2))
        if subsession.round_number <= int(Constants.num_rounds/2): # in first half
            g.MPCR = Constants.MPCR1
        else: # second half
            g.MPCR = Constants.MPCR2


def setPayoffs(g: Group):
    # First: total contribution by all members of the group
    total_contribution = 0
    for p in g.get_players():
        total_contribution += p.contribution

    # Second: calculate the earnings for each group member
    for p in g.get_players():
        print('payoff', p.participant.payoff)
        print('endowment', Constants.endowment)
        print('contribution', p.contribution)
        print('total contribution', total_contribution)
        print('MPCR', g.MPCR)
        p.participant.payoff = float(Constants.endowment
                                     - p.contribution
                                     + total_contribution)
        p.participant.vars['totalEarnings'] += p.participant.payoff


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


class ResultsWaitPage(WaitPage):
    after_all_players_arrive = 'setPayoffs'
    body_text = "Please wait for everyone to make their decisions."


class Results(Page):
    pass # when there isn't anything to do logically


page_sequence = [Action, ResultsWaitPage, Results]