from otree.api import Bot, SubmissionMustFail
from . import *
import random

class PlayerBot(Bot):
    def play_round(self):

        import random

        if self.player.round_number == 1:
            yield Introduction


        if self.player.participant.treatment == 'SelfSelection':
            if self.player.round_number == 1 or self.player.round_number == self.player.participant.PayRounds:
                number = random.choice([6, 9, 12])
                yield Explanation, dict(num_payrows=number)
#            if self.player.round_number == 1 or self.player.round_number == self.player.participant.PayRounds:
#                if self.player.round_number == self.player.participant.round_attention_check:
#                    number = random.choice([6, 9, 12])
#                    yield Explanation, dict(attention_check=1,
#                                            num_payrows=number)
        else:
            if self.player.round_number == 1:
                yield Explanation
            if self.player.round_number > 1 and self.player.round_number == self.player.participant.PayRounds:
                number = random.choice([9, 14, 18])
                yield Explanation, dict(num_payrows_assign=number)
#            if self.player.round_number == 1 or self.player.round_number == self.player.participant.PayRounds:
#                if self.player.round_number == self.player.participant.round_attention_check:
#                    yield Explanation, dict(attention_check=1)


        random_answer = random.randint(1, 12)
        yield Task, dict(Row_1=random_answer)


        if self.player.round_number == C.NUM_ROUNDS:
            yield ThankYou