import random 

class CommonFunctions:

    def __eq__(self, other):
        return True

    def __init__(self):
        self.context_sequence_group1 = random.sample([i for i in range(1, 31)], 15)
        self.context_sequence_group2 = random.sample([i for i in range(1, 31)], 15)
        self.action_sequence_group1 = [random.sample([i for i in range(1, 5)], 4) for i in range(0, 15)]
        self.action_sequence_group2 = [random.sample([i for i in range(1, 5)], 4) for i in range(0, 15)]

    def getContextImagesGroup(self, group_number, round_number):
        if group_number == 1:
            return self.context_sequence_group1[round_number-1]
        else:
            return self.context_sequence_group2[round_number-1]

    def getActionImagesGroup(self, group_number, round_number):
        if group_number == 1:
            return self.action_sequence_group1[round_number-1]
        else:
            return self.action_sequence_group2[round_number-1]

    def getContextImagesPath(self, group_number, round_number):
        context_images_group = self.getContextImagesGroup(group_number, round_number)
        return [f'Image/{group_number}Env{context_images_group}{i}.png' for i in range(1, 5)]

    def getContextImagesName(self, group_number, round_number):
        context_images_group = self.getContextImagesGroup(group_number, round_number)
        return [f'{group_number}Env{context_images_group}{i}.png' for i in range(1, 5)]

    def getActionImagesPath(self, group_number, round_number):
        action_images_group = self.getActionImagesGroup(group_number, round_number)
        return [f'Image/{group_number}Action{i}.png' for i in action_images_group]

    def getSequenceOfTickCrossImages(self, players):
        return ['Image/tick.png' if player.check_action_choice == 1 else 'Image/cross.png' for player in players]