import pygame
import random

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

class Element():
    def __init__(self, typ):
        # wskaznik wybranego elementu ubioru
        self.wybrany = 0
        # lista obrazow
        self.lista_obrazow = []
        for i in range(1, 5):
            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 >2:
            self.wybrany = 0

    def losujNastepny(self):
        self.wybrany = random.randint(0, 2)

    def wybrany_Obraz(self):
        return self.lista_obrazow[self.wybrany].obraz

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

class Zbroja(Element):
    def __init__(self):
        super().__init__('body')

class oczy(Element):
    def __init__(self):
        super().__init__('eye')
        
class bron(Element):
    def __init__(self):
        super().__init__('weapon')