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

class new_step(Page):
    def is_displayed(self):
        return self.round_number == 1
    
    def vars_for_template(self):
        self.player.create_loteries()


#############################################################################################
#               CONSIGNE
##############################################################################################

class p1_consigne(Page):
    def vars_for_template(self):
        self.player.chosen_asso()

        boule = self.player.create_boule_entrainement()
        boule_prev = self.player.create_boule_prev_entrainement()
        return {
        'boule': boule,
        'boule_prev': boule_prev
    }
    
    def is_displayed(self):
        return self.round_number == 1


#############################################################################################
#               START GAME
##############################################################################################

class start_game(Page):
    def is_displayed(self):
        return self.round_number == 1
    

#############################################################################################
#               TREATMENT
##############################################################################################

class UrnePage_1(Page):
    form_model = 'player'
    form_fields = ['choix' + str(j) for j in range(1, Constants.num_choices + 1)]
    
    def vars_for_template(self):
        self.player.TREATMENT = "TRADEOFF_SELF_SURE_vs_CHARITY_LOT"
        self.player.chosen_asso()
        
        boule = self.player.create_boule()
        boule_prev = self.player.create_boule_prev()
        return {
        'boule': boule,
        'boule_prev': boule_prev
    }
    

# SELECTION POUR PAIEMENT
class FIN_PREV(Page):
    def is_displayed(self):
        return self.round_number == Constants.num_rounds



# Définissez la séquence de pages, en insérant les pages mélangées
page_sequence = [new_step, 
                p1_consigne, 
                start_game, 
                UrnePage_1, 
                FIN_PREV]