from .models import *

class Overall_Instru(Page):
    form_model = 'player'
    def is_displayed(self):
        player = self.player
        return player.round_number == 1
class Instru_roll(Page):
    form_model = 'player'
    form_fields = ['final_die', 'all_dice']
class Report(Page):
    form_model = 'player'
    form_fields = ['report_num']
    def before_next_page(self):
        timeout_happened = self.timeout_happened
        session = self.session
        subsession = self.subsession
        player = self.player
        player = player
        if player.report_num==1:
            player.payoff=subsession.a1
        elif player.report_num==2:
            player.payoff=subsession.a2
        elif player.report_num==3:
            player.payoff=subsession.a3
        elif player.report_num==4:
            player.payoff=subsession.a4
        elif player.report_num==5:
            player.payoff=subsession.a5
        else:
            player.payoff=subsession.a6
class Calculation(Page):
    form_model = 'player'
    def is_displayed(self):
        player = self.player
        return player.round_number == 24 
    def before_next_page(self):
        timeout_happened = self.timeout_happened
        player = self.player
        player.my_method()
        player.total_payoff=player.in_round(player.s1).payoff+player.in_round(player.s2).payoff+player.in_round(player.s3).payoff+player.in_round(player.s4).payoff+player.in_round(player.s5).payoff+player.in_round(player.s6).payoff
        if player.total_payoff>6:
            player.total_payoff=6
        player.total_payoff_dollar=player.total_payoff+1
class Demographics(Page):
    form_model = 'player'
    form_fields = ['worker_id', 'assignment_id', 'income', 'seven11', 'seven12', 'seven13', 'seven14', 'seven15', 'seven16', 'seven21', 'seven22', 'seven23', 'seven24', 'seven25', 'seven26', 'seven31', 'seven32', 'seven33', 'seven34', 'seven35', 'seven36', 'seven41', 'seven42', 'seven43', 'seven44', 'seven45', 'seven46', 'nine11', 'nine12', 'nine13', 'nine14', 'nine21', 'nine22', 'nine23', 'nine24', 'nine31', 'nine32', 'nine33', 'nine34', 'ten11', 'ten12', 'ten13', 'ten14', 'ten15', 'ten21', 'ten22', 'ten23', 'ten24', 'ten25', 'ten31', 'ten32', 'ten33', 'ten35', 'ten34', 'ten41', 'ten42', 'ten43', 'ten44', 'ten45', 'eleven11', 'eleven12', 'eleven13', 'eleven14', 'eleven15', 'eleven16', 'eleven17', 'eleven21', 'eleven22', 'eleven23', 'eleven24', 'eleven25', 'eleven26', 'eleven27', 'eleven31', 'eleven32', 'eleven33', 'eleven34', 'eleven35', 'eleven36', 'eleven37', 'eleven41', 'eleven42', 'eleven43', 'eleven44', 'eleven45', 'eleven46', 'eleven47', 'twelve11', 'twelve12', 'twelve13', 'twelve15', 'twelve21', 'twelve22', 'twelve23', 'twelve24', 'twelve25', 'twelve31', 'twelve32', 'twelve33', 'twelve34', 'twelve35', 'twelve41', 'twelve42', 'twelve43', 'twelve44', 'twelve45', 'test_lie1', 'twelve14']
    def is_displayed(self):
        player = self.player
        return player.round_number == 24 
class Payoff(Page):
    form_model = 'player'
    def is_displayed(self):
        player = self.player
        return player.round_number == 24
page_sequence = [Overall_Instru, Instru_roll, Report, Calculation, Demographics, Payoff]