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


author = 'Patrick Sewell & Amil Camilo (Adapted from Felix Holzmeister)'

doc = """
Staircase time elicitation task as proposed by Falk et al. (2016), Working Paper.
"""

# def record_time():
#     import time
#     curr_time = time()
#     return curr_time

# ******************************************************************************************************************** #
# *** CLASS SUBSESSION
# ******************************************************************************************************************** #
class Subsession(BaseSubsession):

    # initiate list of sure payoffs and implied switching row in first round
    # ------------------------------------------------------------------------------------------------------------
    def creating_session(self):
        if self.round_number == 1:
            for p in self.get_players():
                p.participant.vars['time_payoffs'] = [Constants.first_number]
                p.participant.vars['icl_switching_row'] = 2 ** Constants.num_choices
                p.participant.vars['patience_result'] = 0


# ******************************************************************************************************************** #
# *** CLASS GROUP
# ******************************************************************************************************************** #
class Group(BaseGroup):
    pass


# ******************************************************************************************************************** #
# *** CLASS PLAYER
# ******************************************************************************************************************** #
class Player(BasePlayer):

    # add model fields to class player
    # ----------------------------------------------------------------------------------------------------------------
    random_draw = models.IntegerField()
    payoff_relevant = models.StringField()
    time_payoff = models.IntegerField()
    choice = models.StringField()
    switching_row = models.IntegerField()
    last_number = models.IntegerField()
    patience_result = models.IntegerField()
    decision_time = models.IntegerField()


    def record_decision_time(self):
        self.decision_time = int(time.time()) - self.participant.vars['start_time'] # records DT
        self.participant.vars['start_time'] = int(time.time()) # updates start time

    # set sure payoff for next choice for TIME ELICITATION
    # ----------------------------------------------------------------------------------------------------------------
    def set_sure_payoffs(self):

        # add current round's sure payoff to model field
        # ------------------------------------------------------------------------------------------------------------
        self.last_number = int(self.participant.vars['time_payoffs'][self.round_number - 1])

        # determine sure payoff for next choice and append list of sure payoffs
        # ------------------------------------------------------------------------------------------------------------
        if not self.round_number == Constants.num_choices:

            if self.choice == 'A':
                self.last_number = Constants.time_matrix[self.last_number][0]
                self.participant.vars['time_payoffs'].append(
                    int(self.last_number))
                #     c(self.participant.vars['time_sure_payoffs'][self.round_number - 1]
                #     + Constants.delta / 2 ** (self.round_number - 1)) #this is changed
                # )
            elif self.choice == 'B':
                self.last_number = Constants.time_matrix[self.last_number][1]
                self.participant.vars['time_payoffs'].append(
                    int(self.last_number))
                #     c(self.participant.vars['icl_sure_payoffs'][self.round_number - 1]
                #     - Constants.delta / 2 ** (self.round_number - 1))
                # )
            else:
                pass
        self.time_payoff = self.last_number
        # self.decision_time = int(time.time()) - self.participant.vars['start_time'] #records decision time
        # self.participant.vars['start_time'] = int(time.time()) #updates the start_time

    # update implied switching row each round
    # ----------------------------------------------------------------------------------------------------------------
    def update_switching_row(self):

        if self.choice == 'B':
            self.participant.vars['icl_switching_row'] -= 2 ** (Constants.num_choices - self.round_number)

        elif self.choice == 'I':
            self.participant.vars['icl_switching_row'] /= 2

    # set payoffs
    # ----------------------------------------------------------------------------------------------------------------
    def set_payoffs(self):

        current_round = self.subsession.round_number
        current_choice = self.in_round(current_round).choice

    def set_patience_result(self):
        if self.choice == 'A':
            self.participant.vars['patience_result'] = Constants.time_matrix[self.last_number][0]
            self.patience_result = self.participant.vars['patience_result']

        elif self.choice == 'B':
            self.participant.vars['patience_result'] = Constants.time_matrix[self.last_number][1]
            self.patience_result = self.participant.vars['patience_result']

        else:
            pass

        # if care about payoffs:
        # else:

        #
        # # set payoff if all choices have been completed or if "indifferent" was chosen
        # # ------------------------------------------------------------------------------------------------------------
        # if current_round == Constants.num_rounds or current_choice == 'I':
        #
        #     # randomly determine which choice to pay
        #     # --------------------------------------------------------------------------------------------------------
        #     completed_choices = len(self.participant.vars['time_payoffs'])
        #     self.participant.vars['icl_choice_to_pay'] = random.randint(1, completed_choices)
        #     choice_to_pay = self.participant.vars['icl_choice_to_pay']
        #
        #     # random draw to determine whether to pay the "high" or "low" lottery outcome
        #     # --------------------------------------------------------------------------------------------------------
        #     self.in_round(choice_to_pay).random_draw = random.randint(1, 100)
        #
        #     # determine whether the lottery or sure payoff is relevant for payment
        #     # --------------------------------------------------------------------------------------------------------
        #     self.in_round(choice_to_pay).payoff_relevant = random.choice(['A','B']) \
        #         if self.in_round(choice_to_pay).choice == 'I' \
        #         else self.in_round(choice_to_pay).choice
        #
        #     # set player's payoff
        #     # --------------------------------------------------------------------------------------------------------
        #     if self.in_round(choice_to_pay).payoff_relevant == 'A':
        #         self.in_round(choice_to_pay).payoff = Constants.lottery_hi \
        #             if self.in_round(choice_to_pay).random_draw <= Constants.probability \
        #             else Constants.lottery_lo
        #     elif self.in_round(choice_to_pay).payoff_relevant == 'B':
        #         self.in_round(choice_to_pay).payoff = self.participant.vars['time_payoffs'][choice_to_pay - 1]
        #
        #     # set payoff as global variable
        #     # --------------------------------------------------------------------------------------------------------
        #     self.participant.vars['icl_payoff'] = self.in_round(choice_to_pay).payoff
        #
        #     # implied switching row
        #     # --------------------------------------------------------------------------------------------------------
        #     self.in_round(choice_to_pay).switching_row = self.participant.vars['icl_switching_row']