from otree.api import *
c = cu

doc = 'Initial Room Division For 4 to 1  Divided Public Goods Game'
class Constants(BaseConstants):
    name_in_url = 'InitialRoomD_4t1Division'
    players_per_group = None
    num_rounds = 1
class Subsession(BaseSubsession):
    pass
class Group(BaseGroup):
    pass
class Player(BasePlayer):
    pass
class PlayerDivision(Page):
    form_model = 'player'
    @staticmethod
    def app_after_this_page(player, upcoming_apps):
        session = player.session
        participant = player.participant
        if player.participant.vars['partyid'] == "Democrat":
            d = 0
            m = False
            for pl in player.participant.get_players():
                if pl.participant.vars['partyid'] == "Democrat":
                    d = d + 1
                if player.participant.id_in_session == pl.participant.id_in_session and d % 5 == 0:
                    m = True
            if m == True:
                return upcoming_apps[1]
            else:
                return upcoming_apps[0]
        else:
            r = 0
            m = False
            for pl in player.participant.get_players():
                if pl.participant.vars['partyid'] == "Republican":
                    r = r + 1
                if player.participant.id_in_session == pl.participant.id_in_session and r % 5 == 0:
                    m = True
            if m == True:
                return upcoming_apps[0]
            else:
                return upcoming_apps[1]
page_sequence = [PlayerDivision]