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

doc = ''
class Constants(BaseConstants):
    name_in_url = 'instructions'
    players_per_group = None
    num_rounds = 1
    initial_mean_energy_level = c(105)
    ll_rounds = 8
    ul_rounds = 16
    ll_initial_energy_level = c(96)
    ul_initial_energy_level = c(120)
class Subsession(BaseSubsession):
    pass
class Group(BaseGroup):
    cooperation_payback = models.IntegerField()
    steal_failure_payoff_absolute = models.IntegerField()
    def group_formation(self):
        import random
        import numpy as np
        self.cooperation_payback = self.session.config['cooperation_payoff'] + self.session.config['victims_payoff']
        self.steal_failure_payoff_absolute = np.abs(self.session.config["steal_failure_payoff"])
        self.session.vars['nrounds'] = random.uniform (Constants.ll_rounds, Constants.ul_rounds)
        self.session.vars['mocks_done'] = False
        n = self.session.config['num_demo_participants']
        self.session.vars['num_partners'] = n - 1
        starting = np.random.normal(loc=self.session.config['initial_mean_energy_level'],scale=self.session.config['initial_std_energy_level'],size=n).round().astype('int').clip(100,200)
        for i in range(n):
            p = self.get_player_by_id(i+1)
            p.participant.payoff = int(starting[i])
            p.participant.vars['mocks_done'] = False
            p.participant.vars['rounds_below_threshold'] = 0

class Player(BasePlayer):
    test1 = models.StringField(label='If you work alone, then how much can you expect your energy points level to change that round?')
    def test1_choices(self):
        return [[f'{self.session.config["lone_payoff"]-5} points', f'{self.session.config["lone_payoff"]-5} points'], [f'{self.session.config["lone_payoff"]} points', f'{self.session.config["lone_payoff"]} points'], [f'{self.session.config["lone_payoff"]+5} points', f'{self.session.config["lone_payoff"]+5} points']]
    test2 = models.StringField(label='Say you have 100 points, you and one other person in the group decide to cooperate, and no-one decides to steal, how many points would you have afterwards?')
    def test2_choices(self):
        return [[f'{100 + self.session.config["cooperation_payoff"]-5} points', f'{100 + self.session.config["cooperation_payoff"]-5} points'], [f'{100 + self.session.config["cooperation_payoff"]} points', f'{100 + self.session.config["cooperation_payoff"]} points'], [f'{100 + self.session.config["cooperation_payoff"]+5} points', f'{100 + self.session.config["cooperation_payoff"]+5} points']]
    test3 = models.StringField(label='Say your points level is 110, you decide to steal, everyone else in the group decides to cooperate, and you get away with it. What do you expected your energy points level to be after the round?')
    def test3_choices(self):
        return [[f'{110+self.session.config["steal_success_payoff"]-5} points', f'{110+self.session.config["steal_success_payoff"]-5} points'], [f'{110+self.session.config["steal_success_payoff"]-10} points', f'{110+self.session.config["steal_success_payoff"]-10} points'], [f'{110+self.session.config["steal_success_payoff"]} points', f'{110+self.session.config["steal_success_payoff"]} points']]
    consent = models.StringField(choices=[['Yes', 'Yes']], label='I consent to take part in this study. ', widget=widgets.RadioSelect)
    prolific_id = models.StringField(label='Please re-enter your Prolific ID')