from otree.api import (
    models, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
)


class Constants(BaseConstants):
    name_in_url = 'four_players_four_period_play'
    players_per_group = 4
    num_rounds = 1
    first_pay = 25
    sec_pay = 15
    other_pay = 5


class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    asset_up_r1 = models.BooleanField()
    asset_up_r2 = models.BooleanField()
    asset_up_r3 = models.BooleanField()
    asset_up_r4 = models.BooleanField()


class Player(BasePlayer):
    invest_amount_r1 = models.IntegerField(label="כמה מטבעות תרצה/' להשקיע בנכס המסוכן?", min=0, max=100, blank=False)
    invest_profit_r1 = models.FloatField()
    rank_r1 = models.IntegerField()
    invest_amount_r2 = models.IntegerField(label="כמה מטבעות תרצה/' להשקיע בנכס המסוכן?", min=0, max=100, blank=False)
    invest_profit_r2 = models.FloatField()
    rank_r2 = models.IntegerField()
    invest_amount_r3 = models.IntegerField(label="כמה מטבעות תרצה/' להשקיע בנכס המסוכן?", min=0, max=100, blank=False)
    invest_profit_r3 = models.FloatField()
    rank_r3 = models.IntegerField()
    invest_amount_r4 = models.IntegerField(label="כמה מטבעות תרצה/' להשקיע בנכס המסוכן?", min=0, max=100, blank=False)
    invest_profit_r4 = models.FloatField()
    rank_r4 = models.IntegerField()
    total_score = models.FloatField(initial=0)
    payout = models.IntegerField()
    start_invest = models.FloatField()
    finish_invest = models.FloatField()