from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c, currency_range
)
from gneezy_potters.config import *
import random
from random import randrange
from random import random

author = 'Your name here'

doc = """
Gneezy Potters method to elicit risk preference as in their 1995 QJE paper.
"""

class Constants(BaseConstants):
    name_in_url = 'gneezy_potters'
    # group size
    players_per_group = None
    # endowment amount for each different round
    endowment = (20, 20, 20, 20)
    # probability of success
    probability = (1/2, 1/2, 1/2, 1/2)
    # multiplier = return rate + 1
    multiplier = (1.7, 1.35, 1.5, 1.35)
    # multiplier2 = return rate2 + 1
    multipliersecond = (0.7, 0.75, 0.75, 0.7)
    # the number of total rounds
    num_rounds = len(probability)


class Subsession(BaseSubsession):
    rand_round = models.IntegerField()

    def creating_session(self):
        self.rand_round = randrange(1, Constants.num_rounds + 1)
        if self.round_number == 1:
            for p in self.get_players():

                round_list = (j for j in range(1, Constants.num_rounds+1))
                probabilities = Constants.probability
                multipliers = Constants.multiplier
                multipliersecond = Constants.multipliersecond
                endowments = Constants.endowment
                p.participant.vars['environment'] = list(zip(round_list, endowments, probabilities, multipliers,
                                                             multipliersecond))
                p.participant.vars['environment2'] = list(zip(round_list, endowments, probabilities, multipliersecond))
                p.participant.vars['outcomes'] = [None for j in range(0, Constants.num_rounds)]
                p.participant.vars['investment'] = [None for j in range(0, Constants.num_rounds)]
                p.participant.vars['payoff'] = [None for j in range(0, Constants.num_rounds)]


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    random_draw = models.FloatField()
    winner = models.IntegerField()
    investment = models.CurrencyField(
        min=0, max=Constants.endowment[0],
        doc="""The amount invested by the subject""",
    )

    def set_payoffs(self, round_num):
        self.random_draw = random()
        if self.random_draw < self.participant.vars['environment'][round_num-1][2]:
            self.payoff = self.participant.vars['environment'][round_num - 1][1] - self.investment + \
                          self.investment * self.participant.vars['environment'][round_num - 1][3]
            self.winner = 1
        else:
            self.payoff = self.participant.vars['environment'][round_num - 1][1] - self.investment + \
                          self.investment * self.participant.vars['environment'][round_num - 1][4]
            self.winner = 0

        self.participant.vars['investment'][round_num - 1] = self.investment
        self.participant.vars['payoff'][round_num - 1] = self.payoff
        self.participant.payoff = self.payoff