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)
        else:
            pass
        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],
                                    'q11': Constants.answer_keys_covid[10],
                                    'q12': Constants.answer_keys_covid[11],
                                    'q13': Constants.answer_keys_covid[12],
                                    'q14': Constants.answer_keys_covid[13],
                                    'q15': Constants.answer_keys_covid[14]})
        yield (pages.End)