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

import random

class lotteryComplete(Page):
    pass


class surveyOne(Page): # BTN Items
    form_model = 'player'
    form_fields = ['schwartz_item_1',
                   'schwartz_item_2',
                   'schwartz_item_3',]

class surveyTwo(Page): # Schwarz Items

    form_model = 'player'
    form_fields = ['btn_item_1',
                   'btn_item_2',
                   'btn_item_3',
                   'btn_item_4', ]


    def vars_for_template(self):
        self.player.get_Bonus()
        return dict(
            bonus=self.participant.vars['final_bonus']
        )


page_sequence = [lotteryComplete, surveyOne, surveyTwo]