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


class P1Bienvenue(Page):
    form_model = 'player'
    form_fields = ['agree']

    def before_next_page(self):
        if self.player.id_in_group % 2 == 1:
            self.player.treatment = 1
        else:
            self.player.treatment = 2
        self.player.participant.vars['treatment'] = self.player.treatment

class Questionnaire0(Page):
    form_model = 'player'
    form_fields = ['nextq0', 'q0', 'b0']

    def before_next_page(self):
        self.player.participant.vars['nextq0'] = self.player.nextq0
        if self.player.nextq0 == 1 :
            self.player.eliminated = 1
            self.player.gaineliminated = 0
            self.player.participant.vars['eliminated'] = self.player.eliminated
            self.player.participant.vars['gaineliminated'] = self.player.gaineliminated

class Questionnaire0Bis(Page):
    timeout_seconds = 15
    form_model = 'player'
    form_fields = []
    def is_displayed(self):
        return self.player.nextq0 != 1

class P25QuestionnaireFinal(Page):
    form_model = 'player'
    form_fields = [
        'q_a',
        'q_b',
        'q_c',
                   ]
    def is_displayed(self):
        return self.player.nextq0 != 1

class Questionnaire5(Page):
    form_model = 'player'
    form_fields = ['fear_5a', 'b5']

    def is_displayed(self):
        return self.player.nextq0 != 1

    def before_next_page(self):
        self.player.participant.vars['b5'] = self.player.b5
        if self.player.b5 != 1 :
            self.player.eliminated = 2
            self.player.gaineliminated = Constants.bonus1
            self.player.participant.vars['eliminated'] = self.player.eliminated
            self.player.participant.vars['gaineliminated'] = self.player.gaineliminated

class Questionnaire5Bis(Page):
    timeout_seconds = 15
    form_model = 'player'
    form_fields = []

    def is_displayed(self):
        return self.player.nextq0 != 1 and self.player.b5 == 1

    def before_next_page(self):
        self.player.gaineliminated = 0
        self.player.participant.vars['eliminated'] = self.player.eliminated
        self.player.participant.vars['gaineliminated'] = self.player.gaineliminated
#-------------------------------------------------------------------------------------#

page_sequence = [
    P1Bienvenue,
    Questionnaire0,
    Questionnaire0Bis,
    P25QuestionnaireFinal,
    Questionnaire5,
    Questionnaire5Bis,
]