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

import random

#--------------------------------------- jeu 2 undercutting (contre tout le monde)
class pagenext(Page):
    form_model = 'player'
    form_fields = []
    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.treatment = self.player.participant.vars['treatment']

class MyPage2(Page):
    form_model = 'player'
    form_fields = ['playagain']

    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):

        if self.player.playagain == 0:
            self.player.gaintask2 = Constants.surepayoff
            # -------------------------------------------# new oct 2022
            self.player.participant.vars['gain_jeu3'] = self.player.gaintask2
            # -------------------------------------------#

class MyPage3(Page):
    form_model = 'player'
    form_fields = ['choice2']

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

         #return self.player.playagain == 1 and

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

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

        self.player.gaintask2 = Constants.l1[self.player.choice2][self.player.choicejoueur2_task2]

# -------------------------------------------# new oct 2022
        self.player.participant.vars['gain_jeu3'] = self.player.gaintask2
# -------------------------------------------#

page_sequence = [
    #Quiz1,
    #Quiz2,
    #MyPage1,
    pagenext,
    MyPage2,
    MyPage3,
    #Results,
]