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

from otree.widgets import Slider



class Constants(BaseConstants):
    name_in_url = 'split_dot_fraction'
    players_per_group = None

    dot_conditions = 250
    numbers_probs = [5, 10, 25, 50, 75, 90, 95, 100]

    succession_rounds = [2, 2, 2, 2]
    num_rounds = 2

class Subsession(BaseSubsession):

    def before_session_starts(self):
        if self.round_number == 1:
            for p in self.get_players():
                # c = random.sample(Constants.dot_conditions, 1)
                if 'dot_block' in p.participant.vars:
                    p.participant.vars['dot_block'] += 1
                else:
                    p.participant.vars['dot_block'] = 1
                    p.participant.vars['dots_paying_round'] = random.randint(1, sum(Constants.succession_rounds))
                    list_of_tasks = [(x, Constants.dot_conditions) for x in Constants.numbers_probs]
                    p.participant.vars['dot_sequence'] = random.sample(list_of_tasks, len(list_of_tasks))
        for p in self.get_players():
            if self.round_number <= Constants.succession_rounds[p.participant.vars['dot_block']-1]:
                p.set_task()
                if p.round_number + sum(Constants.succession_rounds[:(p.participant.vars['dot_block']-1)]) == p.participant.vars['dots_paying_round']:
                    p.on_paying_round = True

class Group(BaseGroup):
    pass


class Player(BasePlayer):

    guess = models.FloatField()
    task_identifier_percent = models.IntegerField()
    task_identifier_cloud = models.IntegerField()
    on_paying_round = models.BooleanField(initial=False)
    dot_block = models.IntegerField()

    def set_task(self):
        rounds_previous_blocks = sum(Constants.succession_rounds[:(self.participant.vars['dot_block']-1)])
        task_tuple = self.participant.vars['dot_sequence'][self.round_number + rounds_previous_blocks - 1]
        setattr(self, 'task_identifier_percent', task_tuple[0])
        setattr(self, 'task_identifier_cloud', task_tuple[1])
        setattr(self, 'dot_block', self.participant.vars['dot_block'])

    def set_payoffs(self):
        payoff = 0
        correct_guess = self.task_identifier_cloud * self.task_identifier_percent/100
        if abs(self.guess - correct_guess) <= 5:
            payoff = self.session.config['incentive_dots']

        if 'possible_payoffs' in self.participant.vars:
            self.participant.vars['possible_payoffs'].append(payoff)
        else:
            self.participant.vars['possible_payoffs'] = [payoff]