extern/pywal
1
0
Fork 0
mirror of https://github.com/dylanaraps/pywal.git synced 2024-12-01 20:33:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Basic functionality

Browse Source
This commit is contained in:
Lorenz Leitner 2019-03-15 09:11:32 +01:00
parent e10337c838
commit e0c30a34a0
1 changed files with 38 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
pywal/image.py
View File

@ -12,6 +12,22 @@ from . import util
from . import wallpaper from . import wallpaper
def get_image_dir_recursive(img_dir):
"""Get all images in a directory recursively."""
current_wall = wallpaper.get()
current_wall = os.path.basename(current_wall)
file_types = (".png", ".jpg", ".jpeg", ".jpe", ".gif")
images = []
for path, subdirs, files in os.walk(img_dir):
for name in files:
print(os.path.join(path, name))
images.append(os.path.join(path, name))
return images, current_wall
def get_image_dir(img_dir): def get_image_dir(img_dir):
"""Get all images in a directory.""" """Get all images in a directory."""
current_wall = wallpaper.get() current_wall = wallpaper.get()
@ -31,16 +47,29 @@ def get_random_image(img_dir):
images.remove(current_wall) images.remove(current_wall)
elif not images: elif not images:
logging.error("No images found in directory, picking random directory.") logging.error("No images found in directory.")
logging.info("Found dirs: {}".format([dir_file.name for dir_file in
os.scandir(img_dir)]))
sys.exit(1) sys.exit(1)
random.shuffle(images) random.shuffle(images)
return os.path.join(img_dir, images[0]) return os.path.join(img_dir, images[0])
def get_random_image_recursive(img_dir):
"""Pick a random image file from a directory recursively."""
images, current_wall = get_image_dir_recursive(img_dir)
if len(images) > 2 and current_wall in images:
images.remove(current_wall)
elif not images:
logging.error("No images found in directory.")
sys.exit(1)
print(images)
random.shuffle(images)
return os.path.join("", images[0])
def get_next_image(img_dir): def get_next_image(img_dir):
"""Get the next image in a dir.""" """Get the next image in a dir."""
images, current_wall = get_image_dir(img_dir) images, current_wall = get_image_dir(img_dir)
@ -64,6 +93,7 @@ def get_next_image(img_dir):
def get(img, cache_dir=CACHE_DIR, iterative=False, recursive=False): def get(img, cache_dir=CACHE_DIR, iterative=False, recursive=False):
"""Validate image input.""" """Validate image input."""
recursive = True # TODO: Remove
if os.path.isfile(img): if os.path.isfile(img):
wal_img = img wal_img = img
@ -71,6 +101,9 @@ def get(img, cache_dir=CACHE_DIR, iterative=False, recursive=False):
if iterative: if iterative:
wal_img = get_next_image(img) wal_img = get_next_image(img)
else:
if recursive:
wal_img = get_random_image_recursive(img)
else: else:
wal_img = get_random_image(img) wal_img = get_random_image(img)