SI_InteligentnyWozekWidlowy/util/PathByEnum.py
Jakub-Prus e5a7a975e8 .
2022-06-02 11:13:21 +02:00

18 lines
556 B
Python

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"