from otree.api import Currency as c, currency_range
from . import pages
from ._builtin import Bot
from .models import Constants


class PlayerBot(Bot):

    def play_round(self):
        if self.player.id_in_group == 1 and self.group.CO < Constants.players_per_group:
            yield (pages.Player1, {'p1sel': 'Do not delete any link'})
        if self.player.id_in_group == 3 and self.group.CO < Constants.players_per_group:
            yield (pages.Player3, {'p3sel': 'Do not delete any link'})
        if self.player.id_in_group == 2 and self.group.CO < Constants.players_per_group:
            yield (pages.Player2, {'p2sel': 'Do not delete any link'})
        if self.group.CO >= Constants.players_per_group and self.group.done < 2:
            yield (pages.Results)