from otree.api import Bot, Submission, cu
from . import Introduction, Send, SendBack, Results


class PlayerBot(Bot):
    def play_round(self):
        yield Introduction
        if self.player.id_in_group == 1:
            yield Submission(Send, dict(sent_amount=cu(3)))
        else:
            yield Submission(SendBack, dict(sent_back_amount=cu(4)))
        yield Results