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


class Risk_Attitude_Task(Page):
    form_model = 'player'
    timeout_seconds = 60


class Bomb(Page):
    form_model = 'player'
    form_fields = ['deletedCount']

    def before_next_page(self):
        import random

        self.player.Bomb_count = self.player.deletedCount
        self.player.Bomb_location = random.randint(1, 100)
        if self.player.Bomb_count < self.player.Bomb_location:
            self.player.payoff_bomb = self.player.Bomb_count * 0.1
        else:
            self.player.payoff_bomb = 0


class Loss_aversion_task(Page):
    form_model = 'player'
    form_fields = ['Choice1', 'Choice2', 'Choice4', 'Choice3', 'Choice6', 'Choice5']

    def before_next_page(self):
        import random

        self.player.n = random.randint(1, 6)
        self.player.n2 = random.randint(1, 2)
        if self.player.n == 1:
            if self.player.Choice1 == True:
                if self.player.n2 == 1:
                    self.player.payoff_questionnaire = -0.03
                else:
                    self.player.payoff_questionnaire = 0.15
            else:
                self.player.payoff_questionnaire = 0
        elif self.player.n == 2:
            if self.player.Choice2 == True:
                if self.player.n2 == 1:
                    self.player.payoff_questionnaire = -0.06
                else:
                    self.player.payoff_questionnaire = 0.15
            else:
                self.player.payoff_questionnaire = 0
        elif self.player.n == 3:
            if self.player.Choice3 == True:
                if self.player.n2 == 1:
                    self.player.payoff_questionnaire = -0.09
                else:
                    self.player.payoff_questionnaire = 0.15
            else:
                self.player.payoff_questionnaire = 0
        elif self.player.n == 4:
            if self.player.Choice4 == True:
                if self.player.n2 == 1:
                    self.player.payoff_questionnaire = -0.12
                else:
                    self.player.payoff_questionnaire = 0.15
            else:
                self.player.payoff_questionnaire = 0
        elif self.player.n == 5:
            if self.player.Choice5 == True:
                if self.player.n2 == 1:
                    self.player.payoff_questionnaire = -0.15
                else:
                    self.player.payoff_questionnaire = 0.15
            else:
                self.player.payoff_questionnaire = 0
        else:
            if self.player.Choice6 == True:
                if self.player.n2 == 1:
                    self.player.payoff_questionnaire = -0.15
                else:
                    self.player.payoff_questionnaire = 0.15
            else:
                self.player.payoff_questionnaire = 0


class Final_Results(Page):
    form_model = 'player'
    timeout_seconds = 60


page_sequence = [Risk_Attitude_Task, Bomb, Loss_aversion_task, Final_Results]