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

class InputSubjectID(Page):
	form_model = models.Player
	form_fields = ['subject_ID', 'gender', 'age','timeonpage_InputSubjectID']
	def is_displayed(self):
		return self.round_number == 1
	def before_next_page(self):
		self.participant.vars['wrong_math_answers']=0
		self.participant.vars['cum_payoff']=0
		self.player.CreateAgents()


class MathProblemLevelOfEffort(Page):
	form_model = models.Player
	form_fields = ['problem_difficulty','timeonpage_MathProblemLevelOfEffort']
	
	def before_next_page(self):
		self.player.GetMathProblem(self.player.problem_difficulty)
		self.player.setCondition()

		

class MathProblemInput(Page):
	timeout_seconds = 90
	timeout_submission = None
	form_model = models.Player
	form_fields = ['input_answer','timeonpage_MathProblemInput']
	
	def before_next_page(self):
		self.player.CheckIfWrong(self.player.math_problem_ans, self.player.input_answer)
	
	def vars_for_template(self):
		return {'problem': self.player.math_problem}
	

class MathProblemFeedback(Page):
	timeout_seconds = 5
	form_model = models.Player
	form_fields=['timeonpage_MathProblemFeedback']
	
	def vars_for_template(self):
		return {'answer'	: self.player.math_problem_ans,
				'input_ans' : self.player.input_answer,
				'correct' : self.player.math_problem_ans == self.player.input_answer}
	
	def before_next_page(self):
		self.player.set_payoffs()

	body_text = "Waiting for other participants to contribute"
	

class GroupWaitPage(Page):
	timeout_seconds = random.randrange(0,models.Constants.GroupWaitPageMaxDelay+1)
		
	def vars_for_template(self):
		return {'title_text': "Wait Page"}


class Results(Page):
	form_model = models.Player
	form_fields=['timeonpage_Results']

	#timeout_seconds = 120
	def vars_for_template(self):
		return {
		'is_control': self.player.condition==0,
		'cum_payoff': self.participant.vars['cum_payoff'],
		'possible_payoff': self.round_number*130,
		'agents': self.participant.vars["agents"]
		}
	def before_next_page(self):
		self.player.create_counterfactual_json(),
		
		
		
# set to a number other than 1 if debugging rest of experiment
instructions_round_number = 1	 

	
class Instructions1(Page):
	form_model = models.Player
	form_fields = ['timeonpage_Instructions1']
	def is_displayed(self):
		return self.round_number == instructions_round_number

class Instructions2(Page):
	form_model = models.Player
	form_fields = ['timeonpage_Instructions2']
	def is_displayed(self):
		return self.round_number == instructions_round_number

class Instructions3(Page):
	form_model = models.Player
	form_fields = ['timeonpage_Instructions3']
	def is_displayed(self):
		return self.round_number == instructions_round_number

class Instructions4(Page):
	form_model = models.Player
	form_fields = ['timeonpage_Instructions4']
	def is_displayed(self):
		return self.round_number == instructions_round_number

class InstructionsQuiz(Page):
	form_model=models.Player
	form_fields=['IQ1','IQ2','IQ3','timeonpage_InstructionsQuiz']
	def is_displayed(self):
		return self.round_number == instructions_round_number

class InstructionsQuizFeedback(Page):
	form_model=models.Player
	form_fields=['timeonpage_InstructionsQuizFeedback']
	def is_displayed(self):
		return self.round_number == instructions_round_number
	def vars_for_template(self):
		return {
			'1_correct': self.player.IQ1==4,
			'2_correct': self.player.IQ2==1,
			'3_correct': self.player.IQ3==4
		}


class DebriefQuestions(Page):
	form_model = models.Player
	form_fields=['DBQ1','DBQ2','DBQ3','DBQ4','DBQ5','DBQ6','timeonpage_DebriefQuestions1']
	
	def before_next_page(self):
		self.player.CalculateTotalPayoff()
	
	def is_displayed(self):
		return self.round_number == models.Constants.num_rounds
	
	def get_form_fields(self):
		fields=self.form_fields[:(len(self.form_fields)-1)]
		random.shuffle(fields)
		return fields+self.form_fields[(len(self.form_fields)-1):]

class DebriefQuestions2(Page):
	form_model = models.Player
	form_fields=['DB_CF1','Debrief_CFComments','DB_CF2','Debrief_OtherComments','timeonpage_DebriefQuestions2']
	
	def before_next_page(self):
		self.player.CalculateTotalPayoff()
	
	def is_displayed(self):
		return self.round_number == models.Constants.num_rounds
	

class Counterfactuals_new(Page):
	form_model = models.Player
	form_fields=['timeonpage_Counterfactuals']
	def is_displayed(self):
		return self.player.condition!=0

# page_sequence = [Counterfactuals_new]

# modify sequence for testing, original seqeuence below
page_sequence = [
	GroupWaitPage,
	InputSubjectID,
	Instructions1,
	Instructions2,
	Instructions3,
	InstructionsQuiz,
	InstructionsQuizFeedback,
	GroupWaitPage,
	MathProblemLevelOfEffort,
	MathProblemInput,
	MathProblemFeedback,	
	GroupWaitPage,
	Results,
	Counterfactuals_new,
	DebriefQuestions,
	DebriefQuestions2
]