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


class Welcome(Page):
    """Beschreibung des Experiments"""

    timeout_seconds = 14*60



    def before_next_page(self):
        if self.timeout_happened:
            self.participant.vars['is_dropout'] = True



class Introduction(Page):

    def get_timeout_seconds(self):
        if self.participant.vars.get('is_dropout'):
            return 1  # instant timeout, 1 second
        else:
            return 15 * 60

    def before_next_page(self):
        if self.timeout_happened:
            self.participant.vars['is_dropout'] = True


class Survey1(Page):
    form_model = 'player'
    form_fields = ['f1', 'f2', 'f3', 'f4', 'f5', 'f6']

    def is_displayed(self):
        return self.session.config['color'] == 'red' or self.session.config['color'] == 'blue'

    def get_timeout_seconds(self):
        if self.participant.vars.get('is_dropout'):
            return 1  # instant timeout, 1 second
        else:
            return 7 * 60

    def before_next_page(self):
        if self.timeout_happened:
            self.participant.vars['is_dropout'] = True


class Survey1Club(Page):
    form_model = 'player'
    form_fields = ['f1', 'f2', 'f3', 'f4', 'fc5', 'fc6']

    def is_displayed(self):
        return self.session.config['color'] == 'green'

    def get_timeout_seconds(self):
        if self.participant.vars.get('is_dropout'):
            return 1  # instant timeout, 1 second
        else:
            return 7 * 60

    def before_next_page(self):
        if self.timeout_happened:
            self.participant.vars['is_dropout'] = True


class Survey1Results(Page):

    def is_displayed(self):
        return self.session.config['color'] == 'red' or self.session.config['color'] == 'blue'

    def get_timeout_seconds(self):
        if self.participant.vars.get('is_dropout'):
            return 1  # instant timeout, 1 second
        else:
            return 5 * 60

    def before_next_page(self):
        if self.timeout_happened:
            self.participant.vars['is_dropout'] = True


class Survey1ResultsClub(Page):

    def is_displayed(self):
        return self.session.config['color'] == 'green'

    def get_timeout_seconds(self):
        if self.participant.vars.get('is_dropout'):
            return 1  # instant timeout, 1 second
        else:
            return 5 * 60

    def before_next_page(self):
        if self.timeout_happened:
            self.participant.vars['is_dropout'] = True

page_sequence = [Welcome,
                 Introduction,
                 Survey1,
                 Survey1Club,
                 Survey1Results,
                 Survey1ResultsClub
                 ]