SI_InteligentnyWozekWidlowy/util/PathByEnum.py

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"