from otree.api import *
c = cu

doc = '\nOne player decides how to divide a certain amount between himself and the other\nplayer.\nSee: Kahneman, Daniel, Jack L. Knetsch, and Richard H. Thaler. "Fairness\nand the assumptions of economics." Journal of business (1986):\nS285-S300.\n'
class C(BaseConstants):
    NAME_IN_URL = 'dictatorParallel'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1
    ENDOWMENT1 = cu(100)
    ENDOWMENT2 = cu(100)
    INSTRUCTIONS_TEMPLATE = 'dictatorParallel/instructions.html'
class Subsession(BaseSubsession):
    pass
class Group(BaseGroup):
    kept1 = models.CurrencyField(doc='Amount dictator decided to keep for himself', label='Ich möchte behalten:', max=C.ENDOWMENT1, min=0)
    kept2 = models.CurrencyField(label='Ich möchte behalten:', max=C.ENDOWMENT2, min=0)
def set_payoffs1(group: Group):
    p1 = group.get_player_by_id(1)
    p2 = group.get_player_by_id(2)
    p1.payoff = group.kept1 +(C.ENDOWMENT2 - group.kept2)
    p2.payoff = group.kept2 + (C.ENDOWMENT1 - group.kept1)
def set_payoffs2(group: Group):
    p1 = group.get_player_by_id(1)
    p2 = group.get_player_by_id(2)
    p1.payoff = group.kept1 +(C.ENDOWMENT2 - group.kept2)
    p2.payoff = group.kept2 + (C.ENDOWMENT1 - group.kept1)
class Player(BasePlayer):
    Geschlecht = models.StringField(blank=True, choices=[['Mann,Frau,Divers', 'Mann'], ['Frau', 'Frau'], ['Divers', 'Divers']])
class Introduction(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 1
class Angebot(Page):
    form_model = 'group'
    form_fields = ['kept1']
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 1
class Angebot2(Page):
    form_model = 'group'
    form_fields = ['kept2']
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 2
class ResultsWaitPage1(WaitPage):
    after_all_players_arrive = set_payoffs1
class ResultsWaitPage2(WaitPage):
    after_all_players_arrive = set_payoffs2
class Ergebnis1(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 1
        
        return True
    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(offer1=C.ENDOWMENT1 - group.kept1)
class Ergebnis2(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 2
        
        return True
    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(offer2=C.ENDOWMENT2 - group.kept2)
class Ende(Page):
    form_model = 'player'
    @staticmethod
    def is_displayed(player: Player):
        return player.round_number == C.NUM_ROUNDS
        
        return True
page_sequence = [Introduction, Angebot, Angebot2, ResultsWaitPage1, ResultsWaitPage2, Ergebnis1, Ergebnis2, Ende]