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


class PlayerBot(Bot):
    def play_round(self):
        yield Questionnaire1, dict(dem_age=18, dem_gender=1, dem_study=1, dem_grad=1, dem_understand=1,)
        yield Questionnaire2, dict(attack_profit=1, attack_spite=1, attack_super=1, attack_fight1=1,
                                   attack_fight2=1, attack_encourage=1, attack_chance=1, attack_mistake=1,
                                   attack_understand=1)
        yield Questionnaire3, dict(noattack_risk=1, noattack_efficient=1,
                                   noattack_ownprofit=1, noattack_otherprofit=1,
                                   noattack_discourage=1, noattack_mistake=1,
                                   noattack_understand=1)
        if self.player.type == 1:
            yield Questionnaire4A, dict(arming_attack=1, arming_security=1,
                                        arming_noattack=1, arming_inefficient=1,
                                        arming_unequalA1=1,
                                        arming_unequalA2=1,
                                        arming_understand=1)
        if self.player.type == 2:
            yield Questionnaire4B, dict(arming_attack=1, arming_security=1,
                                        arming_noattack=1, arming_inefficient=1,
                                        arming_unequalB1=1,
                                        arming_unequalB2=1,
                                        arming_understand=1)