klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-10 21:29:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Create images directory if doesn't exist, to store image search results

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-07-28 21:30:31 +04:00
parent a29af70de5
commit 4d5183063c
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/search_type/image_search.py
View File

@@ -190,6 +190,10 @@ def collate_results(hits, image_names, output_directory, image_files_url, count=
target_image_name = f"{index}{source_path.suffix}" target_image_name = f"{index}{source_path.suffix}"
target_path = resolve_absolute_path(f"{output_directory}/{target_image_name}") target_path = resolve_absolute_path(f"{output_directory}/{target_image_name}")
# Create output directory, if it doesn't exist
if not target_path.parent.exists():
target_path.parent.mkdir(exist_ok=True)
# Copy the image to the output directory # Copy the image to the output directory
shutil.copy(source_path, target_path) shutil.copy(source_path, target_path)