from otree.api import *
c = cu

doc = '\nThis is the ultimatum game. One player makes an offer and the other either accepts or rejects it.'
class Constants(BaseConstants):
    name_in_url = 'Ultimatum'
    players_per_group = 2
    num_rounds = 6
    endowment = 50
    instructions_template = 'Ultimatum/instructions.html'
def creating_session(subsession):
    session = subsession.session
    subsession.group_randomly()
class Subsession(BaseSubsession):
    pass
def set_payoffs(group):
    for p in group.get_players():
        set_payoff(p)
class Group(BaseGroup):
    proposal = models.IntegerField(max=Constants.endowment, min=0)
    response = models.StringField(choices=[['Accept', 'Accept'], ['Reject', 'Reject']], widget=widgets.RadioSelect)
def other_player(player):
    group = player.group
    return player.get_others_in_group()[0]
def set_payoff(player):
    group = player.group
    if player.id_in_group == 1:
        if group.response == "Accept":
            player.payoff = Constants.endowment - group.proposal
        else:
            player.payoff = 0
    else:
        if group.response == "Accept":
            player.payoff = group.proposal
        else:
            player.payoff = 0
class Player(BasePlayer):
    pass
class Introduction(Page):
    form_model = 'player'
    timeout_seconds = 100
    @staticmethod
    def is_displayed(player):
        return player.round_number == 1
class Proposer(Page):
    form_model = 'group'
    form_fields = ['proposal']
    @staticmethod
    def is_displayed(player):
        group = player.group
        return player.id_in_group == 1
class WaitForP1C1(WaitPage):
    @staticmethod
    def is_displayed(player):
        group = player.group
        return player.id_in_group == 2
class Reciever(Page):
    form_model = 'group'
    form_fields = ['response']
    @staticmethod
    def is_displayed(player):
        group = player.group
        return player.id_in_group == 2
class WaitForP2C2(WaitPage):
    @staticmethod
    def is_displayed(player):
        group = player.group
        return player.id_in_group == 1
class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs
class Results(Page):
    form_model = 'player'
page_sequence = [Introduction, Proposer, WaitForP1C1, Reciever, WaitForP2C2, ResultsWaitPage, Results]