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


class Pre(Page):
    pass

class Pre2(Page):
    pass


class Sequence1(Page):
    form_model = 'player'
    form_fields = ['consumption_A1','saving_A1']



class Sequence2(Page):
    form_model = 'player'
    form_fields = ['consumption_A2','saving_A2']

    def vars_for_template(self):
        player = self.player
        player.income_A = (player.saving_A1 + 100)
        player.child_number = random.randint(1,4)
        if player.child_number == 1:
            player.consumption_C1 = 20
            player.saving_C1 = 0
            player.child = 'あなたの子供は将来を全く考えずに現在を楽しむためにお金お使うようです。'
        elif player.child_number == 2:
            player.consumption_C1 = 14
            player.saving_C1 = 6
            player.child = 'あなたの子供は一応貯金をしているようですが現在の消費にお金を使いがちです。'
        elif player.child_number == 3:
            player.consumption_C1 = 10
            player.saving_C1 = 10
            player.child = 'あなたの子供は貯金と現在の消費に使うお金を半々にしているようです。'
        else:
            player.consumption_C1 = 6
            player.saving_C1 = 14
            player.child = 'あなたの子供は現在の消費を切り詰めて将来のためになるべく貯金しているようです。'
        return {
            'consumption_C1':player.consumption_C1,
            'saving_C1':player.saving_C1,
            'child':player.child,
            'income_A':player.income_A,
        }


class Results(Page):
    def vars_for_template(self):
        player = self.player
        player.consumption_C2 = player.saving_C1 + player.saving_A2 - 100
        player.point_C = player.consumption_C1 * player.consumption_C2
        player.point_A = player.consumption_A1 * player.consumption_A2 * player.point_C

        return {
            'point_C':player.point_C,
            'point_A':player.point_A,
        }


page_sequence = [
    Pre,
    Pre2,
    Sequence1,
    Sequence2,
    Results,
]