from random import random from data.enum.ItemType import ItemType class PathByEnum: def get_random_path(self, item:ItemType): if item == ItemType.DOOR: a = str(random.randint(1, 10)) return "item_images/door/drzwi" + a + ".jpg" if item == ItemType.EGG: a = str(random.randint(1, 10)) return "item_images/refrigerator/lodowka" + a + ".jpg" if item == ItemType.SHELF: a = str(random.randint(1, 10)) return "item_images/shelf/polka" + a + ".jpg"