Untitled

mport pygame

# classe ausiliaria

class Image(pygame.sprite.Sprite):
    def __init__(self, path):
        super().__init__()
        self.image = pygame.image.load(path)

# classe base

class Element():
    def __init__(self, type):
        # indicatore del capo di abbigliamento scelto
        self.chosen = 0
        # lista delle immagini
        self.image_list = []
        # utilizzo di un ciclo per caricare tutte le immagini dalla directory
        for i in range(1, 4):
            path = f'images/{type}{i}.png'
            loaded_image = Image(path)
            self.image_list.append(loaded_image)

    def choose_next(self):
        self.chosen += 1
        if self.chosen > 2:
            self.chosen = 0

    def chosen_image(self):
        return self.image_list[self.chosen].image

class HeadCover(Element):
    def __init__(self):
        super().__init__('head')

class ClothingElement(Element):
    def __init__(self):
        super().__init__('body')

class EyesElement(Element):
    def __init__(self):
        super().__init__('eye')

class WeaponElement(Element):
    def __init__(self):
        super().__init__('weapon')