import pygame

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


class Element:
    def __init__(self,typ):
        self.wybrane = 0
        self.lista_obrazuw = []
        
        for i in range(1,4):
            sciezka = f"images/{typ}{i}.png"
            wycytany_obraz = Obraz(sciezka)
            self.lista_obrazuw.append(wycytany_obraz)
    def wybierz_nastepny(self):
        self.wybrane += 1
        if self.wybrane >= 2:
            self.wybrane = 0
            


    def wybrany_obraz(self):
        return self.lista_obrazuw[self.wybrany].obraz