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

doc = '\nThis is a sequential PG game'
class Constants(BaseConstants):
    name_in_url = 'Treatment1'
    players_per_group = 6
    num_rounds = 12
    endowment = c(100)
    multiplier = 2.4
    part_fee = 2
    my_curr = '$'
    admin_report_template = 'Treatment1/admin_report.html'
    instructions_template = 'Treatment1/instructions.html'
class Subsession(BaseSubsession):
    def vars_for_admin_report(self):
        contributions = [
            p.contribution for p in self.get_players() if p.contribution != None
        ]
        if contributions:
            return dict(
                avg_contribution=sum(contributions) / len(contributions),
                min_contribution=min(contributions),
                max_contribution=max(contributions),
            )
        else:
            return dict(
                avg_contribution='(no data)',
                min_contribution='(no data)',
                max_contribution='(no data)',
            )
    def creating_session(self):
        import random
        if self.round_number == 1:
            paying_round = random.randint(3, Constants.num_rounds)
            self.session.vars['paying_round'] = paying_round


class Group(BaseGroup):
    total_contribution = models.CurrencyField()
    individual_share = models.CurrencyField()
    def set_payoffs(self):
        self.total_contribution = sum([p.contribution for p in self.get_players()])
        self.individual_share = (
            self.total_contribution * Constants.multiplier / Constants.players_per_group
        )
        for p in self.get_players():
            if self.subsession.round_number == self.session.vars['paying_round']:
                        p.payoff = (Constants.endowment - p.contribution) + self.individual_share
    def position1(self):
        return self.get_players()[0]
    def position2(self):
        return self.get_players()[1]
    def position3(self):
        return self.get_players()[2]
    def position4(self):
        return self.get_players()[2]
    def position5(self):
        return self.get_players()[4]
    def position6(self):
        return self.get_players()[5]
class Player(BasePlayer):
    contribution = models.CurrencyField(choices=[[100, 'Contribute 100 points'], [0, 'Contribute 0 point']])