From c4c7f38b15719d4db931a8ee3ccc3fe2772c81f7 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Wed, 20 Jul 2022 03:40:49 +0400 Subject: [PATCH] Fix extracting image names from multiple image directories --- src/search_type/image_search.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/search_type/image_search.py b/src/search_type/image_search.py index 6efcea7e..3cb2314d 100644 --- a/src/search_type/image_search.py +++ b/src/search_type/image_search.py @@ -34,11 +34,12 @@ def extract_entries(image_directories, verbose=0): image_names = [] for image_directory in image_directories: image_directory = resolve_absolute_path(image_directory, strict=True) - image_names = list(image_directory.glob('*.jpg')) + image_names.extend(list(image_directory.glob('*.jpg'))) image_names.extend(list(image_directory.glob('*.jpeg'))) if verbose > 0: - print(f'Found {len(image_names)} images in {image_directory}') + image_directory_names = ', '.join([str(image_directory) for image_directory in image_directories]) + print(f'Found {len(image_names)} images in {image_directory_names}') return image_names