from otree.api import Page, WaitPage
from .models import C


P_FAIR = .5
P_BIASED = 1

class Introduction(Page):
    """
    Introduction page providing some information about the game.
    """
    def is_displayed(self):
        return self.round_number == 1

class RoundInfo(Page):
    """
    Page displaying round information
    """
    def vars_for_template(self):
        return {
            'round_number': self.round_number
        }
    def is_displayed(self):
        return self.round_number <= C.NUM_ROUNDS


class ChooseCoin(Page):
    """
    Page where participant chooses between 'fair' and 'biased' coin.
    """
    form_model = 'player'
    form_fields = ['coin_choice']

    def before_next_page(self):
        # Flip the chosen coin after the player makes a choice.
        self.player.flip_chosen_coin(p_fair=P_FAIR, p_biased=P_BIASED)

    def is_displayed(self):
        return self.round_number <= C.NUM_ROUNDS


class RevealCoinOutcome(Page):
    """
    Page revealing the outcome of the chosen coin.
    """
    def vars_for_template(self):
        return {
            'chosen_coin_result': self.player.chosen_coin_result
        }
    
    def is_displayed(self):
        return self.round_number <= C.NUM_ROUNDS

class ChoosePermutation(Page):
    """
    Page where participant chooses between four options.
    """
    form_model = 'player'
    form_fields = ['coin_permutation_choice']

    def before_next_page(self):
        # Flip the chosen coin after the player makes a choice.
        self.player.flip_chosen_coin(p_fair=P_FAIR, p_biased=P_BIASED)

    def is_displayed(self):
        return self.round_number <= C.NUM_ROUNDS


class Results(Page):
    """
    Show overall results or feedback.
    """
    def vars_for_template(self):
        # Sum over all the winnings from each player (i.e. each round)
        return {
            'winnings': sum([player.total_winnings for player in self.player.in_all_rounds()])
        }
    def is_displayed(self):
        return self.round_number == C.NUM_ROUNDS

page_sequence = [Introduction, RoundInfo, ChooseCoin, RevealCoinOutcome, ChoosePermutation, Results]