from otree.api import (
    models, widgets, BaseConstants, BaseSubsession, BaseGroup, BasePlayer,
    Currency as c, currency_range, Page, WaitPage
)
import csv
cu = c

doc = ''
class Constants(BaseConstants):
    name_in_url = 'practice_real_effort'
    players_per_group = 3
    payment_per_correct_answer = 1
    score_per_correct_answer = 1

    with open('s1_real_effort/answer_key.csv') as questions_file:
        questions = list(csv.DictReader(questions_file))

    # num_rounds = len(questions)
    num_rounds = 15

    player1_role = 'Player 1'
    player2_role = 'Player 2'
    player3_role = 'Player 3'   

class Subsession(BaseSubsession):  

    def creating_session(self):
        if self.round_number == 1:
            self.session.vars['questions'] = Constants.questions.copy()  

        for p in self.get_players():
            question_data = p.current_question()
            p.round_number = self.round_number

        for g in self.get_groups():  
            if (g.id_in_subsession == 1):
                g.question_id = int(question_data['question_id_1'])
                g.solution = int(question_data['solution_1'])
            elif ( g.id_in_subsession == 2):
                g.question_id = int(question_data['question_id_2'])
                g.solution = int(question_data['solution_2'])
            else:
                g.question_id = int(question_data['question_id_3'])
                g.solution = int(question_data['solution_3'])

    def set_timer(self):
        import time
        for player in self.get_players():
            player.participant.vars['expiry'] = time.time() + 3 * 60   

    def after_all_players_arrive(self):
        for group in self.get_groups():
            group.s1_team_check_correct()                      

class Group(BaseGroup):
    question_id = models.IntegerField()
    solution = models.IntegerField()
    s1_team_is_correct = models.IntegerField(initial=-1)
    s1_team_total_correct = models.IntegerField(initial=-1)     

    def s1_team_check_correct(group):
        players = group.get_players()
        team_answers = [p.s1_is_correct for p in players if p.s1_is_correct !=-1]
        if (sum(team_answers) == 3 and len(team_answers) == 3):
            group.s1_team_is_correct = 1
        elif (sum(team_answers) != 3 and len(team_answers) == 3):
            group.s1_team_is_correct = 0
        else:
            group.s1_team_is_correct = -1

        team_in_all_rounds = group.in_all_rounds()
        filterT = [g.s1_team_is_correct for g in team_in_all_rounds if g.s1_team_is_correct !=-1] 
        group.s1_team_total_correct = sum(filterT)
   
class Player(BasePlayer):
    s1_task_answers = models.IntegerField(label="Answer (the number of 4s):")
    s1_is_correct = models.IntegerField(initial=-1)
    s1_player_payoff = models.FloatField()
    s1_timeSpent = models.FloatField(initial=0)

    def role(self):
        if self.id_in_group == 1:
            return 'Player 1'
        if self.id_in_group == 2:
            return 'Player 2' 
        if self.id_in_group == 3:
            return 'Player 3'              

    def current_question(self):
        return self.session.vars['questions'][self.round_number - 1]

    def s1_check_correct(self):
        group = self.group
        if (self.s1_task_answers == self.group.solution and self.s1_task_answers != -1):
         self.s1_is_correct = 1
        elif (self.s1_task_answers != self.group.solution and self.s1_task_answers != -1): 
         self.s1_is_correct = 0
        else:
         self.s1_is_correct = -1