# Definition of views/pages for the survey.
# Please note: When using oTree 2.x, this file should be called "pages.py" instead of "views.py"
#

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

from otreeutils.surveys import SurveyPage, setup_survey_pages

import random

class ExtraShop01(Page):
    form_model = 'player'
    form_fields = ['extra_01']
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['ES01']
    def before_next_page(self):
        shop2value = {0: 0, 1: 0.4,  2: 0.2,   3: 0.7,   4: 0,
                      5: 0.6,    6: 0.1, 7: 0.3, 8: 0.5};
        self.player.payoff += shop2value[self.player.extra_01]

class ExtraShop02(Page):
    form_model = 'player'
    form_fields = ['extra_02']
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['ES02']
    def before_next_page(self):
        shop2value = {0: 0, 1: -0.4,   2: -0.1,   3: -0.2,   4: -0.3,
                      5: 0.1,   6: 0.4, 7: 0.2, 8: 0.3};
        self.player.payoff += shop2value[self.player.extra_02]

class ExtraShop03(Page):
    form_model = 'player'
    form_fields = ['extra_03']
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['ES03']
    def before_next_page(self):
        shop2value = {0: 0, 1: -0.1,   2: -0.4,   3: 0.3,   4: 0.2,
                      5: -0.3,   6: -0.2, 7: 0.1, 8: 0.4};
        self.player.payoff += shop2value[self.player.extra_03]

class ExtraShop04(Page):
    form_model = 'player'
    form_fields = ['extra_04']
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['ES04']
    def before_next_page(self):
        shop2value = {0: 0, 1: 0,   2: 0,   3: -0.2,   4: 0.4,
                      5: -0.3,   6: -0.2, 7: 0.2, 8: 0.2};
        self.player.payoff += shop2value[self.player.extra_04]

class ExtraShop05(Page):
    form_model = 'player'
    form_fields = ['extra_05']
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['ES05']
    def before_next_page(self):
        shop2value = {0: 0, 1: 0.1,   2: 0.4,   3: 0.1,   4: 0.7,
                      5: 0,   6: 0.1, 7: 0.5, 8: 0.6};
        self.player.payoff += shop2value[self.player.extra_05]

class ExtraShop06(Page):
    form_model = 'player'
    form_fields = ['extra_06']
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['ES06']
    def before_next_page(self):
        shop2value = {0: 0, 1: -0.3,   2: 0.3,   3: -0.1,   4: 0.3,
                      5: 0,   6: 0.1, 7: -0.2, 8: 0.1};
        self.player.payoff += shop2value[self.player.extra_06]

class ListShop01(Page):
    form_model = 'player'
    form_fields = ['list_01']
   
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['LS01']
    def before_next_page(self):
        shop2value = {0: 0, 1: 0,   2: 0.6,   3: 0.5,   4: 0.3,
                      5: 0.1,   6: 0.7, 7: 0.2, 8: 0.4};
        self.player.payoff += shop2value[self.player.list_01]

class ListShop02(Page):
    form_model = 'player'
    form_fields = ['list_02']
   
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['LS02']
    def before_next_page(self):
        shop2value = {0: 0, 1: 0.7,   2: 0.1,   3: 0.4,   4: 0,
                      5: 0.2,   6: 0.5, 7: 0.6, 8: 0.2};
        self.player.payoff += shop2value[self.player.list_02]

class ListShop03(Page):
    form_model = 'player'
    form_fields = ['list_03']
    
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['LS03']
    def before_next_page(self):
        shop2value = {0: 0, 1: 0.6,   2: 0.7,   3: 0.4,   4: 0.1,
                      5: 0.3,   6: 0.2, 7: 0, 8: 0.4};
        self.player.payoff += shop2value[self.player.list_03]

class ListShop04(Page):
    form_model = 'player'
    form_fields = ['list_04']
    
    def is_displayed(self):
        return self.round_number == self.participant.vars['round']['LS04']
    def before_next_page(self):
        shop2value = {0: 0, 1: 0.6,   2: 0.3,   3: 0.4,   4: 0.2,
                      5: 0.6,   6: 0.1, 7: 0, 8: 0.4};
        self.player.payoff += shop2value[self.player.list_04]


page_sequence = [
    ExtraShop01,  #Potato-Impuls
    ExtraShop02,  #Corkscrew-Impuls
    ExtraShop03,  #Flower-Impuls
    ExtraShop04,  #Napkins-Impuls
    ExtraShop05,  #Onion-Impuls
    ExtraShop06,  #Tiramisu-Impuls
    ListShop01,  #Salmon-List
    ListShop02,  #White Wine-List
    ListShop03,  #Lettuce-List
    ListShop04,  #Tomato-List
]