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)
|
img_path = os.path.join(TRACKS_FOLDER, track, "ui", config)
|
||||||
for name in os.listdir(img_path):
|
for name in os.listdir(img_path):
|
||||||
lower = name.lower()
|
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 os.path.join(img_path, name)
|
||||||
|
|
||||||
img_path = os.path.join(TRACKS_FOLDER, track, config)
|
img_path = os.path.join(TRACKS_FOLDER, track, config)
|
||||||
for name in os.listdir(img_path):
|
for name in os.listdir(img_path):
|
||||||
lower = name.lower()
|
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 os.path.join(img_path, name)
|
||||||
|
|
||||||
return ""
|
return ""
|
||||||
|
|||||||
Reference in New Issue
Block a user