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


class Introduction1S(Page):
    def is_displayed(self):
        return self.round_number == 1


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

    form_model = models.Group
    form_fields = ['offer']

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

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

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

    form_model = models.Group
    form_fields = ['actual_accept']


class ResultsWaitPage(WaitPage):

    def after_all_players_in_group_arrive(self):
        self.group.set_payoffs()


class Results(Page):
    def vars_for_template(self):
        return {
            'total_payoff': sum([p.payoff
                                 for p in self.player.in_all_rounds()]),
            'paying_round': self.session.vars['paying_round'],
            'player_in_all_rounds': self.player.in_all_rounds(),
        }


page_sequence = [
    Introduction1S,
    Offer,
    GetStrategy,
    WaitPage,
    AcceptOffer,
    ResultsWaitPage,
    Results
]