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



class finish_game(Page):
    form_model = 'player'

    def vars_for_template(self):
        if self.round_number == 1:
            self.participant.vars['payoff_pilot3'] = self.player.payoff_pilot3

        if self.participant.vars['paying_round'] == 1:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][0]

        if self.participant.vars['paying_round'] == 2:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][1]

        if self.participant.vars['paying_round'] == 3:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][2]

        if self.participant.vars['paying_round'] == 4:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][3]

        if self.participant.vars['paying_round'] == 5:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][4]

        if self.participant.vars['paying_round'] == 6:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][5]

        if self.participant.vars['paying_round'] == 7:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][6]

        if self.participant.vars['paying_round'] == 8:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][7]

        if self.participant.vars['paying_round'] == 9:
            self.player.payoff_pilot3 = self.participant.vars['payoff_choice'][0]

        return dict(
            solved=self.participant.vars['solved'],
            time=self.participant.vars['time'],
            moves=self.participant.vars['moves'],
            endow=self.participant.vars['endow'],
            payoff=self.participant.vars['payoff'],
            bonus=self.participant.vars['bonus'],
            deduction=self.participant.vars['deduction'],
            payinground=self.participant.vars['paying_round'],
            payoffpilot3=self.participant.vars['payoff_pilot3'],
            solved_choice=self.participant.vars['solved_choice'],
            time_choice=self.participant.vars['time_choice'],
            deduction_choice=self.participant.vars['deduction_choice'],
            moves_choice=self.participant.vars['moves_choice'],
            endow_choice=self.participant.vars['endow_choice'],
            payoff_choice=self.participant.vars['payoff_choice'],
            bonus_choice=self.participant.vars['bonus_choice'],
        )

    def js_vars(self):
        return {
            'paying_round': self.participant.vars['paying_round'],
            'payoff_pilot3': self.participant.vars['payoff_pilot3'],
        }


class PaymentInfo(Page):
    form_model = 'player'
    form_fields = ['paying_round', 'payoff_pilot3']

    def vars_for_template(self):
        if self.round_number == 1:
            self.participant.vars['payoff_pilot3'] = self.player.payoff_pilot3

        if self.participant.vars['paying_round'] == 1:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][0]

        if self.participant.vars['paying_round'] == 2:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][1]

        if self.participant.vars['paying_round'] == 3:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][2]

        if self.participant.vars['paying_round'] == 4:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][3]

        if self.participant.vars['paying_round'] == 5:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][4]

        if self.participant.vars['paying_round'] == 6:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][5]

        if self.participant.vars['paying_round'] == 7:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][6]

        if self.participant.vars['paying_round'] == 8:
            self.player.payoff_pilot3 = self.participant.vars['payoff'][7]

        if self.participant.vars['paying_round'] == 9:
            self.player.payoff_pilot3 = self.participant.vars['payoff_choice'][0]

        return dict(
            solved=self.participant.vars['solved'],
            time=self.participant.vars['time'],
            moves=self.participant.vars['moves'],
            endow=self.participant.vars['endow'],
            payoff=self.participant.vars['payoff'],
            bonus=self.participant.vars['bonus'],
            deduction=self.participant.vars['deduction'],
            payinground=self.participant.vars['paying_round'],
            payoffpilot3=self.participant.vars['payoff_pilot3'],
            solved_choice=self.participant.vars['solved_choice'],
            time_choice=self.participant.vars['time_choice'],
            deduction_choice=self.participant.vars['deduction_choice'],
            moves_choice=self.participant.vars['moves_choice'],
            endow_choice=self.participant.vars['endow_choice'],
            payoff_choice=self.participant.vars['payoff_choice'],
            bonus_choice=self.participant.vars['bonus_choice'],
        )

    def js_vars(self):
        return {
            'paying_round': self.participant.vars['paying_round'],
            'payoff_pilot3': self.participant.vars['payoff_pilot3'],
        }


page_sequence = [finish_game, PaymentInfo]