from otree.api import *


author = 'Your name here'
doc = """
Your app description
"""


class C(BaseConstants):
    NAME_IN_URL = 'common_commitment'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1
    INVESTMENT_FACTOR = 2
    BUDGET = 10


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    total_contribution = models.FloatField()
    total_profit = models.FloatField()


class Player(BasePlayer):
    contribution = models.FloatField(
        min=0, max=C.BUDGET, label="How much do you want to invest?"
    )


# FUNCTIONS
# PAGES
class Contribution(Page):
    form_model = "player"
    form_fields = ["contribution"]



class ResultsWaitPage(WaitPage):
    @staticmethod
    def after_all_players_arrive(group: Group):
        total_contribution = 0
        for player in group.get_players():
            total_contribution += player.contribution
        group.total_contribution = total_contribution
        group.total_profit = total_contribution * C.INVESTMENT_FACTOR


class Results(Page):

    @staticmethod
    def vars_for_template(player: Player):

        list_of_contributions = [player.contribution]
        for p in player.get_others_in_group():
            list_of_contributions.append(p.contribution)
        return {
            "list_of_contributions": list_of_contributions,
            "player_payoff":  ((C.BUDGET - min(list_of_contributions))+min(list_of_contributions)*C.INVESTMENT_FACTOR),
            "minimum_amount": min(list_of_contributions)
        }

page_sequence = [Contribution, ResultsWaitPage, Results]