SI_InteligentnyWozekWidlowy/util/PathByEnum.py

18 lines
550 B
Python
Raw Normal View History

2022-05-27 00:03:24 +02:00
import random
2022-05-26 19:22:51 +02:00
2022-05-26 19:27:40 +02:00
from data.enum.ItemType import ItemType
2022-05-26 19:22:51 +02:00
class PathByEnum:
2022-05-27 00:03:24 +02:00
@staticmethod
def get_random_path(item: ItemType):
2022-05-26 19:27:40 +02:00
if item == ItemType.DOOR:
2022-05-26 19:22:51 +02:00
a = str(random.randint(1, 10))
return "item_images/door/drzwi" + a + ".jpg"
if item == ItemType.FRIDGE:
2022-05-26 19:22:51 +02:00
a = str(random.randint(1, 10))
return "item_images/refrigerator/lodowka" + a + ".jpg"
2022-05-26 19:27:40 +02:00
if item == ItemType.SHELF:
2022-05-26 19:22:51 +02:00
a = str(random.randint(1, 10))
return "item_images/shelf/polka" + a + ".jpg"