SI_InteligentnyWozekWidlowy/util/PathByEnum.py

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"