from . import pages
from ._builtin import Bot
from .models import Constants

import random


class PlayerBot(Bot):

    def play_round(self):

        yield (pages.Wait)

        if self.round_number == 21 or self.round_number == 41 or self.round_number == 61:
            yield (pages.Attention)

        if self.round_number == 1:

            yield (pages.DecisionP, {
                'practice_submitted_decision': random.randint(1, 2)})

            if self.player.practice_submitted_decision == 1:

                yield (pages.ChoiceP1, {
                    'practice_submitted_choice': random.randint(1, 2),
                    'new_practice_time_choice': random.randint(1, 10)})

                yield (pages.EstimateP, {
                    'practice_submitted_belief': random.randint(1, 100),
                    'new_practice_time_belief': random.randint(1, 10)})

            if self.player.practice_submitted_decision == 2:

                yield (pages.EstimateP, {
                    'practice_submitted_belief': random.randint(1, 100),
                    'new_practice_time_belief': random.randint(1, 10)})

                yield (pages.ChoiceP2, {
                    'practice_submitted_choice': random.randint(1, 2),
                    'new_practice_time_choice': random.randint(1, 10)})

            yield (pages.FeedbackP)

        yield (pages.Decision, {
            'submitted_decision': random.randint(1, 2)})

        if self.player.submitted_decision == 1:

            yield (pages.Choice1, {
                'submitted_choice': random.randint(1, 2),
                'new_time_choice': random.randint(1, 10)})

            yield (pages.Estimate, {
                'submitted_belief': random.randint(1, 100),
                'new_time_belief': random.randint(1, 10)})

        if self.player.submitted_decision == 2:

            yield (pages.Estimate, {
                'submitted_belief': random.randint(1, 100),
                'new_time_belief': random.randint(1, 10)})

            yield (pages.Choice2, {
                'submitted_choice': random.randint(1, 2),
                'new_time_choice': random.randint(1, 10)})

        yield (pages.Feedback)

        if self.round_number == Constants.num_rounds:
            yield (pages.End)