import pygame

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

class Element():
    def __init__(self, typ):
        self.wybrany = 0 
        self.lista_obrazow = []

        for i in range(1,4):
            scieszka = f"Images/{typ}{i}.png"
            wczytany_obraz = Obraz(scieszka)
            self.lista_obrazow.append(wczytany_obraz)

    def wybierzNastepny(self):
        self.wybrany += 1
        if self.wybrany > 2:
            self.wybrany = 0

    def wybranyObraz(self):
        return self.lista_obrazow[self.wybrany].obraz
    
class UbranieElement(Element):
    def __init__(self):
        super().__init__('body')

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

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

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


#cd '.\Lekcja 19,20\'