import pygame

#klasa pomocnicza Obraz
class Obraz(pygame.sprite.Sprite):
    def __init__(self, sciezka):
        super().__init__()
        self.obraz = pygame.image.load(sciezka)

#klasa bazowa
class Element():
    def __init__(self, typ, ilosc_elementow):
        #wskaźnik wybranego elementu ubioru
        self.wybrany = 0
        #lista obrazów
        self.lista_obrazow = []
        #użycie pętli aby zaczytać wszystkie obrazy z folderu
        for i in range(1, ilosc_elementow + 1):
            sciezka = f'images/{typ}{i}.png'
            wczytany_obraz = Obraz(sciezka)
            self.lista_obrazow.append(wczytany_obraz)
    def wybierzNastepny(self):
        self.wybrany += 1
        if self.wybrany > len(self.lista_obrazow)-1:
            self.wybrany = 0
    def wybranyObraz(self):
        return self.lista_obrazow[self.wybrany].obraz
    
class NakrycieGlowy(Element):
    def __init__(self):
        super().__init__('head', 3)

class UbranieElement(Element):
    def __init__(self):
        super().__init__('body', 3)


class OczyElement(Element):
    def __init__(self):
        super().__init__('eye', 3)


class BronElement(Element):
    def __init__(self):
        super().__init__('weapon', 5)