import pygame

class Element():
    def __init__(self, typ, ilosc=3):
        self.lista_obrazów = []
        self.wybór = 1

        for i in range(1, ilosc+1):
            sciezka = f"resources/{typ}{i}.png"
            powierzchnia = pygame.image.load(sciezka)
            self.lista_obrazów.append(powierzchnia)

    def daj_wybrany(self):
        return self.lista_obrazów[self.wybór-1]
    


class NakrycieGlowy(Element):
    def __init__(self):
        super().__init__("head")

class Ubranie(Element):
    def __init__(self):
        super().__init__("body")
    
class Oczy(Element):
    def __init__(self):
        super().__init__("eye")

class Broń(Element):
    def __init__(self):
        super().__init__("weapon")