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

from certainty_eq.config import *
import random as rrandom
from random import *
from statistics import median

author = 'JUBO YAN--NTU Econ'

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


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

    def creating_session(self):
        self.chosen_rnd_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
                lottery_hi = Constants.lottery_hi
                lottery_lo = Constants.lottery_lo
                endowments = Constants.endowment

                p.participant.vars['ce_environment'] = list(zip(round_list, endowments, lottery_hi, lottery_lo,
                                                                probabilities))
                p.participant.vars['ce_outcomes'] = [None for j in range(0, Constants.num_rounds)]
                # p.participant.vars['ce'] = [None for j in range(0, Constants.num_rounds)]
                # p.participant.vars['ce_payoff'] = [None for j in range(0, Constants.num_rounds)]


class Group(BaseGroup):

    pass


class Player(BasePlayer):

    ce = models.CurrencyField(
        doc="""The amount you choose in exchange of the lottery""",
    )

    rnd_draw = models.FloatField()
    rnd_bdm = models.FloatField()
    payoff = models.FloatField()
    winner = models.IntegerField()

    def set_participant_investment(self, round_num):
        # self.random_round = randrange(1, Constants.num_rounds + 1)
        # if the randomly drawn number is smaller than the specified probability, then win the lottery
        self.rnd_draw = random()
        # determine the BDM draw. lower bound and upper bound are the
        self.rnd_bdm = rrandom.uniform(Constants.lottery_lo[round_num - 1], Constants.lottery_hi[round_num - 1])

        if self.rnd_draw < self.participant.vars['ce_environment'][round_num - 1][4]:
            self.winner = 1
        else:
            self.winner = 0

        if self.rnd_bdm > self.ce:
            self.payoff = self.rnd_bdm
        else:
            if self.winner == 1:
                self.payoff = Constants.lottery_hi[round_num - 1]
            else:
                self.payoff = Constants.lottery_lo[round_num - 1]

        self.participant.vars['ce_outcomes'][round_num - 1] = [self.ce, self.rnd_draw, self.rnd_bdm, self.payoff,
                                                               self.winner, self.payoff + Constants.endowment[round_num - 1]]