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


class PlayerBot(Bot):

    cases = [
        'success',  # players agree on an amount under the threshold
        'greedy',  # players ask for too much so end up with nothing
    ]

    def play_round(self):

        # start
        yield pages.Introduction

        if self.case == 'success':
            request = c(10)
            yield pages.Request, dict(request=request)
            yield pages.Results
            expect(self.player.payoff, request)

        if self.case == 'greedy':
            yield pages.Request, dict(request=Constants.amount_shared)
            yield pages.Results
            expect(self.player.payoff, 0)