from otree.api import *
c = cu

doc = ''
class C(BaseConstants):
    NAME_IN_URL = 'Voto1'
    PLAYERS_PER_GROUP = 3
    NUM_ROUNDS = 1
class Subsession(BaseSubsession):
    pass
class Group(BaseGroup):
    ganador = models.StringField()
def Vote1(group: Group):
    session = group.session
    group = group
    players = group.get_players()
    
    if sum([p.vote for p in players]) <= 4 :
        group.ganador = "Igual repartición de agua"
    else :
        group.ganador = "Igual repartición de agua con mercado"
    
    group.session.vars["ganador1"] = group.ganador
    
    for p in players :
        if sum([p.vote for p in players]) <= 4 :
            p.participant.etapa5=p.participant.etapa1 
        else :
            p.participant.etapa5=p.participant.etapa2
    
    
Group.Vote1 = Vote1
class Player(BasePlayer):
    vote = models.IntegerField(choices=[[1, '1'], [2, '2']])
    type = models.IntegerField()
    type2 = models.IntegerField()
class IntroVoto(Page):
    form_model = 'player'
class Stage3intro(Page):
    form_model = 'player'
    form_fields = ['vote']
class Stage3calculation(WaitPage):
    after_all_players_arrive = Vote1
class Stage3results(Page):
    form_model = 'player'
page_sequence = [IntroVoto, Stage3intro, Stage3calculation, Stage3results]