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


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


class Test(Page):
    form_model = "player"
    form_fields = ["test_1", "test_2", "test_3",
                   "test_4", "test_5", "test_6", "test_7"]

    def is_displayed(self):
        return self.round_number == 1

    def test_1_error_message(self, test_1):
        if test_1 != True:
            return '* 実験1は二人一組で行います。'

    def test_2_error_message(self, test_2):
        if test_2 != True:
            return '* この実験は実験1を10ラウンド、実験２を10ラウンドの合計20ラウンド行います。'

    def test_3_error_message(self, test_3):
        if test_3 != True:
            return '* ペアはラウンドごとに変わります。'

    def test_4_error_message(self, test_4):
        if test_4 != True:
            return '* 販売単価を求める式は' \
                   '121－あなたの生産量－相手の生産量－1です。'

    def test_5_error_message(self, test_5):
        if test_5 != True:
            return '* 販売単価は101です。'

    def test_6_error_message(self, test_6):
        if test_6 != True:
            return '* あなたの利得は1800です。'

    def test_7_error_message(self, test_7):
        if test_7 != True:
            return '* あなたの利得は1125です。'

class Decide(Page):
    form_model = 'player'
    form_fields = ['units']


class ResultsWaitPage(WaitPage):
    body_text = "Waiting for the other participant to decide."

    after_all_players_arrive = 'set_payoffs'


class Results(Page):
    def vars_for_template(self):
        return dict(other_player_units=self.player.other_player().units)


page_sequence = [Introduction, Test, Decide, ResultsWaitPage, Results]