from . import pages
from ._builtin import Bot

import random


class PlayerBot(Bot):

    def play_round(self):

        yield (pages.QuestionnaireA, {
            'statement1': random.randint(0, 3),
            'statement2': random.randint(0, 3),
            'statement3': random.randint(0, 3),
            'statement4': random.randint(0, 3),
            'statement5': random.randint(0, 3),
            'statement6': random.randint(0, 3),
            'statement7': random.randint(0, 3),
            'statement8': random.randint(0, 3),
            'statement9': random.randint(0, 3),
            'statement10': random.randint(0, 3),
            'statement11': random.randint(0, 3),
            'statement12': random.randint(0, 3),
            'statement13': random.randint(0, 3),
            'statement14': random.randint(0, 3),
            'statement15': random.randint(0, 3),
            'statement16': random.randint(0, 3),
            'statement17': random.randint(0, 3),
            'statement18': random.randint(0, 3),
            'statement19': random.randint(0, 3),
            'statement20': random.randint(0, 3),
            'statement21': random.randint(0, 3),
            'statement22': random.randint(0, 3)})

        yield (pages.QuestionnaireB)

        yield (pages.QuestionnaireB1, {
            'rank1': random.randint(1, 4),
            'rank2': random.randint(1, 4),
            'rank3': random.randint(1, 4),
            'rank4': random.randint(1, 4)})

        yield (pages.QuestionnaireB2, {
            'choice': random.randint(1, 2)})

        yield (pages.QuestionnaireB3, {
            'belief': random.randint(0, 100)})

        yield (pages.QuestionnaireC, {
            'age': random.randint(18, 30),
            'gender': random.randint(1, 4),
            'nationality': random.randint(1, 9),
            'study': "Economics",
            'year': random.randint(1, 5)})

        yield (pages.End)

        yield (pages.Earnings)