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):
        # wskaźnik wybranego elementu
        self.wybrany = 0
        self.lista_obrazow = []

        # zaczytać wszystkie obrazy z folderu 
        for i in range(1,4):
            sciezka = f'images/{typ}{i}.png'
            wyczytany_obraz = Obraz(sciezka)
            self.lista_obrazow.append(wyczytany_obraz)
        
    def wybierz_nastepny(self):
        self.wybrany += 1
        if self.wybrany > 2:
            self.wybrany = 0
    
    def wybrany_obraz(self):
        return self.lista_obrazow[self.wybrany].obraz