import Experiment as pages
from . import *
c = cu

class PlayerBot(Bot):
    def play_round(self):
        if self.player.role == C.MANAGER_ROLE:
            yield ManagerSelection, dict()
        if self.player.role == C.WORKER_A_ROLE:
            yield WorkerASelection, dict(worker_a_effort=0)
        if self.player.role == C.WORKER_B_ROLE:
            yield WorkerBSelection, dict(worker_b_effort=0)
        yield SetWaitPageArrival