from . import models
from ._builtin import Page, WaitPage
from .functions import slider
from otree.api import Currency as c, currency_range
from .models import Constants

class RoleInstraction(Page):
    pass

class Instruction(Page):
    pass

class DecessorInstructions(Page):
    def before_next_page(self):
        self.player.set_payoff()

    def is_displayed(self):
        return self.player.id_in_group == 1

class SliderPrimaryDiscrete(Page):
    def before_next_page(self):
        self.player.set_payoff()

    def is_displayed(self):
        return self.player.id_in_group == 1

    form_model = models.Player
    form_fields = ['slider1',
                   'slider2',
                   'slider3',
                   'slider4',
                   'slider5',
                   'slider6',
                ]

    def vars_for_template(self):
        return {'slider_items': [1, 2, 3, 4, 5, 6]}

    def before_next_page(self):
        chosen_values = {
            'item1': self.player.slider1,
            'item2': self.player.slider2,
            'item3': self.player.slider3,
            'item4': self.player.slider4,
            'item5': self.player.slider5,
            'item6': self.player.slider6,
        }
        mean_allocations = slider.mean_allocations_discrete(chosen_values)
        svo_slider_angle = slider.svo_angle(mean_allocations['self'], mean_allocations['other'])
        self.player.slider_angle = svo_slider_angle
        self.player.slider_classification = slider.svo_classification(svo_slider_angle)

class ReceiverInstructions(Page):
    def is_displayed(self):
        return self.player.id_in_group == 2

class DebugDisplayPage(Page):
    def vars_for_template(self):
        return {'angle': self.player.slider_angle}


class ResultsWaitPage(WaitPage):
    # wait_for_all_groups = True
    after_all_players_arrive = 'set_payoffs'
    template_name = 'SSM/ResultsWaitPage.html'


class Results(Page):
    """Players payoff: How much each has earned"""

page_sequence = [
    RoleInstraction,
    Instruction,
    SliderPrimaryDiscrete,
    ReceiverInstructions,
    ResultsWaitPage,
]