General: Better solution to bug.

dylanaraps · Jun 28, 2017 · a249f20 · a249f20
1 parent 153a7f9
commit a249f20
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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)