extern/pywal
1
0
Fork 0
mirror of https://github.com/dylanaraps/pywal.git synced 2025-02-18 03:01:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

General: Fix bug when shuffling in a directory with only one image.

Browse Source
This commit is contained in:
Dylan Araps 2017-06-28 20:21:22 +10:00
parent 3121a1a5c8
commit c00df332aa
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pywal/gen_colors.py
View File

@ -17,13 +17,17 @@ def random_img(img_dir):
current_wall = pathlib.Path(CACHE_DIR / "wal")
if current_wall.is_file():
current_wall = util.read_file(current_wall)
current_wall = os.path.basename(current_wall[0])
wallpaper = util.read_file(current_wall)
current_wall = os.path.basename(wallpaper[0])
# Add all images to a list excluding the current wallpaper.
file_types = (".png", ".jpg", ".jpeg", ".jpe", ".gif")
images = [img for img in os.scandir(img_dir)
if img.name.endswith(file_types) and img.name != current_wall]
if img.name.endswith(file_types) and img.name != wallpaper]
# If no images are found, use the current wallpaper.
if not images:
return pathlib.Path(wallpaper)
return pathlib.Path(img_dir / random.choice(images).name)