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


class PaymentInfo(Page):

    def vars_for_template(self):
        participant = self.participant
        # print(self.participant.vars)
        payoffs = [round(x/10000,2) for x in self.participant.vars['payoffs']]
        # print(payoffs)
        if "payoff_invest" in self.participant.vars:
            payoff_invest = round(self.participant.vars["payoff_invest"],2)
        else:
            payoff_invest =round(random.choice(payoffs),2)
            self.participant.vars["payoff_invest"]=payoff_invest
        payoff_dem = 1
        payoff_flquiz = round(self.participant.vars["num_right"]*0.25,2)
        payoff_total = round(payoff_invest + payoff_dem + payoff_flquiz,2)

        self.player.payoff=payoff_total


        return {
            'redemption_code': participant.label or participant.code,
            'payoffs': payoffs,
            'payoff_invest': payoff_invest,
            'payoff_dem': payoff_dem,
            'payoff_flquiz': payoff_flquiz,
            'payoff_total':payoff_total,
        }


page_sequence = [PaymentInfo]