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

import random

class PlayerBot(Bot):

    def play_round(self):
        yield (pages.Introduction)
        yield (pages.ComprehensionCheck, {'comp_checkA': random.randint(0,1),
                                          'comp_checkB': random.randint(0,1),
                                          'comp_checkC': random.randint(0,1),
                                          'comp_checkD': random.randint(0,1),
                                          'comp_checkE': random.randint(0,1)})

        if self.player.comp_checkA == 0 or self.player.comp_checkB == 0 or self.player.comp_checkC == 1 or self.player.comp_checkD == 0 or self.player.comp_checkE == 0:
            yield (pages.ComprehensionCheckFail)
            yield (pages.StartPage)
        else:
            yield (pages.StartPage)
        yield (pages.QuestionPage, {'q1': Constants.answer_keys_covid[0],
                                    'q2': Constants.answer_keys_covid[1],
                                    'q3': Constants.answer_keys_covid[2],
                                    'q4': Constants.answer_keys_covid[3],
                                    'q5': Constants.answer_keys_covid[4],
                                    'q6': Constants.answer_keys_covid[5],
                                    'q7': Constants.answer_keys_covid[6],
                                    'q8': Constants.answer_keys_covid[7],
                                    'q9': Constants.answer_keys_covid[8],
                                    'q10': Constants.answer_keys_covid[9]})
        yield (pages.End)