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.round_number == 1:
            yield (pages.Welcome)
            yield (pages.Instructions)
            yield (pages.Test)
        if self.round_number in [Constants.belief_round,
                                 Constants.num_rounds/2 - Constants.belief_round,
                                 Constants.num_rounds/2 + Constants.belief_round,
                                 Constants.num_rounds - Constants.belief_round]:
            yield (pages.Beliefs, {'belief_sender': 0, 'belief_receiver': 100})
        if self.player.role() == 'sender':
            yield (pages.Sender, {'message': self.group.state})
        if self.player.role() == 'receiver':
            yield (pages.Receiver, {'action': self.group.message})
        yield (pages.Results)