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):
        yield Start
        if self.participant.vars['role'] == 'P' and self.participant.vars['treatment'] == 'OB':
            yield BonusReasons, dict(Choice_quality=1, Outcome_quality=2, Counterfactual=3, altruism=4)
        if self.participant.vars['role'] == 'P' and self.participant.vars['treatment'] == 'NOB':
            yield BonusReasons, dict(Choice_quality=1, altruism=4)
        if self.participant.vars['role'] == 'A':
            yield DecisionReasons, dict(A_good=1, A_bad=2, A_bonus=3)
            yield Comprehension, dict(b_cond_choice=1, b_cond_outcomes=1)


        yield Cognitive2, dict(Crt_barrel=4, Crt_student=29, Crt_pig=20, Crt_simon=1, Know2=0)
        yield Cognitive, dict(Crt_Bat=4, Crt_Widget=29, Crt_Lake=20, Know=0)
        yield Risk, dict(Q_risk=3)
        yield Demographics, dict(Age=23, gender='Male', Nation='Bavaria', Major='Law', Education='Masters', HH_income='1000-1500' )