from otree.api import Currency as c, currency_range
from ._builtin import Page, WaitPage
from .models import Constants


class CRT(Page):

    form_model = 'player'
    form_fields = ['crt_bat',
                       'crt_widget',
                       'crt_lake']

    #check for correctness of CRT and calculate score before the player reaches the next page
    def before_next_page(self):
        self.player.set_is_crt_bat_correct()
        self.player.set_is_crt_lake_correct()
        self.player.set_is_crt_widget_correct()
        self.player.set_crt_score()

    def app_after_this_page(self, upcoming_apps):
        if self.participant.vars['treatment'] == 'seg_A':
            return upcoming_apps[1]

        if self.participant.vars['treatment'] == 'agg_B':
            return upcoming_apps[2]

        if self.participant.vars['treatment'] == 'seg_B':
            return upcoming_apps[3]


page_sequence = [CRT]