import random from data.enum.ItemType import ItemType class PathByEnum: @staticmethod def get_random_path(item: ItemType): if item == ItemType.DOOR: a = str(random.randint(1, 10)) return "item_images/door/drzwi" + a + ".jpg" if item == ItemType.REFRIGERATOR: 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"