from otree.api import Currency as c, currency_range
from . import pages
from ._builtin import Bot
from .models import Constants

import random


class PlayerBot(Bot):

    def play_round(self):
        yield (pages.Start)
        if self.player.leader_nudge == 0:
            yield (pages.IntroVideoControl)
        elif self.player.leader_nudge == 1:
            yield (pages.IntroVideoTreatment)

        yield (pages.Send, {"sent_amount": 0})

        yield (pages.SendBack, {'sent_back_amount1': random.randint(1, 30),
                                'sent_back_amount2': random.randint(1, 60),
                                'sent_back_amount3': random.randint(1, 90),
                                'sent_back_amount4': random.randint(1, 120),
                                'sent_back_amount5': random.randint(1, 150),
                                'sent_back_amount6': random.randint(1, 180),
                                'sent_back_amount7': random.randint(1, 210),
                                'sent_back_amount8': random.randint(1, 240),
                                'sent_back_amount9': random.randint(1, 270),
                                'sent_back_amount10': random.randint(1, 300)
                                })