from otree.api import *
from .Constants import *

class InfluenceMatrix(Page):
    form_model = "player"
    form_fields = [
        "player1_influence", 
        "player2_influence", 
        "player3_influence", 
        "player4_influence", 
        "ai_influence"
    ]
    
    def before_next_round(self):
        prev_round_score = self.group.in_previous_rounds()[-1].score
        self.session.vars['prev_round_score'] = prev_round_score

    def vars_for_template(self):
        selected_question = self.group.get_selected_question()
        players = self.group.get_players()
        id_to_player = {player.id_in_group: player for player in players}
        vars = {
            'correct_answer': selected_question['Answer']
        }
        for i in id_to_player:
            player = id_to_player[i]
            vars[f'player{i}_nickname'] = player.nickname
            vars[f'player{i}_color'] = player.color
            vars[f'player{i}_is_correct'] = player.correct_answer_selected
        return vars