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


class MusicPage(Page):
    form_model = 'player'
    form_fields = ['level1', 'level2', 'level3']

class LandingPage(Page):
    def before_next_page(self):
        print(self.player.round_number)
    pass

class NDA(Page):
    pass

class Instructions(Page):
    pass

class DoNotTalkPage(Page):
    pass

class FirstSong(Page):
    form_model = 'player'
    form_fields = ['level1']

class SecondSong(Page):
    form_model = 'player'
    form_fields = ['level2']

class ThirdSong(Page):
    form_model = 'player'
    form_fields = ['level3']

class TransitionTo2(Page):
    def vars_for_template(self):
        all_players = self.subsession.get_players()
        group_players = int(len(all_players)/2)
        return {'group_players':group_players}
    pass


class Matching(WaitPage):
    title_text = "Group Formation"
    body_text = "The groups are now being formed. Please wait."
    def after_all_players_arrive(self):
        players = self.subsession.get_players()
        self.subsession.grouping_players(players)
        #self.subsession.print_player_vars(players)
        self.subsession.making_dictators(players)
        self.subsession.dictator_sequences(players)
        self.subsession.match(players)

page_sequence = [
    LandingPage,
    DoNotTalkPage,
    NDA,
    Instructions,
    FirstSong,
    SecondSong,
    ThirdSong,
    Matching,
    TransitionTo2,
]