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

import random

class new_step(Page):
    def is_displayed(self): 
        return self.round_number == 1
    
    def before_next_page(self):
        self.player.state = random.choice(['aligned', 'unaligned'])

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


class ChoiceToRevealPage(Page):
    form_model = 'player'
    form_fields = ['reveal_state']

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

    def is_displayed(self): 
        return self.round_number == 1


class MakeChoicePage(Page):
    form_model = 'player'
    form_fields = ['choice']

    def is_displayed(self): 
        return self.round_number == 1



class RevealedUnalignedPage(Page):
    form_model = 'player'
    form_fields = ['choice']

    def vars_for_template(self):
        self.player.state = "unaligned"

    def is_displayed(self): 
        return self.round_number == 2



class RevealedAlignedPage(Page):
    form_model = 'player'
    form_fields = ['choice']

    def vars_for_template(self):
        self.player.state = "aligned"

    def is_displayed(self): 
        return self.round_number == 3

    def before_next_page(self):
        self.player.paiement_dana()







# Modifiez la séquence de pages
page_sequence = [new_step, consigne_dana, ChoiceToRevealPage, MakeChoicePage, RevealedUnalignedPage, RevealedAlignedPage]