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


class Instruction(Page):
    pass

class Offer(Page):
    form_model = 'group'
    form_fields=['offer']

    def is_displayed(self):
        return self.player.proposer

class PlayerBWaitPage(WaitPage):
    def is_displayed(self):
        return self.player.proposer == 0

class Response(Page):
    form_model='group'
    form_fields=['accept']

    def is_displayed(self):
        return self.player.proposer == 0

class ResultsWaitPage(WaitPage):
    pass

class Results(Page):
    def vars_for_template(self):
        role='Player A' if self.player.proposer else 'Player B'
        accepted='Yes' if self.group.accept else 'No'
        return {
            'your_role':role,
            'offer_amount':self.group.offer,
            'offer_accepted':accepted,
            'payoff':self.player.final_payoff
        }


page_sequence = [Instruction, Offer, PlayerBWaitPage, Response, ResultsWaitPage, Results]