from . import models
from ._builtin import Page, WaitPage
from .models import Constants, cost_from_effort
from otree.common import safe_json


class Introduction(Page):
    pass


class Offer(Page):
    def is_displayed(self):
        return self.player.role() == 'principal'

    form_model = models.Group
    form_fields = ['agent_fixed_pay', 'agent_return_share']


class OfferWaitPage(WaitPage):
    def vars_for_template(self):
        if self.player.role() == 'agent':
            body_text = "You are Participant B. Waiting for Participant A to propose a contract."
        else:
            body_text = "Waiting for Participant B."
        return {'body_text': body_text}


class Accept(Page):
    def is_displayed(self):
        return self.player.role() == 'agent'

    form_model = models.Group
    form_fields = ['contract_accepted', 'agent_work_effort']

    timeout_submission = {
        'contract_accepted': False,
        'agent_work_effort': 1,
    }

    def vars_for_template(self):
        return {
            'EFFORT_TO_RETURN': safe_json(Constants.EFFORT_TO_RETURN),
            'EFFORT_TO_COST': safe_json(Constants.EFFORT_TO_COST),
        }


class ResultsWaitPage(WaitPage):
    def body_text(self):
        if self.player.role() == 'principal':
            return "Waiting for Participant B to respond."

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


class Results(Page):
    def vars_for_template(self):
        return {
            'fixed_pay_int': int(self.group.agent_fixed_pay),
            'received': self.player.payoff - Constants.bonus,
            'effort_cost': cost_from_effort(self.group.agent_work_effort),
        }


page_sequence = [Introduction,
                 Offer,
                 OfferWaitPage,
                 Accept,
                 ResultsWaitPage,
                 Results]