import pygame

class Picture():
    def __init__(self, image):
        self.picture = pygame.image.load(image)

class Element():

    def __init__(self, type):
        self.choice = 0
        self.photos = []

        for i in range(1, 4):
            image = f'images/{type}{i}.png'
            choosen_img = Picture(image)
            self.photos.append(choosen_img)

    def get_next(self):
        self.choice += 1
        if self.choice > 2:
            self.choice = 0

    def show_element(self):
        return self.photos[self.choice].picture
    
class Head(Element):
    def __init__(self):
        super().__init__('head')

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

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

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