also check for jpgs
This commit is contained in:
@@ -21,13 +21,13 @@ def get_path_image(image_name, track: str, config: str = ""):
|
||||
img_path = os.path.join(TRACKS_FOLDER, track, "ui", config)
|
||||
for name in os.listdir(img_path):
|
||||
lower = name.lower()
|
||||
if image_name in lower and lower.endswith(".png"):
|
||||
if image_name in lower and lower.endswith(".png") or lower.endswith(".jpg"):
|
||||
return os.path.join(img_path, name)
|
||||
|
||||
img_path = os.path.join(TRACKS_FOLDER, track, config)
|
||||
for name in os.listdir(img_path):
|
||||
lower = name.lower()
|
||||
if image_name in lower and lower.endswith(".png"):
|
||||
if image_name in lower and lower.endswith(".png") or lower.endswith(".jpg"):
|
||||
return os.path.join(img_path, name)
|
||||
|
||||
return ""
|
||||
|
||||
Reference in New Issue
Block a user