import random
from ._builtin import Page, WaitPage
from .models import Constants


#class AA(Page):
#    form_model = 'player'
#    form_fields = ['a1', 'a2', 'a3', 'a4', 'a5', 'a6', 'a7', 'a8']

#    def get_form_fields(self):
#        fields = self.form_fields
#        random.shuffle(fields)
#        return fields


# class FA(Page):
    #    form_model = 'player'
    #    form_fields = ['f1','f2','f3','f4','f5','f6','f7','f8']

#    def get_form_fields(self):
#        fields = self.form_fields
#        random.shuffle(fields)
#        return fields


class PM(Page):
    form_model = 'player'
    form_fields = ['pm1', 'pm2', 'pm3', 'pm4', 'pm5', 'pm6', 'pm7', 'pm8', 'pm9', 'pm10', 'f3', 'f4', 'f6']

    def get_form_fields(self):
        fields = self.form_fields
        random.shuffle(fields)
        return fields


class EHA(Page):
    form_model = 'player'
    form_fields = ['e1', 'e2', 'e3', 'e4', 'h1', 'h2', 'h3', 'h4']

    def get_form_fields(self):
        fields = self.form_fields
        random.shuffle(fields)
        return fields


class K(Page):
    form_model = 'player'
    form_fields = ['k1','k2','k3','k4','k5','k6','k7','k8']

    def get_form_fields(self):
        fields = self.form_fields
        random.shuffle(fields)
        return fields

    def before_next_page(self):
        if self.player.k1 == 1:
            self.participant.payoff += 50
        else:
            pass
        if self.player.k2 == 1:
            self.participant.payoff += 50
        else:
            pass
        if self.player.k3 == 1:
            self.participant.payoff += 50
        else:
            pass
        if self.player.k4 == 1:
            self.participant.payoff += 50
        else:
            pass
        if self.player.k5 == 1:
            self.participant.payoff += 50
        else:
            pass
        if self.player.k6 == 1:
            self.participant.payoff += 50
        else:
            pass
        if self.player.k7 == 1:
            self.participant.payoff += 50
        else:
            pass
        if self.player.k8 == 1:
            self.participant.payoff += 50
        else:
            pass


page_sequence = [PM, EHA, K]