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




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

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


class OfferWaitPage(WaitPage):
    def vars_for_template(self):
        if self.player.role() == 'agent':
            body_text = "You are Participant 2. Waiting for Participant 1 to propose a wage offer."
        else:
            body_text = "Waiting for Participant 2."
        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']

    timeout_submission = {
#        'contract_accepted': False,
        'agent_work_effort': 0,
    }

class WorkEffort(Page):
    def is_displayed(self):
        #if self.group.contract_accepted == True:
        return self.player.role() == 'agent'

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

 #   timeout_submission = {
#        'contract_accepted': False,
#        'agent_work_effort': 0,
    }




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

    def after_all_players_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()]),
            'player_in_all_rounds': self.player.in_all_rounds()
        }




page_sequence = [
                 Offer,
                 OfferWaitPage,
                 #Accept,
                 WorkEffort,
                 ResultsWaitPage,
                 Results]