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.IntroPage
        if self.round_number == 41:
            yield pages.NewBlock
        if self.round_number > 40:
            if self.player.id_in_group == 1:
                yield pages.DecisionPage, {'contribute': True}
            else:
                yield pages.DecisionPage, {'contribute': False}
        else:
            if self.player.id_in_group == 1:
                yield pages.DecisionPage, {'contribute': True}
            else:
                yield pages.DecisionPage, {'contribute': False}
            if self.player.id_in_group == 1:
                yield pages.DecisionPage
            else:
                yield pages.DecisionPage, {'contribute': False}
            if self.player.id_in_group == 1:
                yield pages.DecisionPage
            else:
                yield pages.DecisionPage, {'contribute': False}
        yield pages.Results