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




class aIntroduction(Page):
    def is_displayed(self):
        return self.subsession.round_number == 1
    pass

class aContribute(Page):
    form_model = models.Player
    form_fields = ['contribution']


class bPleaseproceed(Page):
    timeout_seconds = 20


class SimpleWaitPage(WaitPage):
    def vars_for_template(self):
        self.group.set_payoffs()

    body_text = "Waiting for other group members."

    pass

class cResults(Page):
    def vars_for_template(self):
        return self.group.set_payoffs()


class Comprehension1(Page):

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

    def error_message(self, values):
        if values["check1"] != '90':
            self.player.s2q1attempt += 1
            if self.player.s2q1attempt >= 3:
                return ("You have reached the maximum number of attempts, please raise your hand now.")
            return 'WRONG! Please try again. Raise your hand if you need help.'

    pass

class Comprehension2(Page):

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

    def error_message(self, values):
        if values["check2"] != 'Group Enforcer':
            self.player.s2q2attempt += 1
            if self.player.s2q2attempt >= 3:
                return ("You have reached the maximum number of attempts, please raise your hand now.")
            return 'WRONG! Please try again. Raise your hand if you need help.'

    pass


class Comprehension3(Page):

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

    def error_message(self, values):
        if values["check3"] != 'Removes points from the Group Members using Enforcement Tax Points':
            self.player.s2q3attempt += 1
            if self.player.s2q3attempt >= 3:
                return ("You have reached the maximum number of attempts, please raise your hand now.")
            return 'WRONG! Please try again. Raise your hand if you need help.'

    pass

page_sequence = [
    aIntroduction,
    Comprehension1,
    Comprehension2,
    Comprehension3,
    aContribute,
    bPleaseproceed,
    SimpleWaitPage,
    cResults,
]