from otree.api import *


doc = """
This application provides a webpage instructing participants how to get paid.
Examples are given for the lab and Amazon Mechanical Turk (AMT).
"""


class Constants(BaseConstants):
    name_in_url = 'payment_info'
    players_per_group = None
    num_rounds = 1


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    pass


class Player(BasePlayer):
    endow1 = models.FloatField()
    total = models.FloatField()
    total2 = models.FloatField()
    ftotal = models.FloatField()


# FUNCTIONS
# PAGES
class PaymentInfo(Page):
    @staticmethod
    def vars_for_template(player: Player):
        participant = player.participant
        participant.finished = True
        player.endow1 = participant.final_endow
        player.total = round(player.endow1,2)
        player.total2 = round((player.total * 0.005), 2)
        player.payoff = player.total2
        player.ftotal = round((player.total * 0.005) + 2, 2)


page_sequence = [PaymentInfo]