from otree.api import *
c = cu

doc = '\nOne player decides how to divide a certain amount between himself and the other\nplayer.\n\nSee: Kahneman, Daniel, Jack L. Knetsch, and Richard H. Thaler. "Fairness\nand the assumptions of economics." Journal of business (1986):\nS285-S300.\n\n'
class C(BaseConstants):
    NAME_IN_URL = 'dictator_high_endowment'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 4
    ENDOWMENT = cu(40)
    LANGUAGE = 'PL'
    INSTRUCTIONS_TEMPLATE = 'dictator_high_endowment/instructions.html'
    DALEJ_TEMPLATE = 'dictator_high_endowment/dalej.html'
class Subsession(BaseSubsession):
    pass
def pairs1low(subsession: Subsession):
    session = subsession.session
    if subsession.round_number==1:
        subsession.set_group_matrix([[1,4],[3,2]])
def pairs2low(subsession: Subsession):
    session = subsession.session
    if subsession.round_number==2:
        subsession.set_group_matrix([[1,2],[3,4]])
def pairs3low(subsession: Subsession):
    session = subsession.session
    if subsession.round_number==3:
        subsession.set_group_matrix([[1,3],[2,4]])
def pairs3b_low(subsession: Subsession):
    session = subsession.session
    if subsession.round_number==4:
        subsession.set_group_matrix([[3,1],[4,2]])
Subsession.pairs1low = pairs1low
Subsession.pairs2low = pairs2low
Subsession.pairs3low = pairs3low
Subsession.pairs3b_low = pairs3b_low
class Group(BaseGroup):
    kept = models.CurrencyField(doc='Amount dictator decided to keep for himself', label='Chcę zatrzymać', min=0)
    send = models.CurrencyField(label='To oznacza, że przekażę', min=0)
    satisfying = models.CurrencyField(label='W takim przypadku uważam, że satysfakcjonowałoby mnie otrzymanie', min=0)
def set_payoffs(group: Group):
    p1 = group.get_player_by_id(1)
    p2 = group.get_player_by_id(2)
    p1.payoff = group.kept
    p2.payoff = C.ENDOWMENT - group.kept
Group.set_payoffs = set_payoffs
class Player(BasePlayer):
    pass
class MyWaitPageRound5(WaitPage):
    wait_for_all_groups = True
    after_all_players_arrive = pairs1low
    title_text = 'Proszę czekać'
    body_text = 'Proszę zaczekać, aż pozostali uczestnicy eksperymentu podejmą decyzje'
    @staticmethod
    def is_displayed(player: Player):
        session = player.session
        subsession = player.subsession
        return subsession.round_number == 1
class MyWaitPageRound6(WaitPage):
    wait_for_all_groups = True
    after_all_players_arrive = pairs2low
    title_text = 'Proszę czekać'
    body_text = 'Proszę zaczekać, aż pozostali uczestnicy eksperymentu podejmą decyzje'
    @staticmethod
    def is_displayed(player: Player):
        session = player.session
        subsession = player.subsession
        return subsession.round_number == 2
class MyWaitPageRound7(WaitPage):
    wait_for_all_groups = True
    after_all_players_arrive = pairs3low
    title_text = 'Proszę czekać'
    body_text = 'Proszę zaczekać, aż pozostali uczestnicy eksperymentu podejmą decyzje'
    @staticmethod
    def is_displayed(player: Player):
        session = player.session
        subsession = player.subsession
        return subsession.round_number == 3
class MyWaitPageRound4(WaitPage):
    wait_for_all_groups = True
    after_all_players_arrive = pairs3b_low
    title_text = 'Proszę czekać'
    body_text = 'Proszę zaczekać, aż pozostali uczestnicy eksperymentu podejmą decyzje'
    @staticmethod
    def is_displayed(player: Player):
        session = player.session
        subsession = player.subsession
        return subsession.round_number == 4
class InfoRound(Page):
    form_model = 'player'
    @staticmethod
    def vars_for_template(player: Player):
        session = player.session
        subsession = player.subsession
        return dict(runda = subsession.round_number+4)
class Offer(Page):
    form_model = 'group'
    form_fields = ['kept', 'send']
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 1
    @staticmethod
    def error_message(player: Player, values):
        print('values is', values)
        if values['kept'] + values['send'] != 40:
            return 'Kwota przeznaczona do zatrzymania i kwota przesyłana do Gracza Odbierającego musi sumować się do 40'
class WaitScreenOnlyPlayer2(Page):
    form_model = 'group'
    form_fields = ['satisfying']
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 2
    @staticmethod
    def error_message(player: Player, values):
        print('values is', values)
        if values['satisfying'] > 40:
            return 'Kwota, którą  nie może być większa niż zasób posiadany przez Gracza Decydującego'
class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs
    title_text = 'Proszę czekać'
    body_text = 'Proszę zaczekać, aż pozostali uczestnicy eksperymentu podejmą decyzje'
class Results(Page):
    form_model = 'player'
    @staticmethod
    def vars_for_template(player: Player):
        session = player.session
        subsession = player.subsession
        group = player.group
        return dict(offer=C.ENDOWMENT - group.kept,
                    runda = subsession.round_number+4)
    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        session = player.session
        subsession = player.subsession
        participant = player.participant
        if subsession.round_number == 1:
            participant.vars["r5"] = player.payoff
        elif subsession.round_number == 2:
            participant.vars["r6"] = player.payoff
        elif subsession.round_number == 3:
            participant.vars["r7"] = player.payoff
        elif subsession.round_number == 4:
            participant.vars["r8"] = player.payoff
page_sequence = [MyWaitPageRound5, MyWaitPageRound6, MyWaitPageRound7, MyWaitPageRound4, InfoRound, Offer, WaitScreenOnlyPlayer2, ResultsWaitPage, Results]