from otree.api import Currency as c, currency_range, expect, Bot
from risk_tasks import *

from otree.api import Currency as c, currency_range, expect, Bot
from . import *


class PlayerBot(Bot):
    def play_round(self):
        if self.participant.vars['role'] == 'P' and self.player.round_number == 1:
            yield InstructionsP2
        if self.participant.vars['role'] == 'P' and self.player.round_number <= 3:
            yield HL, dict(A_1="A", A_2="A", A_3="A", A_4="A", A_5="A", A_6="B", A_7="B", A_8="B", A_9="B", A_10="B", A_11="B")
        if self.participant.vars['role'] == 'P' and self.player.round_number == 4:
            yield InstructionsP3
            yield Choice, dict(choice_s=1)
        if self.participant.vars['role'] == 'P' and self.player.round_number > 4:
            yield Choice, dict(choice_s=1)