from otree.api import *
c = cu

doc = ''
class C(BaseConstants):
    NAME_IN_URL = 'dictator1'
    PLAYERS_PER_GROUP = 2
    NUM_ROUNDS = 1
    ENDOWMENT = cu(158)
    INSTRUCTIONS_TEMPLATE = 'dictator1/instructions.html'
class Subsession(BaseSubsession):
    pass
class Group(BaseGroup):
    kept = models.CurrencyField(label='Ich möchte behalten:', max=C.ENDOWMENT, 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
class Player(BasePlayer):
    UserID = models.LongStringField(label='Bitte geben Sie Ihre zugewiesene Spieler-ID ein:')
    Spende = models.IntegerField(label='Angenommen, Sie werden Gewinner des Spiels. Bitte geben Sie an, welchen Anteil des Gewinns ( in Prozent)  Sie an eine wohltätige Einrichtung spenden möchten.', max=100, min=0)
    Einrichtung = models.StringField(blank=True, choices=[['Caritas', 'Caritas'], ['Diakonie', 'Diakonie'], ['Misereor', 'Misereor'], ['Brot für die Welt', 'Brot für die Welt'], ['Unicef', 'Unicef'], ['Greenpeace', 'Greenpeace'], ['Aktion Deutschland Hilft (Türkei Erdbeben)', 'Aktion Deutschland Hilft (Türkei Erdbeben)'], ['Aktion Deutschland Hilft (Ukraine)', 'Aktion Deutschland Hilft (Ukraine)']], label='An welche der folgenden Einrichtungen soll gespendet werden:')
class Introduction(Page):
    form_model = 'player'
    form_fields = ['UserID', 'Spende', 'Einrichtung']
    @staticmethod
    def before_next_page(player: Player, timeout_happened):
        participant = player.participant
        player.participant.vars['UserID'] = player.UserID
        player.participant.vars['Spende'] = player.Spende
        player.participant.vars['Einrichtung'] = player.Einrichtung
        
        
class Angebot(Page):
    form_model = 'group'
    form_fields = ['kept']
    @staticmethod
    def is_displayed(player: Player):
        group = player.group
        return player.id_in_group == 1
class ResultsWaitPage(WaitPage):
    after_all_players_arrive = set_payoffs
class Ergebnis(Page):
    form_model = 'player'
    @staticmethod
    def vars_for_template(player: Player):
        group = player.group
        return dict(offer=C.ENDOWMENT - group.kept)
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, ResultsWaitPage, Ergebnis, Ende]