fix for source path

Author: Acidham

Image Shrinker.alfredworkflow

@@ -16,20 +16,48 @@
     'heic',
 ]
 
+image_files = []
+for file_path in sys.argv[1:]:
+    file_extension = Path(file_path).suffix[1:].lower()
+    if file_extension in WHITELIST_EXT:
+        img_suffix = os.getenv('img_suffix')
+        target_folder = os.getenv('target_folder')
+        path = Path(file_path)
+
+        if not (os.getenv('target_folder')) and not (img_suffix):
+            img_suffix = '_shrinked'
+
+        if target_folder:
+            path = Path(target_folder) / path.name
+
+        if img_suffix:
+            new_file_name = f"{path.stem}{img_suffix}{path.suffix}"
+            new_file_path = path.with_name(new_file_name)
+            shutil.copyfile(file_path, new_file_path)
+            file_path = str(new_file_path)
+
+        image_files.append(file_path)
+"""
 image_files = []
 for f in sys.argv[1::]:
     ext = Path(f).suffix.replace('.', '').lower()
     if ext in WHITELIST_EXT:
+        img_suffix = os.getenv('img_suffix')
+        path, filename = os.path.split(f)
+        if not (os.getenv('target_folder')) and not (img_suffix):
+            img_suffix = '_shrinked'
+        if os.getenv('target_folder'):
+            path = os.getenv('target_folder')
         # create a copy if img_suffix is not empty in WF config
-        if os.getenv('img_suffix'):
-            path, filename = os.path.split(f)
+        if img_suffix:
             basename, extension = os.path.splitext(filename)
-            new_file_name = f"{basename}{os.getenv('img_suffix')}{extension}"
+            new_file_name = f"{basename}{img_suffix}{extension}"
             new_file = os.path.join(path, new_file_name)
             shutil.copyfile(f, new_file)
             f = new_file
 
         image_files.append(f)
+"""
 
 files = "\t".join(image_files)
 

src/filecheck.py

@@ -0,0 +1,6 @@
+import os
+import sys
+
+source_file = sys.argv[1]
+source_path = os.path.dirname(source_file)
+sys.stdout.write(source_path)