from otree.api import expect, Submission
from IPDARM_EQUAL_POST import pages
from ._builtin import Bot
from IPDARM_EQUAL_POST.models import Constants


class PlayerBot(Bot):
    def play_round(self):
        if self.player.round_number == 1:
            yield pages.Introduction

        yield Submission(pages.Temp, check_html=False)

        if self.participant.vars['role'] == "A":
            yield pages.Decision, dict(ADA=0, ANA=0, AD1=0, AN1A=0, decision=0)

        if self.participant.vars['role'] == "B":
            yield pages.Decision, dict(ADB=0, ANB=0, AD1=0, AN1B=0, decision=0)

        yield Submission(pages.Temp2, check_html=False)

        yield pages.Results

        yield pages.Communication, dict(Message='Vous jouez avec un robot pour cette période, il effectuera le choix Y par défault')

        yield pages.Reception