from otree.api import *


doc = """
Your app description
"""


class C(BaseConstants):
    NAME_IN_URL = 'trust_trial'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1
    ENDOWMENT = cu(10)
    MULTIPLIER= 3

class Subsession(BaseSubsession):
    pass


class Group(BaseGroup):
    give_amount = models.CurrencyField(
        choices = currency_range(cu(0), C.ENDOWMENT, cu(1)),
        label = "あなたはプレイヤー2にいくら渡しますか？",
        widget = widgets.RadioSelectHorizontal)
    
    back_amount = models.CurrencyField(
        label = "あなたはプレイヤー1にいくら返しますか？",
        widget = widgets.RadioSelectHorizontal)


class Player(BasePlayer):
    pass

def compute_payoffs(group: Group):
    p1 = group.get_player_by_id(1)
    p2 = group.get_player_by_id(2)
    p1.payoff = C.ENDOWMENT - group.give_amount + group.back_amount
    p2.payoff = group.give_amount * C.MULTIPLIER - group.back_amount

def back_amount_choices(group: Group):
    return currency_range(cu(0), group.give_amount * C.MULTIPLIER, cu(1))


# PAGES
class Page1(Page):
    pass

class Page2(Page):
    form_model = 'group'
    form_fields = ['give_amount']   

    @staticmethod
    def is_displayed(player: Player):
        return player.id_in_group == 1


class Page3(WaitPage):
    pass


class Page4(Page):
    form_model = 'group'
    form_fields = ['back_amount']

    @staticmethod
    def is_displayed(player: Player):
        return player.id_in_group == 2

    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(
            multi_amount = group.give_amount * C.MULTIPLIER
        )

class Page5(WaitPage):
    after_all_players_arrive = compute_payoffs

class Page6(Page):
    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(
            multi_amount = group.give_amount * C.MULTIPLIER
        )

page_sequence = [Page1, Page2, Page3, Page4, Page5, Page6]