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


class Constants(BaseConstants):
    name_in_url = 'two_players_four_period_play'
    players_per_group = 2
    num_rounds = 1
    higher_pay = 20
    lower_pay = 10


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.StringField()
    invest_amount_r2 = models.IntegerField(label="כמה מטבעות תרצה/' להשקיע בנכס המסוכן?", min=0, max=100, blank=False)
    invest_profit_r2 = models.FloatField()
    rank_r2 = models.StringField()
    invest_amount_r3 = models.IntegerField(label="כמה מטבעות תרצה/' להשקיע בנכס המסוכן?", min=0, max=100, blank=False)
    invest_profit_r3 = models.FloatField()
    rank_r3 = models.StringField()
    invest_amount_r4 = models.IntegerField(label="כמה מטבעות תרצה/' להשקיע בנכס המסוכן?", min=0, max=100, blank=False)
    invest_profit_r4 = models.FloatField()
    rank_r4 = models.StringField()
    total_score = models.FloatField(initial=0)
    start_invest = models.FloatField()
    finish_invest = models.FloatField()