From a249f2083b5c88902eb656ecea2ba3e05962b74b Mon Sep 17 00:00:00 2001 From: Dylan Araps Date: Wed, 28 Jun 2017 20:38:11 +1000 Subject: [PATCH] General: Better solution to bug. --- pywal/gen_colors.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pywal/gen_colors.py b/pywal/gen_colors.py index e1e3478..3f5e16e 100755 --- a/pywal/gen_colors.py +++ b/pywal/gen_colors.py @@ -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)