General: Better solution to bug.

This commit is contained in:
Dylan Araps 2017-06-28 20:38:11 +10:00
parent 153a7f92d2
commit a249f2083b

View File

@ -17,8 +17,8 @@ def random_img(img_dir):
current_wall = pathlib.Path(CACHE_DIR / "wal")
if current_wall.is_file():
wallpaper = util.read_file(current_wall)
current_wall = os.path.basename(wallpaper[0])
current_wall = util.read_file(current_wall)
current_wall = os.path.basename(current_wall[0])
# Add all images to a list excluding the current wallpaper.
file_types = (".png", ".jpg", ".jpeg", ".jpe", ".gif")
@ -27,7 +27,8 @@ def random_img(img_dir):
# If no images are found, use the current wallpaper.
if not images:
return pathlib.Path(wallpaper[0])
print("image: No new images found (nothing to do), exiting...")
quit(1)
return pathlib.Path(img_dir / random.choice(images).name)