from otree.api import Currency as c, currency_range
from otree.api import expect
from . import pages
from ._builtin import Bot
from .models import Constants, Player
from random import randint
from otree.api import Submission


class PlayerBot(Bot):

    def play_round(self):
        yield (pages.TaskGInstructionsRound1, {'grit_decision_round1': 'easy'})
        expect(self.player.grit_decision_round1, 'easy')
        yield (pages.TaskGRound1, {'grit_num_correct_round1': 3})
        expect(self.player.grit_num_correct_round1, 3)
        yield (pages.TaskGResultsRound1)
        if self.player.grit_round1_hardship:
            expect(self.player.grit_payoff_round1, 8)
        else:
            expect(self.player.grit_payoff_round1, 3)

        yield (pages.TaskGInstructionsRound2, {'grit_decision_round2': 'difficult'})
        expect(self.player.grit_decision_round2, 'difficult')
        yield (pages.TaskGRound2, {'grit_num_correct_round2': 3})
        expect(self.player.grit_num_correct_round2, 3)
        yield (pages.TaskGResultsRound2)
        expect(self.player.grit_payoff_round2, 8)