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


class IntroWaitPagePhase1(WaitPage):
    """WaitPage to continue when everyone is ready for the next phase"""
    def is_displayed(self):
        return self.round_number in Constants.phase1

    def after_all_players_arrive(self):
        self.group.set_distribution_rule()


class IntroWaitPagePhase2(WaitPage):
    """WaitPage to continue when everyone is ready for the next phase"""
    def is_displayed(self):
        return self.round_number in Constants.phase2

    def after_all_players_arrive(self):
        self.group.set_distribution_rule()


class IntroWaitPagePhase3(WaitPage):
    def is_displayed(self):
        return self.round_number in Constants.phase3

    def after_all_players_arrive(self):
        return


class Phase1Intro(Page):
    """Introduction page for phase 1"""
    def is_displayed(self):
        return self.round_number == min(Constants.phase1)

    timeout_seconds = 15


class Phase2Intro(Page):
    """Introduction page for phase 2"""
    def is_displayed(self):
        return self.round_number == min(Constants.phase2)

    timeout_seconds = 15


class Phase3Intro(Page):
    """Introduction page for phase 3"""
    def is_displayed(self):
        return self.round_number == min(Constants.phase3)

    timeout_seconds = 15


class DistributionRuleVote(Page):
    """The Page on which players may choose for a distribution rule in Phase 3."""
    def is_displayed(self):
        return self.round_number == min(Constants.phase3)

    form_model = models.Player
    form_fields = ['prule']


class DistributionWaitPage(WaitPage):
    """"WaitPage to calculate the distribution rule in Phase 3"""
    def is_displayed(self):
        return self.round_number in Constants.phase3

    def after_all_players_arrive(self):
        self.group.set_distribution_rule()


class DistributionRuleAnnounce(Page):
    timeout_seconds = 15


class Threshold(Page):
    """Players may propose a threshold level for the current round."""
    form_model = models.Player
    form_fields = ['proposal']


class ThresholdWaitPage(WaitPage):

    """"WaitPage to calculate the the threshold level."""

    def after_all_players_arrive(self):
        self.group.set_threshold()

    body_text = "Gelieve te wachten tot elke speler zijn voorstel ingeleverd heeft."


class Contribute(Page):
    """Players have to decide how much points to contribute."""
    form_model = models.Player
    form_fields = ['contribution']

    def set_threshold(self):
        return


class ResultsWaitPage(WaitPage):
    """"WaitPage to calculate the the payoffs."""
    def after_all_players_arrive(self):
        self.group.set_payoffs()

    body_text = "Gelieve te wachten tot elke speler zijn bijdrage heeft gekozen."


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

    def set_payoff(self):
        return

    def vars_for_template(self):
        return {
            'total_earnings': self.group.total_contribution * Constants.efficiency_factor,
        }


class EarningsWaitPage(WaitPage):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    def after_all_players_arrive(self):
        for p in self.group.get_players():
            p.set_payoff()

    body_text = "Gelieve te wachten tot elke speler het experiment afgerond heeft."


class Earnings(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds

    def set_payoff(self):
        return

page_sequence = [
    IntroWaitPagePhase1,
    IntroWaitPagePhase2,
    IntroWaitPagePhase3,
    Phase1Intro,
    Phase2Intro,
    Phase3Intro,
    DistributionRuleVote,
    DistributionWaitPage,
    DistributionRuleAnnounce,
    Threshold,
    ThresholdWaitPage,
    Contribute,
    ResultsWaitPage,
    Results,
    EarningsWaitPage,
    Earnings
]