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

import random

#--------------------------------------- jeu 1 undercutting (contre tout le monde)
class Quiz1(Page):
    form_model = 'player'
    form_fields = ['q_a1', 'q_a2']

    def is_displayed(self):
        return self.player.participant.vars['nextq0'] != 1 and self.player.participant.vars['treatment'] == 2 and self.player.participant.vars['b5'] == 1

    def error_message(self, values):
        print('values is', values)
        if values['q_a1'] != 7 :
            return 'Please correct your answer 1!!'
        elif values['q_a2'] != 28 :
            return 'Please correct your answer 2!!'

    def before_next_page(self):
        self.player.treatment = self.player.participant.vars['treatment']

class Quiz2(Page):
    form_model = 'player'
    form_fields = ['q_b1', 'q_b2']

    def is_displayed(self):
        return self.player.participant.vars['nextq0'] != 1 and self.player.participant.vars['treatment'] == 2 and self.player.participant.vars['b5'] == 1


    def error_message(self, values):
        print('values is', values)
        if values['q_b1'] != 11:
            return 'Please correct your answer 1!!'
        elif values['q_b2'] != 12:
            return 'Please correct your answer 2!!'

class MyPage1(Page):
    form_model = 'player'
    form_fields = ['choice1']

    def is_displayed(self):
        return self.player.participant.vars['nextq0'] != 1 and self.player.participant.vars['treatment'] == 2 and self.player.participant.vars['b5'] == 1


    def before_next_page(self):
        self.player.tirage1 = round(random.random(), 2)

        if self.player.tirage1 <= 0.54:  # voir table 2 (stage 3 - All)
            self.player.choicejoueur2 = 0
        elif self.player.tirage1 <= 0.62:  #
            self.player.choicejoueur2 = 1
        elif self.player.tirage1 <= 0.84:  #
            self.player.choicejoueur2 = 2
        elif self.player.tirage1 <= 0.96:  #
            self.player.choicejoueur2 = 3
        elif self.player.tirage1 <= 0.98:  #
            self.player.choicejoueur2 = 4
        else:
            self.player.choicejoueur2 = 6

        self.player.gaintask1 = Constants.l1[self.player.choice1][self.player.choicejoueur2]

# -------------------------------------------# new oct 2022
        self.player.participant.vars['gain_jeu2'] = self.player.gaintask1
# -------------------------------------------#

page_sequence = [
    #Quiz1,
    #Quiz2,
    MyPage1,
]